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


[ROCm/hip commit: 3564a8ddf9]

projects/hip/api/hip/hip_memory.cpp

@@ -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;
     }