FP8模型不再挑卡!DeepSeek推理成本减半速度翻番,清华团队开源「赤兔」推理引擎

FP8模型不再挑卡!DeepSeek推理成本减半速度翻番,清华团队开源「赤兔」推理引擎
2025年03月14日 11:44 机器之心Pro

机器之心原创

作者:闻菲、张倩

「国产大模型 + 国产引擎 + 国产芯片」的完整技术闭环正在加速形成。

随着 DeepSeek 引燃市场对于大模型私有化部署的热情,模型部署的效率与成本成为企业 AI 落地的关键瓶颈。

今天,清华系科创企业清程极智清华大学翟季冬教授团队联合宣布开源大模型推理引擎「赤兔」(Chitu),率先实现了非 H 卡设备(英伟达 Hopper 架构之前的 GPU 卡及各类国产卡)运行原生 FP8 模型的突破

在 A800 集群上的实测数据显示,用赤兔引擎部署 DeepSeek-671B 满血版推理服务,相比于 vLLM 部署方案,不仅使用的 GPU 数量减少了 50%,而且输出速度还提升了 3.15 倍

赤兔引擎代码已发布至 GitHub 开源社区:

  • https://github.com/thu-pacman/chitu

赤兔 (Chitu) 开源:

国产大模型基础设施又一里程碑

当前,拥有完整自主可控的 AI 技术栈已成为国家战略性需求。DeepSeek、QwQ 等优秀的国产开源大模型正在重塑全球科技产业格局,但在大模型推理部署领域,业界仍高度依赖国外开源工具。

赤兔引擎的开源为业界提供了国产开源新选择,也意味着国产大模型 + 国产引擎 + 国产芯片」的完整技术闭环正在加速形成

「我们看到国内大模型领域取得了显著进步,但在基础设施层面,尤其是推理引擎这一核心环节仍缺乏生产级开源产品。」清程极智 CEO 汤雄表示,开源赤兔引擎是我们助力国内 AI 生态建设的重要一步。」

作为清华大学高性能计算研究所所长,翟季冬教授长期致力于高性能计算与系统软件优化研究。翟教授指出:「赤兔引擎凝结了团队多年的并行计算编译优化技术积累,目标是建立一个真正适合国内多元算力环境的高性能推理引擎,能够弥合先进模型与多样化硬件之间的差距,为中国大模型的产业落地提供关键支撑。」

本次开源的版本特别聚焦于当前市场最迫切的需求 —— 即实现 DeepSeek FP8 精度模型在存量英伟达 GPU 上的无损且高效部署。团队还透露,针对多款国产芯片特别优化的版本也将相继对外开源。这一突破不仅降低了大模型落地门槛,也为国产 AI 算力的发展带来了新的契机。

赤兔引擎核心优势:

全场景性能优化与架构适应性

随着 DeepSeek V3 和 R1 等新一代模型采用 FP8 混合精度训练并发布,企业面临新的部署挑战。这些 FP8 模型针对 NVIDIA Hopper 架构(H100/H200)深度定制,在其他硬件上部署时(包括 NVIDIA 其他架构 GPU 和国产芯片)需要进行额外的适配。

先进的模型与最难获取的硬件绑定,这是当前企业大模型落地的最大痛点之一汤雄表示。随着更新一代模型和芯片往 FP4 等新型数据精度方向的发展,这种代际效应将愈发显著。

另一个挑战是现有开源推理引擎对多元算力环境的支持不足。当前,vLLM 等主流引擎主要针对 NVIDIA 最新架构优化,对国产芯片或较老款 GPU 的适配并不理想。在国内企业私有化部署场景中,NVIDIA 的 Ampere 系列 GPU、国产芯片等多元算力占据了相当比例,这些场景亟需更加灵活的解决方案。

赤兔从一开始就定位于生产级大模型推理引擎,并且充分考虑了企业 AI 落地从小规模试验到大规模部署的渐进式特点,专注于提供以下重要特性:

  • 多元算力适配:不仅支持 NVIDIA 最新旗舰到旧款的多种型号,也为国产芯片提供优化支持。

  • 全场景可伸缩从纯 CPU 部署、单 GPU 部署到大规模集群部署,赤兔引擎提供可扩展的解决方案。

  • 长期稳定运行:可应用于实际生产环境,稳定性足以承载并发业务流量。

