fix memory type

2024-01-26 15:44:24 -08:00 · 2024-01-26 15:44:24 -08:00 · 230dc11603
commit 230dc11603
parent dc7ee041cc
1 changed files with 1 additions and 2 deletions
--- a/src/data/array_interface.cu
+++ b/src/data/array_interface.cu
@ -73,7 +73,6 @@ 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:
@ -81,7 +80,7 @@ bool ArrayInterfaceHandler::IsCudaPtr(void const* ptr) {
    }
 #else
    switch (attr.type) {
-      case hipMemoryTypeUnified:
+      case hipMemoryTypeUnregistered:
      case hipMemoryTypeHost:
        return false;
      default: