finished feature_groups.cu

2023-03-09 21:31:00 +01:00
parent 0ed5d3c849
commit 1e09c21456
2 changed files with 9 additions and 0 deletions
--- a/src/tree/gpu_hist/feature_groups.cu
+++ b/src/tree/gpu_hist/feature_groups.cu
@@ -8,7 +8,12 @@

 #include "feature_groups.cuh"

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

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