From 167d628a49fa8da61871b98053bcea9402ae8447 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
---
 hipamd/api/hip/hip_memory.cpp | 3 +++
 1 file changed, 3 insertions(+)

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