AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-290384 - Enable active wait on CPU if HIP requested

Browse Source 
Change-Id: Idea5adf7a4705cb999da6785e6229fe3200dce17
This commit is contained in:
German Andryeyev
2021-07-19 15:05:24 -04:00
committed by Maneesh Gupta
parent 4589343c3e
commit 2babcfbdbb
3 changed files with 21 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rocclr/device/rocm/rocdevice.cpp
+2 -1
View File
@@ -2546,7 +2546,8 @@ bool Device::IsHwEventReady(const amd::Event& event, bool wait) const {
ClPrint(amd::LOG_INFO, amd::LOG_SIG, "No HW event");
return false;
} else if (wait) {
WaitForSignal(reinterpret_cast<ProfilingSignal*>(hw_event)->signal_);
auto* vdev = event.command().queue()->vdev();
WaitForSignal(reinterpret_cast<ProfilingSignal*>(hw_event)->signal_, vdev->ActiveWait());
return true;
}
return (hsa_signal_load_relaxed(reinterpret_cast<ProfilingSignal*>(hw_event)->signal_) <= 0);