diff --git a/projects/clr/rocclr/device/rocm/rocblit.cpp b/projects/clr/rocclr/device/rocm/rocblit.cpp index 7d12d50dc4..3bf6720f42 100644 --- a/projects/clr/rocclr/device/rocm/rocblit.cpp +++ b/projects/clr/rocclr/device/rocm/rocblit.cpp @@ -660,6 +660,8 @@ bool DmaBlitManager::hsaCopy(const Memory& srcMemory, const Memory& dstMemory, if (val != (kInitVal - 1)) { LogError("Async copy failed"); status = HSA_STATUS_ERROR; + } else { + gpu().hasPendingDispatch(); } } else { LogPrintfError("Hsa copy from host to device failed with code %d", status); @@ -747,6 +749,8 @@ bool DmaBlitManager::hsaCopyStaged(const_address hostSrc, address hostDst, size_ offset += size; } + gpu().hasPendingDispatch(); + return true; }