[AMD] Remove the deprecated C10_WARP_SIZE (#9356)

2025-08-21 18:16:35 -07:00
parent 3cc3d9b950
commit 704ced1b2e
1 changed files with 5 additions and 3 deletions
--- a/sgl-kernel/include/utils.h
+++ b/sgl-kernel/include/utils.h
@@ -331,9 +331,11 @@ inline bool getEnvEnablePDL() {
 #ifndef USE_ROCM
 #define WARP_SIZE 32
 #else
-#include <ATen/cuda/CUDAContext.h>
-#include <c10/macros/Macros.h>
-#define WARP_SIZE C10_WARP_SIZE
+#if defined(__GFX9__) || !defined(__HIP_DEVICE_COMPILE__)
+#define WARP_SIZE 64
+#else
+#define WARP_SIZE 32
+#endif
 #endif

 #ifdef USE_ROCM