Add missing pollForCompletion on clEnqueueVerifyMemoryINTEL

Related-To: NEO-7618

Signed-off-by: Dunajski, Bartosz <bartosz.dunajski@intel.com>
This commit is contained in:
Dunajski, Bartosz
2023-03-02 10:49:50 +00:00
committed by Compute-Runtime-Automation
parent 7903617a44
commit 72ada8b41f
5 changed files with 53 additions and 1 deletions

View File

@@ -5760,6 +5760,13 @@ CL_API_ENTRY cl_int CL_API_CALL clEnqueueVerifyMemoryINTEL(cl_command_queue comm
return retVal;
}
auto activeCopyEngines = pCommandQueue->peekActiveBcsStates();
for (auto &copyEngine : activeCopyEngines) {
if (copyEngine.isValid()) {
pCommandQueue->getBcsCommandStreamReceiver(copyEngine.engineType)->pollForCompletion();
}
}
auto &csr = pCommandQueue->getGpgpuCommandStreamReceiver();
auto status = csr.expectMemory(allocationPtr, expectedData, sizeOfComparison, comparisonMode);
return status ? CL_SUCCESS : CL_INVALID_VALUE;