AI/rocm-systems
2
0
Garpu 0
Kode Masalah Tarik Permintaan Actions Packages Projects Rilis Wiki Kegiatan

SWDEV-466601 - Fix invalid mem acccess in kernarg readback path

Telusuri Sumber 
Change-Id: I4654ae592adc8cf9c687136d45eb1b28d99c7ae1


[ROCm/clr commit: 055e05a12a]
This commit is contained in:
Ioannis Assiouras
2024-06-08 02:56:54 +01:00
orang tua 30c4d5805e
melakukan efba3545ca
2 mengubah file dengan 3 tambahan dan 3 penghapusan
Download Patch File Download Diff File Expand all files Collapse all files
projects/clr/hipamd/src/hip_graph_internal.cpp
+2 -2
Melihat File
@@ -438,11 +438,11 @@ hipError_t GraphExec::CaptureAQLPackets() {
} else if (kernArgImpl == KernelArgImpl::DeviceKernelArgsReadback &&
kernarg_pool_size_graph_ != 0) {
address dev_ptr = kernarg_pool_graph_ + kernarg_pool_size_graph_;
auto kSentinel = *reinterpret_cast<volatile int*>(dev_ptr);
auto kSentinel = *reinterpret_cast<volatile address>(dev_ptr - 1);
_mm_sfence();
*(dev_ptr - 1) = kSentinel;
_mm_mfence();
kSentinel = *reinterpret_cast<volatile int*>(dev_ptr - 1);
kSentinel = *reinterpret_cast<volatile address>(dev_ptr - 1);
}
}
}
projects/clr/rocclr/device/rocm/rocvirtual.cpp
+1 -1
Melihat File
@@ -3246,7 +3246,7 @@ bool VirtualGPU::submitKernelInternal(const amd::NDRangeContainer& sizes,
_mm_sfence();
*(argBuffer + argSize - 1) = *(parameters + argSize - 1);
_mm_mfence();
auto kSentinel = *reinterpret_cast<volatile address*>(argBuffer + argSize - 1);
auto kSentinel = *reinterpret_cast<volatile address>(argBuffer + argSize - 1);
}
}
}