finish constraints.cu

2023-03-09 22:43:51 +01:00 · 2023-03-09 22:43:51 +01:00 · 1530c03f7d
commit 1530c03f7d
parent 309268de02
2 changed files with 9 additions and 0 deletions
--- a/src/tree/constraints.cu
+++ b/src/tree/constraints.cu
@ -14,7 +14,12 @@
 #include "xgboost/span.h"
 #include "constraints.cuh"
 #include "param.h"
+
+#if defined(XGBOOST_USE_hip.CUDA)
 #include "../common/device_helpers.cuh"
+#elif defined(XGBOOST_USE_HIP)
+#include "../common/device_helpers.hip.h"
+#endif

 namespace xgboost {

--- a/src/tree/constraints.hip
+++ b/src/tree/constraints.hip
@ -0,0 +1,4 @@
+
+#if defined(XGBOOST_USE_HIP)
+#include "constraints.cu"
+#endif