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
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)
#include "../common/device_helpers.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 obj {
@@ -49,9 +52,9 @@ void CheckRegInputs(MetaInfo const& info, HostDeviceVector<bst_float> const& pre
}
} // anonymous namespace
#if defined(XGBOOST_USE_CUDA)
#if defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
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> {
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