finished histogram.cu

2023-03-09 21:28:37 +01:00
parent f67e7de7ef
commit 0ed5d3c849
6 changed files with 69 additions and 2 deletions
--- a/src/data/ellpack_page.cuh
+++ b/src/data/ellpack_page.cuh
@@ -8,7 +8,13 @@
 #include <xgboost/data.h>

 #include "../common/compressed_iterator.h"
+
+#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"
 #include "../common/categorical.h"
 #include <thrust/binary_search.h>