Switch back to the GPUIDX macro (#9438)

2023-08-04 00:14:31 -07:00
parent 1aabc690ec
commit bde1ebc209
21 changed files with 85 additions and 88 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(GetGPUId());
+  xgboost::Context ctx = MakeCUDACtx(GPUIDX);
  auto* obj = xgboost::ObjFunction::Create("mylogistic", &ctx);
  ASSERT_EQ(obj->DefaultEvalMetric(), std::string{"logloss"});
  delete obj;