From 2870b7ddd521c7b1cb093b9b9ae14ea6fdd800c3 Mon Sep 17 00:00:00 2001
From: pghafari <pghafari@amd.com>
Date: Fri, 20 May 2022 22:14:18 -0400
Subject: [PATCH] SWDEV-336550 - added nullptr check for svm

Change-Id: Icfd7dab1c7818958108385261542526d4545721d
---
 hipamd/src/hip_memory.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/hipamd/src/hip_memory.cpp b/hipamd/src/hip_memory.cpp
index a824e65914..50c44c5ce8 100644
--- a/hipamd/src/hip_memory.cpp
+++ b/hipamd/src/hip_memory.cpp
@@ -600,6 +600,13 @@ hipError_t hipMemPtrGetInfo(void *ptr, size_t *size) {
 hipError_t hipHostFree(void* ptr) {
   HIP_INIT_API(hipHostFree, ptr);
   CHECK_STREAM_CAPTURE_SUPPORTED();
+  size_t offset = 0;
+  amd::Memory* memory_object = getMemoryObject(ptr, offset);
+  if (memory_object != nullptr) {
+    if (memory_object->getSvmPtr() == nullptr) {
+      return hipErrorInvalidValue;
+    }
+  }
   HIP_RETURN(ihipFree(ptr));
 }