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

fix hip.cc

Browse Source
This commit is contained in:
amdsc21 2023-03-07 05:11:42 +01:00
parent 75fa15b36d
commit 30de728631
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
CMakeLists.txt
View File

@ -199,7 +199,7 @@ if (USE_HIP)
endif()
set(GEN_CODE "")
format_gencode_flags("${GPU_COMPUTE_VER}" GEN_CODE)
add_subdirectory(${PROJECT_SOURCE_DIR}/gputreeshap)
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)

2
src/CMakeLists.txt
View File

@ -17,7 +17,7 @@ if (USE_CUDA)
endif (USE_CUDA)
if (USE_HIP)
file(GLOB_RECURSE HIP_SOURCES *.cu *.cuh)
file(GLOB_RECURSE HIP_SOURCES *.hip.cc *.hip.h)
target_sources(objxgboost PRIVATE ${HIP_SOURCES})
endif (USE_HIP)