Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix memory type

Browse Source
This commit is contained in:
Hui Liu 2024-01-26 15:45:04 -08:00
parent 420f8d6fde
commit 1ce5029a96
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/data/array_interface.cu
View File

@ -73,7 +73,6 @@ bool ArrayInterfaceHandler::IsCudaPtr(void const* ptr) {
} else if (err == hipSuccess) { } else if (err == hipSuccess) {
#if HIP_VERSION_MAJOR < 6 #if HIP_VERSION_MAJOR < 6
switch (attr.memoryType) { switch (attr.memoryType) {
case hipMemoryTypeUnified:
case hipMemoryTypeHost: case hipMemoryTypeHost:
return false; return false;
default: default:
@ -81,7 +80,7 @@ bool ArrayInterfaceHandler::IsCudaPtr(void const* ptr) {
} }
#else #else
switch (attr.type) { switch (attr.type) {
case hipMemoryTypeUnified: case hipMemoryTypeUnregistered:
case hipMemoryTypeHost: case hipMemoryTypeHost:
return false; return false;
default: default: