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

finished gradient_based_sampler.cu

Browse Source
This commit is contained in:
amdsc21 2023-03-09 22:26:08 +01:00
parent df42dd2c53
commit 495816f694
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/tree/gpu_hist/gradient_based_sampler.cuh
View File

@ -6,7 +6,12 @@
#include <xgboost/data.h> #include <xgboost/data.h>
#include <xgboost/span.h> #include <xgboost/span.h>
#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 "../../data/ellpack_page.cuh" #include "../../data/ellpack_page.cuh"
namespace xgboost { namespace xgboost {

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

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