diff --git a/src/linear/linear_updater.cc b/src/linear/linear_updater.cc index e66206196..2aeaeb36c 100644 --- a/src/linear/linear_updater.cc +++ b/src/linear/linear_updater.cc @@ -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 diff --git a/src/objective/objective.cc b/src/objective/objective.cc index d3b01d80b..70746a1f3 100644 --- a/src/objective/objective.cc +++ b/src/objective/objective.cc @@ -42,7 +42,7 @@ void ObjFunction::InitEstimation(MetaInfo const&, linalg::Tensor* base namespace xgboost { namespace obj { // List of files that will be force linked in static links. -#ifdef XGBOOST_USE_CUDA +#if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP) DMLC_REGISTRY_LINK_TAG(regression_obj_gpu); DMLC_REGISTRY_LINK_TAG(quantile_obj_gpu); DMLC_REGISTRY_LINK_TAG(hinge_obj_gpu); @@ -54,6 +54,6 @@ DMLC_REGISTRY_LINK_TAG(quantile_obj); DMLC_REGISTRY_LINK_TAG(hinge_obj); DMLC_REGISTRY_LINK_TAG(multiclass_obj); DMLC_REGISTRY_LINK_TAG(rank_obj); -#endif // XGBOOST_USE_CUDA +#endif // XGBOOST_USE_CUDA, XGBOOST_USE_HIP } // namespace obj } // namespace xgboost diff --git a/src/tree/tree_updater.cc b/src/tree/tree_updater.cc index 286daa4d8..9a3a757a7 100644 --- a/src/tree/tree_updater.cc +++ b/src/tree/tree_updater.cc @@ -34,8 +34,8 @@ DMLC_REGISTRY_LINK_TAG(updater_prune); DMLC_REGISTRY_LINK_TAG(updater_quantile_hist); DMLC_REGISTRY_LINK_TAG(updater_approx); DMLC_REGISTRY_LINK_TAG(updater_sync); -#ifdef XGBOOST_USE_CUDA +#if defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP) DMLC_REGISTRY_LINK_TAG(updater_gpu_hist); -#endif // XGBOOST_USE_CUDA +#endif // XGBOOST_USE_CUDA, XGBOOST_USE_HIP } // namespace tree } // namespace xgboost