finished gradient_based_sampler.cu

2023-03-09 22:26:08 +01:00 · 2023-03-09 22:26:08 +01:00 · 495816f694
commit 495816f694
parent df42dd2c53
2 changed files with 9 additions and 0 deletions
--- a/src/tree/gpu_hist/gradient_based_sampler.cuh
+++ b/src/tree/gpu_hist/gradient_based_sampler.cuh
@ -6,7 +6,12 @@
 #include <xgboost/data.h>
 #include <xgboost/span.h>

+#if defined(XGBOOST_USE_CUDA)
 #include "../../common/device_helpers.cuh"
+#elif defined(XGBOOST_USE_HIP)
+#include "../../common/device_helpers.hip.h"
+#endif
+
 #include "../../data/ellpack_page.cuh"

 namespace xgboost {
--- a/src/tree/gpu_hist/gradient_based_sampler.hip
+++ b/src/tree/gpu_hist/gradient_based_sampler.hip
@ -0,0 +1,4 @@
+
+#if defined(XGBOOST_USE_HIP)
+#include "gradient_based_sampler.cu"
+#endif