finish iterative_dmatrix.cu

2023-03-10 03:47:00 +01:00
parent ec9f500a49
commit 49732359ef
4 changed files with 45 additions and 0 deletions
--- a/src/common/quantile.cuh
+++ b/src/common/quantile.cuh
@@ -5,7 +5,13 @@

 #include "xgboost/span.h"
 #include "xgboost/data.h"
+
+#if defined(XGBOOST_USE_CUDA)
 #include "device_helpers.cuh"
+#elif defined(XGBOOST_USE_HIP)
+#include "device_helpers.hip.h"
+#endif
+
 #include "quantile.h"
 #include "timer.h"
 #include "categorical.h"