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/jvm-packages/xgboost4j-gpu/src/test/java/ml/dmlc/xgboost4j/gpu/java/BoosterTest.java
+++ b/jvm-packages/xgboost4j-gpu/src/test/java/ml/dmlc/xgboost4j/gpu/java/BoosterTest.java
@@ -77,8 +77,8 @@ public class BoosterTest {
        put("objective", "binary:logistic");
        put("num_round", round);
        put("num_workers", 1);
-        put("tree_method", "gpu_hist");
-        put("predictor", "gpu_predictor");
+        put("tree_method", "hist");
+	put("device", "cuda");
        put("max_bin", maxBin);
      }
    };