Move away from Memory management APIs deprecated in ROCM 6.3

ROCm · Oct 15, 2024 · c86d8a0 · c86d8a0
1 parent 65ffb0e
commit c86d8a0
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/onnxruntime/core/providers/migraphx/migraphx_allocator.cc b/onnxruntime/core/providers/migraphx/migraphx_allocator.cc
@@ -71,13 +71,17 @@ void* MIGraphXExternalAllocator::Reserve(size_t size) {
 void* HIPPinnedAllocator::Alloc(size_t size) {
   void* p = nullptr;
   if (size > 0) {
+#ifdef hipExtHostAllocCoherent
+    HIP_CALL_THROW(hipExtHostAlloc((void**)&p, size));
+#else
     HIP_CALL_THROW(hipHostMalloc((void**)&p, size));
+#endif
   }
   return p;
 }
 
 void HIPPinnedAllocator::Free(void* p) {
-  HIP_CALL_THROW(hipHostFree(p));
+  HIP_CALL_THROW(hipFreeHost(p));
 }
 
 }  // namespace onnxruntime
diff --git a/onnxruntime/core/providers/rocm/rocm_allocator.cc b/onnxruntime/core/providers/rocm/rocm_allocator.cc
@@ -89,13 +89,17 @@ void* ROCMExternalAllocator::Reserve(size_t size) {
 void* ROCMPinnedAllocator::Alloc(size_t size) {
   void* p = nullptr;
   if (size > 0) {
+#ifdef hipExtHostAllocCoherent
+    HIP_CALL_THROW(hipExtHostAlloc((void**)&p, size));
+#else
     HIP_CALL_THROW(hipHostMalloc((void**)&p, size));
+#endif
   }
   return p;
 }
 
 void ROCMPinnedAllocator::Free(void* p) {
-  HIP_CALL_THROW(hipHostFree(p));
+  HIP_CALL_THROW(hipFreeHost(p));
 }
 
 }  // namespace onnxruntime