finish gbtree.cu porting

2023-03-08 21:09:56 +01:00
parent cdd7794641
commit 7e1b06417b
4 changed files with 21 additions and 5 deletions
--- a/src/gbm/gbtree.cu
+++ b/src/gbm/gbtree.cu
@@ -1,7 +1,13 @@
 /*!
 * Copyright 2021 by Contributors
 */
+
+#if defined(XGBOOST_USE_CUDA)
 #include "../common/device_helpers.cuh"
+#elif defined(XGBOOST_USE_HIP)
+#include "../common/device_helpers.hip.h"
+#endif
+
 #include "xgboost/context.h"
 #include "xgboost/linalg.h"
 #include "xgboost/span.h"
--- a/src/gbm/gbtree.hip
+++ b/src/gbm/gbtree.hip
@@ -0,0 +1,3 @@
+
+#include "gbtree.cu"
+