From 1ce5029a962eb1bb5b08c62a8d254a9bb9373c37 Mon Sep 17 00:00:00 2001
From: Hui Liu <96135754+hliuca@users.noreply.github.com>
Date: Fri, 26 Jan 2024 15:45:04 -0800
Subject: [PATCH] fix memory type

---
 src/data/array_interface.cu | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/data/array_interface.cu b/src/data/array_interface.cu
index f6644c1fd8b7..2ce80b91abf5 100644
--- 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: