From 7593b415758317530236b258129a8c173bad264a Mon Sep 17 00:00:00 2001
From: Oak Zeng <Oak.Zeng@amd.com>
Date: Mon, 28 Oct 2019 16:16:35 -0500
Subject: [PATCH] Unmap memory from GPU before free

Change-Id: Ic33b17cbaee5de7908d37527254f4f146e6b71e3
Signed-off-by: Oak Zeng <Oak.Zeng@amd.com>
---
 tests/kfdtest/src/KFDMemoryTest.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/kfdtest/src/KFDMemoryTest.cpp b/tests/kfdtest/src/KFDMemoryTest.cpp
index 33ef74862c..747b1daa5a 100644
--- a/tests/kfdtest/src/KFDMemoryTest.cpp
+++ b/tests/kfdtest/src/KFDMemoryTest.cpp
@@ -2165,6 +2165,7 @@ TEST_F(KFDMemoryTest, CacheInvalidateOnCPUWrite) {
     EXPECT_EQ(buffer[100], 0x5678);
 
     // Clean up
+    EXPECT_SUCCESS(hsaKmtUnmapMemoryToGPU(buffer));
     EXPECT_SUCCESS(hsaKmtFreeMemory(buffer, PAGE_SIZE));
     EXPECT_SUCCESS(queue.Destroy());