腾讯混元AI Infra核心技术重磅开源:推理吞吐提升30%!

源自知乎:腾讯技术工程

01-28 16:22

大模型时代,计算效率是关键瓶颈。腾讯混元团队开源的高性能LLM推理算子库HPC-Ops,通过底层架构与指令级优化,在主流推理卡上将核心算子性能逼近硬件极限,显著提升推理吞吐,为开发者提供了生产级的高效解决方案。

腾讯混元AI Infra核心技术重磅开源:推理吞吐提升30%!智能速览

  • 腾讯开源高性能LLM推理算子库HPC-Ops。

  • 通过深度优化,核心算子性能逼近硬件极限。

  • 混元模型推理吞吐提升30%,DeepSeek提升17%。

  • Attention算子最高提升2.22倍,GroupGEMM最高提升1.88倍。

  • 未来将支持稀疏Attention和更多量化方案。

腾讯混元AI Infra核心技术重磅开源:推理吞吐提升30%!精华内容

HPC-Ops的开源,为解决大模型推理效率难题提供了一个新的视角。其如何通过技术与架构创新,实现超越现有SOTA的性能,值得深入探究。

直面推理瓶颈

主流算子库多针对NVIDIA H800等高配训练卡优化,但受限于客观条件,大规模线上推理服务常采用H20等推理卡。现有方案在这些卡上难以发挥硬件峰值性能,且业务对极致吞吐、低延迟及Blockwise FP8等复杂量化策略的需求日益迫切,亟需更适配的底层支持。

核心优化策略

HPC-Ops基于CUDA和CuTe构建,通过任务特性与硬件能力对齐、精细的任务调度和数据重排、聚焦计算逻辑本身三大策略实现突破。

针对访存瓶颈,调整指令顺序进行数据预取,访存带宽可达硬件峰值80%以上。通过创新的数据重排技术,如FP8 Attention中的Interleave重排,减少了线程间数据shuffle。

同时,基于CuTe扩展的vec抽象层,统一负责高效数据搬运,让开发者能聚焦算法,降低维护门槛。

实测性能突破

在真实场景下,HPC-Ops展现出卓越性能。端到端层面,混元模型推理QPM提升30%,DeepSeek模型QPM提升17%。

单算子性能对比更为突出:GroupGEMM在低延迟场景下相比DeepGEMM最高提升1.88倍;FusedMoE相比TensorRT-LLM最高提升1.49倍;Attention算子在Decode场景下BF16精度最高提升2.22倍,FP8精度下最高提升2.0倍。

开源与未来规划

HPC-Ops已在腾讯大规模生产环境验证,提供简洁易用的API,可无缝对接vLLM、SGLang等主流框架,并原生支持BF16、FP8等多精度量化方案。

未来,HPC-Ops将持续深耕,计划研发稀疏Attention算子以解决长上下文模型瓶颈,拓展4bit/8bit混合精度等更多量化方案,并布局计算-通信协同优化内核,为超大规模模型高效部署提供支撑。

腾讯开源的HPC-Ops,不仅是其自身技术实力的体现,更为整个AI社区提供了宝贵的生产级工具。它通过极致性能优化和降低开发门槛,加速了大模型技术的落地应用。未来,围绕更高效、更低成本的推理技术探索,将继续是行业发展的核心议题。

内容由AI生成
0
扫一下,分享更方便,购买更轻松
0评论

当前文章无评论,是时候发表评论了
提示信息

取消
确认
评论举报

最新文章 热门文章