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

finish regression_obj.cu

Browse Source
This commit is contained in:
amdsc21 2023-03-10 06:07:53 +01:00
parent ad710e4888
commit 968a1db4c0
2 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/objective/regression_obj.cu
View File

@ -38,7 +38,10 @@
#if defined(XGBOOST_USE_CUDA) #if defined(XGBOOST_USE_CUDA)
#include "../common/device_helpers.cuh" #include "../common/device_helpers.cuh"
#include "../common/linalg_op.cuh" #include "../common/linalg_op.cuh"
#endif // defined(XGBOOST_USE_CUDA) #elif defined(XGBOOST_USE_HIP)
#include "../common/device_helpers.hip.h"
#include "../common/linalg_op.cuh"
#endif // defined(XGBOOST_USE_CUDA), defined(XGBOOST_USE_HIP)
namespace xgboost { namespace xgboost {
namespace obj { namespace obj {
@ -49,9 +52,9 @@ void CheckRegInputs(MetaInfo const& info, HostDeviceVector<bst_float> const& pre
} }
} // anonymous namespace } // anonymous namespace
#if defined(XGBOOST_USE_CUDA) #if defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
DMLC_REGISTRY_FILE_TAG(regression_obj_gpu); DMLC_REGISTRY_FILE_TAG(regression_obj_gpu);
#endif // defined(XGBOOST_USE_CUDA) #endif // defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
struct RegLossParam : public XGBoostParameter<RegLossParam> { struct RegLossParam : public XGBoostParameter<RegLossParam> {
float scale_pos_weight; float scale_pos_weight;

4
src/objective/regression_obj.hip
View File

@ -0,0 +1,4 @@
#if defined(XGBOOST_USE_HIP)
#include "regression_obj.cu"
#endif