From 3564a8ddf97e9c15611f4fcd11db181e42ae79e2 Mon Sep 17 00:00:00 2001
From: foreman
Date: Fri, 29 Jun 2018 01:59:22 -0400
Subject: [PATCH] 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
---
api/hip/hip_memory.cpp | 3 +++
1 file changed, 3 insertions(+)
diff --git a/api/hip/hip_memory.cpp b/api/hip/hip_memory.cpp
index 424fbec633..98c4c50e03 100644
--- a/api/hip/hip_memory.cpp
+++ b/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;
}