[jvm-packages] Add the new device parameter. (#9385)

2023-07-17 18:40:39 +08:00
parent 2caceb157d
commit f4fb2be101
15 changed files with 112 additions and 47 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,7 +77,8 @@ public class BoosterTest {
        put("objective", "binary:logistic");
        put("num_round", round);
        put("num_workers", 1);
-        put("tree_method", "gpu_hist");
+        put("tree_method", "hist");
+	put("device", "cuda");
        put("max_bin", maxBin);
      }
    };