rocm enable for v2.0.1

2023-10-27 18:50:28 -07:00
parent 2e7e9d3b2d a408254c2f
commit 782b73f2bb
447 changed files with 13518 additions and 8719 deletions
--- a/src/common/stats.cu
+++ b/src/common/stats.cu
@@ -19,10 +19,6 @@ namespace xgboost {
 namespace common {
 namespace cuda_impl {

-#if defined(XGBOOST_USE_HIP)
-namespace cub = hipcub;
-#endif
-
 void Median(Context const* ctx, linalg::TensorView<float const, 2> t,
            common::OptionalWeights weights, linalg::Tensor<float, 1>* out) {
  CHECK_GE(t.Shape(1), 1);