优化 Kimi-K3 路由器权重加载,通过预取技术将单 token 场景的 PDL 等待延迟降低最高 20%。
这个 PR 值得精读,特别是对于从事 GPU 内核优化或关注 vLLM 性能工程的工程师。它展示了一个清晰且实现精良的性能优化案例:如何通过分析硬件特性(PDL 等待期间的空闲)找到优化点,并通过谨慎的设计(预取少量 tile、严格的运行时条件判断、独立缓存)在提升目标路径性能的同时,最小化对通用路径的侵入。核心的设计决策(`_vector_dotprod_prefetched` 的实现、基于 M 的运行时分派)及其背后的硬件权衡(CUDA 核心 vs 张量核心、预取窗口大小选择)具有较高的技术洞察价值。
参与讨论