Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix objective/objective.cc, CMakeFile and setup.py

Browse Source
This commit is contained in:
amdsc21
2023-03-23 20:22:34 +01:00
parent 595cd81251
commit e0716afabf
5 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/objective/objective.cc
View File

@@ -42,7 +42,7 @@ void ObjFunction::InitEstimation(MetaInfo const&, linalg::Tensor<float, 1>* base
namespace xgboost {
namespace obj {
// List of files that will be force linked in static links.
#if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP)
#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);