From 7aba1cf4cd3f4e9ac3d1551e7fdb5dea8ea6e464 Mon Sep 17 00:00:00 2001
From: Sourabh Betigeri <sourabh.betigeri@amd.com>
Date: Thu, 13 Apr 2023 22:59:59 +0000
Subject: [PATCH] SWDEV-392367 - Fixes hipMemGetAddressRange returning device
 buffer size as 0

Change-Id: I6b5640b544390b2dfc53fad271acc7f30240e274
---
 hipamd/src/hip_memory.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/hipamd/src/hip_memory.cpp b/hipamd/src/hip_memory.cpp
index 262ad421ef..602a544b97 100644
--- a/hipamd/src/hip_memory.cpp
+++ b/hipamd/src/hip_memory.cpp
@@ -730,7 +730,8 @@ hipError_t hipMemGetAddressRange(hipDeviceptr_t* pbase, size_t* psize, hipDevice
 
   // Since we are using SVM buffer DevicePtr and HostPtr is the same
   void* ptr = dptr;
-  amd::Memory* svmMem = getMemoryObjectWithOffset(ptr);
+  size_t offset = 0;
+  amd::Memory* svmMem = getMemoryObject(ptr, offset);
   if (svmMem == nullptr) {
     HIP_RETURN(hipErrorNotFound);
   }