diff --git a/cmake/Utils.cmake b/cmake/Utils.cmake index 582e60e12..582b740c3 100644 --- a/cmake/Utils.cmake +++ b/cmake/Utils.cmake @@ -212,12 +212,12 @@ endmacro(xgboost_link_nccl) macro(xgboost_link_rccl target) if(BUILD_STATIC_LIB) - target_include_directories(${target} PUBLIC ${RCCL_INCLUDE_DIR}) + target_include_directories(${target} PUBLIC ${RCCL_INCLUDE_DIR}/rccl) target_compile_definitions(${target} PUBLIC -DXGBOOST_USE_RCCL=1) target_link_directories(${target} PUBLIC ${HIP_LIB_INSTALL_DIR}) target_link_libraries(${target} PUBLIC ${RCCL_LIBRARY}) else() - target_include_directories(${target} PRIVATE ${RCCL_INCLUDE_DIR}) + target_include_directories(${target} PRIVATE ${RCCL_INCLUDE_DIR}/rccl) target_compile_definitions(${target} PRIVATE -DXGBOOST_USE_RCCL=1) target_link_directories(${target} PUBLIC ${HIP_LIB_INSTALL_DIR}) target_link_libraries(${target} PRIVATE ${RCCL_LIBRARY}) diff --git a/rocgputreeshap b/rocgputreeshap index 6ceffde02..2fea6734e 160000 --- a/rocgputreeshap +++ b/rocgputreeshap @@ -1 +1 @@ -Subproject commit 6ceffde024f8752954550ebcca98caa24b5d158d +Subproject commit 2fea6734e83cf147c1bbe580ac4713cd50abcad5 diff --git a/src/data/array_interface.cu b/src/data/array_interface.cu index b27bd1c08..f6644c1fd 100644 --- a/src/data/array_interface.cu +++ b/src/data/array_interface.cu @@ -71,6 +71,7 @@ bool ArrayInterfaceHandler::IsCudaPtr(void const* ptr) { if (err == hipErrorInvalidValue) { return false; } else if (err == hipSuccess) { +#if HIP_VERSION_MAJOR < 6 switch (attr.memoryType) { case hipMemoryTypeUnified: case hipMemoryTypeHost: @@ -78,6 +79,15 @@ bool ArrayInterfaceHandler::IsCudaPtr(void const* ptr) { default: return true; } +#else + switch (attr.type) { + case hipMemoryTypeUnified: + case hipMemoryTypeHost: + return false; + default: + return true; + } +#endif return true; } else { return false;