Skip to content

GitLab

Unverified Commit 704ced1b authored by Hubert Lu's avatar Hubert Lu Committed by GitHub
Browse files

[AMD] Remove the deprecated C10_WARP_SIZE (#9356)

parent 3cc3d9b9
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
sgl-kernel/include/utils.h
View file @ 704ced1b
...@@ -331,9 +331,11 @@ inline bool getEnvEnablePDL() { ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment