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

add hip rocthrust hipcub

Browse Source
This commit is contained in:
amdsc21 2023-03-07 06:35:00 +01:00
parent f13a7f8d91
commit f286ae5bfa
1 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
CMakeLists.txt
View File

@ -59,8 +59,6 @@ option(USE_HIP "Build with GPU acceleration" OFF)
option(USE_RCCL "Build with RCCL to enable distributed GPU support." OFF) option(USE_RCCL "Build with RCCL to enable distributed GPU support." OFF)
option(BUILD_WITH_SHARED_RCCL "Build with shared RCCL library." OFF) option(BUILD_WITH_SHARED_RCCL "Build with shared RCCL library." OFF)
option(BUILD_WITH_HIP_CUB "Build with cub in HIP installation" OFF) option(BUILD_WITH_HIP_CUB "Build with cub in HIP installation" OFF)
set(GPU_COMPUTE_TARGET "" CACHE STRING
"Semicolon separated list of compute versions to be built against, e.g. '908;90a'")
## Copied From dmlc ## Copied From dmlc
option(USE_HDFS "Build with HDFS support" OFF) option(USE_HDFS "Build with HDFS support" OFF)
option(USE_AZURE "Build with AZURE support" OFF) option(USE_AZURE "Build with AZURE support" OFF)
@ -194,16 +192,14 @@ if (USE_HIP)
message(STATUS "Configured HIP host compiler: ${CMAKE_HIP_HOST_COMPILER}") message(STATUS "Configured HIP host compiler: ${CMAKE_HIP_HOST_COMPILER}")
enable_language(HIP) enable_language(HIP)
if (${CMAKE_HIP_COMPILER_VERSION} VERSION_LESS 11.0) find_package(hip REQUIRED)
message(FATAL_ERROR "HIP version must be at least 11.0!") find_package(rocthrust REQUIRED)
endif() find_package(hipcub REQUIRED)
set(GEN_CODE "")
format_gencode_flags("${GPU_COMPUTE_VER}" GEN_CODE) set(CMAKE_HIP_FLAGS "-I${HIP_INCLUDE_DIRS} -I${HIP_INCLUDE_DIRS}/hip")
add_subdirectory(${PROJECT_SOURCE_DIR}/rocgputreeshap) add_subdirectory(${PROJECT_SOURCE_DIR}/rocgputreeshap)
if ((${CMAKE_HIP_COMPILER_VERSION} VERSION_GREATER_EQUAL 11.4) AND (NOT BUILD_WITH_HIP_CUB)) set(BUILD_WITH_HIP_CUB ON)
set(BUILD_WITH_HIP_CUB ON)
endif ()
endif (USE_HIP) endif (USE_HIP)
if (FORCE_COLORED_OUTPUT AND (CMAKE_GENERATOR STREQUAL "Ninja") AND if (FORCE_COLORED_OUTPUT AND (CMAKE_GENERATOR STREQUAL "Ninja") AND