[jvm-packages] support stage-level scheduling (#9775)

2023-11-14 08:59:45 +08:00
parent 162da7b52b
commit 36a552ac98
4 changed files with 298 additions and 13 deletions
--- a/jvm-packages/xgboost4j-spark-gpu/src/test/scala/ml/dmlc/xgboost4j/scala/rapids/spark/GpuXGBoostGeneralSuite.scala
+++ b/jvm-packages/xgboost4j-spark-gpu/src/test/scala/ml/dmlc/xgboost4j/scala/rapids/spark/GpuXGBoostGeneralSuite.scala
@@ -206,7 +206,7 @@ class GpuXGBoostGeneralSuite extends GpuTestSuite {
        .setDevice("cuda:1")
        .fit(trainingDf)
      }
-      assert(thrown.getMessage.contains("`cuda` or `gpu`"))
+      assert(thrown.getMessage.contains("device given invalid value cuda:1"))
    }
  }
 }