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

finish constraints.cu

Browse Source
This commit is contained in:
amdsc21 2023-03-09 22:43:51 +01:00
parent 309268de02
commit 1530c03f7d
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/tree/constraints.cu
View File

@ -14,7 +14,12 @@
#include "xgboost/span.h" #include "xgboost/span.h"
#include "constraints.cuh" #include "constraints.cuh"
#include "param.h" #include "param.h"
#if defined(XGBOOST_USE_hip.CUDA)
#include "../common/device_helpers.cuh" #include "../common/device_helpers.cuh"
#elif defined(XGBOOST_USE_HIP)
#include "../common/device_helpers.hip.h"
#endif
namespace xgboost { namespace xgboost {

4
src/tree/constraints.hip
View File

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