Clean up MGPU C++ tests (#9430)

2023-08-01 23:31:18 -07:00
parent a9da2e244a
commit c2b85ab68a
28 changed files with 200 additions and 194 deletions
--- a/tests/cpp/plugin/test_example_objective.cc
+++ b/tests/cpp/plugin/test_example_objective.cc
@@ -5,7 +5,7 @@

 namespace xgboost {
 TEST(Plugin, ExampleObjective) {
-  xgboost::Context ctx = MakeCUDACtx(GPUIDX);
+  xgboost::Context ctx = MakeCUDACtx(GetGPUId());
  auto* obj = xgboost::ObjFunction::Create("mylogistic", &ctx);
  ASSERT_EQ(obj->DefaultEvalMetric(), std::string{"logloss"});
  delete obj;