add Shap Magic to check if use cat

2023-05-01 21:55:14 +02:00
parent 65d83e288f
commit 313a74b582
4 changed files with 13 additions and 4 deletions
--- a/src/tree/gpu_hist/evaluate_splits.cu
+++ b/src/tree/gpu_hist/evaluate_splits.cu
@@ -14,6 +14,11 @@
 #if defined(XGBOOST_USE_HIP)
 #include <hip/hip_cooperative_groups.h>

+#ifdef __AMDGCN_WAVEFRONT_SIZE
+#undef WAVEFRONT_SIZE
+#define WAVEFRONT_SIZE __AMDGCN_WAVEFRONT_SIZE
+#endif
+
 #define WARP_SIZE WAVEFRONT_SIZE
 #elif defined(XGBOOST_USE_CUDA)
 #define WARP_SIZE 32