P4 to Git Change 1574251 by skudchad@skudchad_rocm on 2018/06/29 01:49:57

SWDEV-145570 - [HIP] Fix hipHostUnregister. Affected files ... ... //depot/stg/opencl/drivers/opencl/api/hip/hip_memory.cpp#35 edit
2018-06-29 01:59:22 -04:00
parent 3f0b25f6bf
commit 3564a8ddf9
1 changed files with 3 additions and 0 deletions
@@ -540,6 +540,9 @@ hipError_t hipHostUnregister(void* hostPtr) {
    amd::Memory* mem = getMemoryObject(hostPtr, offset);

    if(mem) {
+      hip::syncStreams();
+      hip::getNullStream()->finish();
+      amd::MemObjMap::RemoveMemObj(hostPtr);
      mem->release();
      return hipSuccess;
    }