[fed] Fixes for the encrypted GRPC backend. (#10503)

This commit is contained in:
Jiaming Yuan
2024-07-02 15:15:12 +08:00
committed by GitHub
parent 5f0c1e902b
commit a39fef2c67
9 changed files with 192 additions and 109 deletions

View File

@@ -191,8 +191,11 @@ DeviceOrd CUDAOrdinal(DeviceOrd device, bool) {
}
if (device.IsCUDA()) {
device = CUDAOrdinal(device, fail_on_invalid_gpu_id);
if (!device.IsCUDA()) {
// We allow loading a GPU-based pickle on a CPU-only machine.
LOG(WARNING) << "XGBoost is not compiled with CUDA support.";
}
}
return device;
}
} // namespace