intel/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2026-01-01 12:33:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: ensure thread-safety in zeDeviceSynchronize

Browse Source 
get task count and flush stamp within single critical section

Related-To: NEO-14560
Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>
This commit is contained in:
Mateusz Jablonski
2025-07-30 14:09:01 +00:00
committed by Compute-Runtime-Automation
parent b6200738f3
commit abb00a5ce3
3 changed files with 107 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
level_zero/core/source/device/device_imp.cpp
View File

@@ -2249,28 +2249,35 @@ uint32_t DeviceImp::getEventMaxKernelCount() const {
}
ze_result_t DeviceImp::synchronize() {
for (auto &engine : neoDevice->getAllEngines()) {
if (engine.commandStreamReceiver->isInitialized()) {
auto waitStatus = engine.commandStreamReceiver->waitForTaskCountWithKmdNotifyFallback(
engine.commandStreamReceiver->peekTaskCount(),
engine.commandStreamReceiver->obtainCurrentFlushStamp(),
auto waitForCsr = [](NEO::CommandStreamReceiver *csr) -> ze_result_t {
if (csr->isInitialized()) {
auto lock = csr->obtainUniqueOwnership();
auto taskCountToWait = csr->peekTaskCount();
auto flushStampToWait = csr->obtainCurrentFlushStamp();
lock.unlock();
auto waitStatus = csr->waitForTaskCountWithKmdNotifyFallback(
taskCountToWait,
flushStampToWait,
false,
NEO::QueueThrottle::MEDIUM);
if (waitStatus == NEO::WaitStatus::gpuHang) {
return ZE_RESULT_ERROR_DEVICE_LOST;
}
}
return ZE_RESULT_SUCCESS;
};
for (auto &engine : neoDevice->getAllEngines()) {
auto ret = waitForCsr(engine.commandStreamReceiver);
if (ret != ZE_RESULT_SUCCESS) {
return ret;
}
}
for (auto &secondaryCsr : neoDevice->getSecondaryCsrs()) {
if (secondaryCsr->isInitialized()) {
auto waitStatus = secondaryCsr->waitForTaskCountWithKmdNotifyFallback(
secondaryCsr->peekTaskCount(),
secondaryCsr->obtainCurrentFlushStamp(),
false,
NEO::QueueThrottle::MEDIUM);
if (waitStatus == NEO::WaitStatus::gpuHang) {
return ZE_RESULT_ERROR_DEVICE_LOST;
}
auto ret = waitForCsr(secondaryCsr.get());
if (ret != ZE_RESULT_SUCCESS) {
return ret;
}
}