mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-04-12 06:27:02 +08:00
Replace hardcoded WARP_SIZE=32 with the dynamic WARP_SIZE macro from cuda_compat.h to correctly support both Wave64 (MI300X/gfx942) and Wave32 (Strix Halo/gfx1151) architectures. The previous hardcoded value was incorrect for AMD CDNA GPUs which use 64-wide wavefronts. While the current static_assert (kWarpSize >= 4) passes for both 32 and 64, having inconsistent WARP_SIZE definitions across the codebase is a maintenance issue and potential latent bug. Changes: - Add cuda_compat.h include for WARP_SIZE macro - Replace local WARP_SIZE constant with kWarpSize from cuda_compat.h - Update static_assert and comments to use kWarpSize Signed-off-by: c0de128 <kevin.mckay@outlook.com>