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

finish gpu_predictor.cu

Browse Source
This commit is contained in:
amdsc21
2023-03-10 01:29:54 +01:00
parent 1c58ff61d1
commit f0febfbcac
7 changed files with 88 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cmake/Utils.cmake
View File

@@ -194,9 +194,11 @@ function(xgboost_set_hip_flags target)
if (NOT BUILD_WITH_HIP_CUB)
target_compile_definitions(${target} PRIVATE -DXGBOOST_USE_HIP=1 -DTHRUST_IGNORE_CUB_VERSION_CHECK=1)
target_include_directories(${target} PRIVATE ${xgboost_SOURCE_DIR}/rocgputreeshap)
target_include_directories(${target} PRIVATE ${xgboost_SOURCE_DIR}/warp-primitives/include)
else ()
target_compile_definitions(${target} PRIVATE -DXGBOOST_USE_HIP=1)
target_include_directories(${target} PRIVATE ${xgboost_SOURCE_DIR}/rocgputreeshap)
target_include_directories(${target} PRIVATE ${xgboost_SOURCE_DIR}/warp-primitives/include)
endif (NOT BUILD_WITH_HIP_CUB)
set_target_properties(${target} PROPERTIES