fix memory type

ROCm · Jan 26, 2024 · 1ce5029 · 1ce5029
1 parent 420f8d6
commit 1ce5029
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/data/array_interface.cu b/src/data/array_interface.cu
@@ -73,15 +73,14 @@ bool ArrayInterfaceHandler::IsCudaPtr(void const* ptr) {
   } else if (err == hipSuccess) {
 #if HIP_VERSION_MAJOR < 6
     switch (attr.memoryType) {
-      case hipMemoryTypeUnified:
       case hipMemoryTypeHost:
         return false;
       default:
         return true;
     }
 #else
     switch (attr.type) {
-      case hipMemoryTypeUnified:
+      case hipMemoryTypeUnregistered:
       case hipMemoryTypeHost:
         return false;
       default: