EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

[AMD] Remove the deprecated C10_WARP_SIZE (#9356)

Browse Source
This commit is contained in:
Hubert Lu
2025-08-21 18:16:35 -07:00
committed by GitHub
parent 3cc3d9b950
commit 704ced1b2e
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
sgl-kernel/include/utils.h
View File

@@ -331,9 +331,11 @@ inline bool getEnvEnablePDL() {
#ifndef USE_ROCM #ifndef USE_ROCM
#define WARP_SIZE 32 #define WARP_SIZE 32
#else #else
#include <ATen/cuda/CUDAContext.h> #if defined(__GFX9__) || !defined(__HIP_DEVICE_COMPILE__)
#include <c10/macros/Macros.h> #define WARP_SIZE 64
#define WARP_SIZE C10_WARP_SIZE #else
#define WARP_SIZE 32
#endif
#endif #endif
#ifdef USE_ROCM #ifdef USE_ROCM