format

2023-10-23 22:29:48 -07:00
parent 558352afc9
commit 79319dfd4d
8 changed files with 115 additions and 100 deletions
--- a/src/tree/gpu_hist/evaluate_splits.cu
+++ b/src/tree/gpu_hist/evaluate_splits.cu
@@ -11,7 +11,9 @@
 #include "evaluate_splits.cuh"
 #include "expand_entry.cuh"

-#if defined(XGBOOST_USE_HIP)
+#if defined(XGBOOST_USE_CUDA)
+#define WARP_SIZE 32
+#elif defined(XGBOOST_USE_HIP)
 #include <hip/hip_cooperative_groups.h>

 #ifdef __AMDGCN_WAVEFRONT_SIZE
@@ -20,8 +22,6 @@
 #endif

 #define WARP_SIZE WAVEFRONT_SIZE
-#elif defined(XGBOOST_USE_CUDA)
-#define WARP_SIZE 32
 #endif

 #if defined(XGBOOST_USE_HIP)