新增手动融合入口,将激活量化融合从编译器 Pass 迁移至模型代码显式调用。
此 PR 是 vLLM 量化优化架构演进的重要一步,引入了一个设计良好、可扩展的手动融合框架。虽然当前只应用于 Llama 模型,但其接口设计为扩展到其他模型和量化方案奠定了坚实基础。代码实现清晰,测试覆盖全面(包括单元测试和端到端测试)。主要风险在于可能引入的 CI 失败(如 IMA 问题)和性能的精确量化。建议: 1. **值得精读**:对于关注量化优化、编译器融合或模型性能的工程师,本 PR 的实现细节(尤其是融合映射机制和接口设计)非常有参考价值。 2. **重点关注**:需要仔细审查与 CI IMA 问题相关的具体原因,确保稳定性。同时,关注后续的 TP2 一致性测试和 A/B 性能对比结果。 3. **建议动作**:建议在解决 CI 问题后合并,并跟踪其在更大规模模型和硬件上的表现,为后续扩展到其他模型(如 Mixtral)提供数据支持。