From 704ced1b2ec45a87bb42494fe9da18fa5004e546 Mon Sep 17 00:00:00 2001
From: Hubert Lu <55214931+hubertlu-tw@users.noreply.github.com>
Date: Thu, 21 Aug 2025 18:16:35 -0700
Subject: [PATCH] [AMD] Remove the deprecated C10_WARP_SIZE (#9356)

---
 sgl-kernel/include/utils.h | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/sgl-kernel/include/utils.h b/sgl-kernel/include/utils.h
index 56f322764..5cab0786c 100644
--- 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