在性能优化方面,赤兔的设计理念是「对症下药」,支持「低延迟优化」、「高吞吐优化」和「小显存优化」,可以根据不同场景需求,在不同硬件配置和系统环境下,针对不同负载的特性,提供相应的最优解决方案。在实际部署中,这意味着赤兔引擎可以根据系统资源状况,在 GPU 利用率、内存效率和网络传输之间寻找最佳平衡点。

本次开源的技术突破是实现非 H 卡设备原生运行 DeepSeek FP8 精度模型。「我们没有走简单的量化路线,而是通过在算子内部高效处理 FP8 数据,确保模型推理质量不受任何影响。」汤雄表示:「具体来说,我们对 GeMM、MoE 等一系列关键算子进行了指令级的优化,实现了 FP8 数据的原生处理能力。」

这种技术实现的难点在于需要深入理解不同硬件架构的指令集特性,以及 FP8 与其他精度格式之间的数学等效关系,结合硬件特性分析并优化。得益于团队在系统软件人才方面的积累,才能够在较短时间内实现这一突破。传统上为每种芯片架构人工编写优化代码的工作量巨大,而团队所具备的智能编译技术有助于加速高性能算子的开发过程。

在 A800 集群上的实测结果表明,与 vLLM 相比,赤兔引擎实现 DeepSeek-671B 满血版推理所需 GPU 数量减少 50%;尽管只用了一半的算力,但推理速度却达到其 3.15 倍。这意味着企业可以用一半的硬件资源获得更高的推理性能,极大降低了部署门槛和运营成本。

「部署大模型的最低算力成本是一个时常被忽视但极其重要的指标。」清程团队强调,显著降低推理成本对推广 AI 应用至关重要,赤兔推理引擎在设计之初就将降低用户的算力成本纳入核心考量。

开源共建

为国产芯片「抢时间」

赤兔推理引擎的开源标志着清程极智的业务发展进入新的阶段。清程极智在持续提供企业级软件部署运维服务和全套人工智能解决方案的同时,将把更多实际业务中的经验通过开源的方式回馈社区。

「目前英伟达 GPU 在大模型推理市场占据主流地位,未来几年可能出现国外 AI 芯片和国内 AI 芯片百花齐放的态势。」汤雄分析指出:「我们希望赤兔引擎成为连接多元算力与大模型应用的桥梁,同时通过推理一体机产品为企业提供端到端的大模型部署解决方案。」

清程极智的推理一体机产品基于赤兔引擎,适配多种硬件和模型,能够为企业提供开箱即用的私有化部署方案,并配套专业的运维与优化服务。

对于国产 AI 基础设施而言,赤兔引擎的开源有助于弥合国产芯片与国际先进芯片之间的「时间差」—— 当新模型架构和新数据类型出现时,赤兔引擎可以帮助国产芯片快速适配。

其次,开源引擎减轻了芯片厂商的软件开发负担。芯片厂商可以专注于硬件创新,而不必每家都重复开发完整的软件栈。翟季冬教授表示,「我们目前与多家国产芯片厂商进行合作,各方可直接向赤兔项目贡献代码,共同建设一个更强大的开源生态。」

「开源就是做给大家用,用了才知道好不好。我们把代码放到 GitHub 上,有问题提 issue,有想法提 PR,一起把它做得更好。」汤雄超说。

「无论用什么卡,赤兔都是高性价比的选择。」

新浪科技公众号
新浪科技公众号

“掌”握科技鲜闻 (微信搜索techsina或扫描左侧二维码关注)

创事记

科学探索

科学大家

苹果汇

众测

专题

官方微博

新浪科技 新浪数码 新浪手机 科学探索 苹果汇 新浪众测

公众号

新浪科技

新浪科技为你带来最新鲜的科技资讯

苹果汇

苹果汇为你带来最新鲜的苹果产品新闻

新浪众测

新酷产品第一时间免费试玩

新浪探索

提供最新的科学家新闻,精彩的震撼图片