Skip to content

Commit 5ae4f7a

Browse files
hubertlu-twJustinTong0323
authored andcommitted
[AMD] Remove the deprecated C10_WARP_SIZE (sgl-project#9356)
1 parent e34bc1e commit 5ae4f7a

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

sgl-kernel/include/utils.h

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -331,9 +331,11 @@ inline bool getEnvEnablePDL() {
331331
#ifndef USE_ROCM
332332
#define WARP_SIZE 32
333333
#else
334-
#include <ATen/cuda/CUDAContext.h>
335-
#include <c10/macros/Macros.h>
336-
#define WARP_SIZE C10_WARP_SIZE
334+
#if defined(__GFX9__) || !defined(__HIP_DEVICE_COMPILE__)
335+
#define WARP_SIZE 64
336+
#else
337+
#define WARP_SIZE 32
338+
#endif
337339
#endif
338340

339341
#ifdef USE_ROCM

0 commit comments

Comments
 (0)