diff --git a/level_zero/core/source/cmdqueue/cmdqueue.cpp b/level_zero/core/source/cmdqueue/cmdqueue.cpp index 61e3abc805..7dafc94d43 100644 --- a/level_zero/core/source/cmdqueue/cmdqueue.cpp +++ b/level_zero/core/source/cmdqueue/cmdqueue.cpp @@ -165,13 +165,13 @@ ze_result_t CommandQueueImp::synchronize(uint64_t timeout) { } } -ze_result_t CommandQueueImp::synchronizeByPollingForTaskCount(uint64_t timeout) { +ze_result_t CommandQueueImp::synchronizeByPollingForTaskCount(uint64_t timeoutNanoseconds) { UNRECOVERABLE_IF(csr == nullptr); auto taskCountToWait = getTaskCount(); bool enableTimeout = true; - int64_t timeoutMicroseconds = static_cast(timeout); - if (timeout == std::numeric_limits::max()) { + int64_t timeoutMicroseconds = static_cast(timeoutNanoseconds / 1000); + if (timeoutNanoseconds == std::numeric_limits::max()) { enableTimeout = false; timeoutMicroseconds = NEO::TimeoutControls::maxTimeout; } diff --git a/level_zero/core/source/cmdqueue/cmdqueue_imp.h b/level_zero/core/source/cmdqueue/cmdqueue_imp.h index 3f6e1a78b9..d745ea0398 100644 --- a/level_zero/core/source/cmdqueue/cmdqueue_imp.h +++ b/level_zero/core/source/cmdqueue/cmdqueue_imp.h @@ -107,7 +107,7 @@ struct CommandQueueImp : public CommandQueue { MOCKABLE_VIRTUAL NEO::SubmissionStatus submitBatchBuffer(size_t offset, NEO::ResidencyContainer &residencyContainer, void *endingCmdPtr, bool isCooperative); - ze_result_t synchronizeByPollingForTaskCount(uint64_t timeout); + ze_result_t synchronizeByPollingForTaskCount(uint64_t timeoutNanoseconds); void postSyncOperations(bool hangDetected);