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

finished feature_groups.cu

Browse Source
This commit is contained in:
amdsc21
2023-03-09 21:31:00 +01:00
parent 0ed5d3c849
commit 1e09c21456
2 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/tree/gpu_hist/feature_groups.cu
View File

@@ -8,7 +8,12 @@
#include "feature_groups.cuh" #include "feature_groups.cuh"
#if defined(XGBOOST_USE_CUDA)
#include "../../common/device_helpers.cuh" #include "../../common/device_helpers.cuh"
#elif defined(XGBOOST_USE_HIP)
#include "../../common/device_helpers.hip.h"
#endif
#include "../../common/hist_util.h" #include "../../common/hist_util.h"
namespace xgboost { namespace xgboost {

4
src/tree/gpu_hist/feature_groups.hip
View File

@@ -0,0 +1,4 @@
#if defined(XGBOOST_USE_HIP)
#include "feature_groups.cu"
#endif