#54845 [ROCm][Perf] Add low-M FP32 router GEMM for gfx950
原始 PR · 作者 Fangzhou-Ai · 合并时间 2026-09-04 01:42
为gfx950添加专用FP32路由器GEMM Triton内核,显著降低小token批处理延迟。
此PR值得精读,特别是其针对特定硬件和形状的内核设计、与现有分发逻辑的无缝集成方式,以及全面的性能与正确性验证方法。其设计决策(如`M`为运行时参数、基于形状的启动配置、自定义算子包装以支持动态编译和CUDA图)体现了对vLLM编译和捕获流程的深刻理解,具有借鉴价值。对于关注ROCm性能优化、MoE路由器实现或Triton内核开发的工程师,本PR提供了高质量的参考实现。主要关注点是其硬件特异性和形状硬编码带来的维护性问题,以及分发逻辑的优先级管理。
参与讨论