From 68554e155bda9b70825f39e3862f95ab47812f03 Mon Sep 17 00:00:00 2001
From: Rahul Garg <rahul.garg@amd.com>
Date: Wed, 13 Jun 2018 23:10:05 +0530
Subject: [PATCH] Fix retrieved locked ptr offset

---
 hipamd/src/hip_memory.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/hipamd/src/hip_memory.cpp b/hipamd/src/hip_memory.cpp
index b96ca334df..68966abaf1 100644
--- a/hipamd/src/hip_memory.cpp
+++ b/hipamd/src/hip_memory.cpp
@@ -1589,7 +1589,7 @@ hipError_t ihipMemset(void* dst, int  value, size_t sizeBytes, hipStream_t strea
     return e;
 };
 
-hipError_t getLockedPointer(const void *hostPtr, size_t dataLen, void **devicePtrPtr)
+hipError_t getLockedPointer(void *hostPtr, size_t dataLen, void **devicePtrPtr)
 {
         hc::accelerator acc;
 
@@ -1600,7 +1600,8 @@ hipError_t getLockedPointer(const void *hostPtr, size_t dataLen, void **devicePt
 #endif
         am_status_t status = hc::am_memtracker_getinfo(&amPointerInfo, hostPtr);
         if (status == AM_SUCCESS) {
-            *devicePtrPtr =  (char*)amPointerInfo._devicePointer;
+            *devicePtrPtr = static_cast<char*>(amPointerInfo._devicePointer) +
+                (static_cast<char*>(hostPtr) - static_cast<char*>(amPointerInfo._hostPointer));
             return(hipSuccess);
         };
         return(hipErrorHostMemoryNotRegistered);