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

finish multiclass_obj.cu

Browse Source
This commit is contained in:
amdsc21 2023-03-10 06:35:06 +01:00
parent 41407850d5
commit 58a9fe07b6
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/multiclass_obj.cc
View File

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

4
src/objective/multiclass_obj.cu
View File

@ -24,9 +24,9 @@
namespace xgboost { namespace xgboost {
namespace obj { namespace obj {
#if defined(XGBOOST_USE_CUDA) #if defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
DMLC_REGISTRY_FILE_TAG(multiclass_obj_gpu); DMLC_REGISTRY_FILE_TAG(multiclass_obj_gpu);
#endif // defined(XGBOOST_USE_CUDA) #endif // defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
struct SoftmaxMultiClassParam : public XGBoostParameter<SoftmaxMultiClassParam> { struct SoftmaxMultiClassParam : public XGBoostParameter<SoftmaxMultiClassParam> {
int num_class; int num_class;

4
src/objective/multiclass_obj.hip
View File

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