Define the new device parameter. (#9362)

2023-07-13 19:30:25 +08:00
parent 2d0cd2817e
commit 04aff3af8e
63 changed files with 827 additions and 477 deletions
--- a/tests/cpp/tree/test_prediction_cache.cc
+++ b/tests/cpp/tree/test_prediction_cache.cc
@@ -70,9 +70,9 @@ class TestPredictionCache : public ::testing::Test {
      Context ctx;
      ctx.InitAllowUnknown(Args{{"nthread", "8"}});
      if (updater_name == "grow_gpu_hist") {
-        ctx.gpu_id = 0;
+        ctx = ctx.MakeCUDA(0);
      } else {
-        ctx.gpu_id = Context::kCpuId;
+        ctx = ctx.MakeCPU();
      }

      ObjInfo task{ObjInfo::kRegression};