From 34d02d8422641f445aea4cde63801b5249a6a7d6 Mon Sep 17 00:00:00 2001
From: foreman <dl.constructicon@amd.com>
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


[ROCm/hip commit: 3564a8ddf97e9c15611f4fcd11db181e42ae79e2]
---
 projects/hip/api/hip/hip_memory.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/projects/hip/api/hip/hip_memory.cpp b/projects/hip/api/hip/hip_memory.cpp
index 424fbec633..98c4c50e03 100644
--- a/projects/hip/api/hip/hip_memory.cpp
+++ b/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;
     }