diff --git a/level_zero/core/source/fence.cpp b/level_zero/core/source/fence.cpp index 495e66f47f..dd92192169 100644 --- a/level_zero/core/source/fence.cpp +++ b/level_zero/core/source/fence.cpp @@ -11,6 +11,8 @@ #include "shared/source/memory_manager/memory_manager.h" #include "shared/source/utilities/cpuintrinsics.h" +#include "hw_helpers.h" + namespace L0 { struct FenceImp : public Fence { @@ -89,6 +91,8 @@ ze_result_t FenceImp::hostSynchronize(uint32_t timeout) { return ZE_RESULT_SUCCESS; } + waitForTaskCountWithKmdNotifyFallbackHelper(cmdQueue->getCsr(), cmdQueue->getTaskCount(), 0, false, false); + if (timeout == 0) { return queryStatus(); }