fix macro XGBOOST_USE_HIP

This commit is contained in:
amdsc21
2023-03-10 07:01:25 +01:00
parent 5edfc1e2e9
commit bde3107c3e
3 changed files with 6 additions and 6 deletions

View File

@@ -30,8 +30,8 @@ DMLC_REGISTER_PARAMETER(LinearTrainParam);
// List of files that will be force linked in static links.
DMLC_REGISTRY_LINK_TAG(updater_shotgun);
DMLC_REGISTRY_LINK_TAG(updater_coordinate);
#ifdef XGBOOST_USE_CUDA
#if defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
DMLC_REGISTRY_LINK_TAG(updater_gpu_coordinate);
#endif // XGBOOST_USE_CUDA
#endif // XGBOOST_USE_CUDA, XGBOOST_USE_HIP
} // namespace linear
} // namespace xgboost