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

finish aft_obj.cu

Browse Source
This commit is contained in:
amdsc21 2023-03-10 06:39:03 +01:00
parent 9bbbeb3f03
commit c073417d0c
3 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/objective/aft_obj.cc
View File

@ -16,6 +16,6 @@ DMLC_REGISTRY_FILE_TAG(aft_obj);
} // namespace obj
} // namespace xgboost
#ifndef XGBOOST_USE_CUDA
#if !defined(XGBOOST_USE_CUDA) && !defined(XGBOOST_USE_HIP)
#include "aft_obj.cu"
#endif // XGBOOST_USE_CUDA

4
src/objective/aft_obj.cu
View File

@ -28,9 +28,9 @@ using AFTLoss = xgboost::common::AFTLoss<Distribution>;
namespace xgboost {
namespace obj {
#if defined(XGBOOST_USE_CUDA)
#if defined(XGBOOST_USE_CUDA) || !defined(XGBOOST_USE_HIP)
DMLC_REGISTRY_FILE_TAG(aft_obj_gpu);
#endif // defined(XGBOOST_USE_CUDA)
#endif // defined(XGBOOST_USE_CUDA) || !defined(XGBOOST_USE_HIP)
class AFTObj : public ObjFunction {
public:

4
src/objective/aft_obj.hip
View File

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