fix: int32 overflow in `trtllm_fp4_block_scale_moe` causing "Unsupported hidden state scale shape" for EP32+ configs#2853

Merged

yzh119 merged 3 commits intoflashinfer-ai:mainfrom

qiching:fix/int32-overflow-fp4-moe

Mar 24, 2026

Commits on Mar 22, 2026

fix: int32 overflow in trtllm_fp4_block_scale_moe for large EP configurations
Albert Cheng (Engrg-Hardware 1)
committed

Commits on Mar 23, 2026

chore: also change hidden_states_scale_vec_size to int64_t for consistency
Albert Cheng (Engrg-Hardware 1)
committed
style: apply clang-format
Albert Cheng (Engrg-Hardware 1)
committed