Merge pull request #518 from gargrahul/fix_pr_484_devptroffset

Fix retrieved locked ptr offset
Этот коммит содержится в:
Maneesh Gupta
2018-06-14 07:37:34 +05:30
коммит произвёл GitHub
родитель 359e6609bc 68554e155b
Коммит f5fccba009
+3 -2
Просмотреть файл
@@ -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);