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

finish hinge.cu

Browse Source
This commit is contained in:
amdsc21
2023-03-10 06:04:59 +01:00
parent 4e3c699814
commit ad710e4888
3 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/objective/hinge.cu
View File

@@ -16,9 +16,9 @@
namespace xgboost {
namespace obj {
#if defined(XGBOOST_USE_CUDA)
#if defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
DMLC_REGISTRY_FILE_TAG(hinge_obj_gpu);
#endif // defined(XGBOOST_USE_CUDA)
#endif // defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
class HingeObj : public ObjFunction {
public: