Media
/
compute-runtime
mirror of https://github.com/intel/compute-runtime.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test: ensure thread safety when capturing input params for wait function 

Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>
This commit is contained in:
Mateusz Jablonski 2025-05-08 08:46:47 +00:00 committed by Compute-Runtime-Automation
parent f6796c7384
commit adc40934ae
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
level_zero/core/test/unit_tests/sources/device/test_l0_device.cpp
View File

@ -6805,6 +6805,7 @@ HWTEST_F(DeviceSimpleTests, givenNoGpuHangWhenSynchronizingDeviceThenCallWaitFor
csr->latestFlushedTaskCount = 0u; csr->latestFlushedTaskCount = 0u;
csr->taskCount = taskCountToWait++; csr->taskCount = taskCountToWait++;
csr->flushStamp->setStamp(flushStampToWait++); csr->flushStamp->setStamp(flushStampToWait++);
csr->captureWaitForTaskCountWithKmdNotifyInputParams = true;
csr->waitForTaskCountWithKmdNotifyFallbackReturnValue = WaitStatus::ready; csr->waitForTaskCountWithKmdNotifyFallbackReturnValue = WaitStatus::ready;
} }
@ -6815,6 +6816,7 @@ HWTEST_F(DeviceSimpleTests, givenNoGpuHangWhenSynchronizingDeviceThenCallWaitFor
csr->latestFlushedTaskCount = 0u; csr->latestFlushedTaskCount = 0u;
csr->taskCount = taskCountToWait++; csr->taskCount = taskCountToWait++;
csr->flushStamp->setStamp(flushStampToWait++); csr->flushStamp->setStamp(flushStampToWait++);
csr->captureWaitForTaskCountWithKmdNotifyInputParams = true;
csr->waitForTaskCountWithKmdNotifyFallbackReturnValue = WaitStatus::ready; csr->waitForTaskCountWithKmdNotifyFallbackReturnValue = WaitStatus::ready;
} }

5
shared/test/common/libult/ult_command_stream_receiver.h
View File

@ -330,7 +330,11 @@ class UltCommandStreamReceiver : public CommandStreamReceiverHw<GfxFamily> {
} }
WaitStatus waitForTaskCountWithKmdNotifyFallback(TaskCountType taskCountToWait, FlushStamp flushStampToWait, bool useQuickKmdSleep, QueueThrottle throttle) override { WaitStatus waitForTaskCountWithKmdNotifyFallback(TaskCountType taskCountToWait, FlushStamp flushStampToWait, bool useQuickKmdSleep, QueueThrottle throttle) override {
if (captureWaitForTaskCountWithKmdNotifyInputParams) {
static std::mutex waitForTaskCountWithKmdNotifyInputParamsMtx;
std::unique_lock<std::mutex> lock(waitForTaskCountWithKmdNotifyInputParamsMtx);
waitForTaskCountWithKmdNotifyInputParams.push_back({taskCountToWait, flushStampToWait, useQuickKmdSleep, throttle}); waitForTaskCountWithKmdNotifyInputParams.push_back({taskCountToWait, flushStampToWait, useQuickKmdSleep, throttle});
}
if (waitForTaskCountWithKmdNotifyFallbackReturnValue.has_value()) { if (waitForTaskCountWithKmdNotifyFallbackReturnValue.has_value()) {
return *waitForTaskCountWithKmdNotifyFallbackReturnValue; return *waitForTaskCountWithKmdNotifyFallbackReturnValue;
} }
@ -667,6 +671,7 @@ class UltCommandStreamReceiver : public CommandStreamReceiverHw<GfxFamily> {
bool isUserFenceWaitSupported = false; bool isUserFenceWaitSupported = false;
bool isAnyDirectSubmissionEnabledCallBase = true; bool isAnyDirectSubmissionEnabledCallBase = true;
bool isAnyDirectSubmissionEnabledResult = true; bool isAnyDirectSubmissionEnabledResult = true;
std::atomic_bool captureWaitForTaskCountWithKmdNotifyInputParams = false;
}; };
} // namespace NEO } // namespace NEO