try

razdoburdin · Aug 13, 2024 · ce73ce3 · ce73ce3
1 parent c25ed19
commit ce73ce3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python-sycl/test_sycl_training_continuation.py b/tests/python-sycl/test_sycl_training_continuation.py
@@ -13,7 +13,7 @@ def run_training_continuation(self, use_json):
         y = np.random.randn(kRows)
         dtrain = xgb.DMatrix(X, y)
         params = {
-            "device": "sycl",
+            "device": "sycl:gpu",
             "max_depth": "2",
             "gamma": "0.1",
             "alpha": "0.01",