[AMD] Remove the deprecated C10_WARP_SIZE (#9356)

704ced1b · Hubert Lu · GitHub · 3cc3d9b9 · 704ced1b
Unverified Commit 704ced1b authored Aug 21, 2025 by Hubert Lu Committed by GitHub Aug 21, 2025
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 3 deletions

sgl-kernel/include/utils.h sgl-kernel/include/utils.h +5 -3

No files found.
--- 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