网易首页 > 网易号 > 正文 申请入驻

梁圣不再独行!AI 自己写算子,屠榜华为算子榜

0
分享至

智猩猩AI整理

编辑:林夕

10月20日,DeepSeek开源推理框架DeepInfer,集成FlashMLA、DeepGEMM等高性能组件。

同日更新的deepseek-recipe,则补充了请求转换、模型输入编码和输出解析等服务接入能力。

这个推理框架的内容,FlashMLA、DeepGEMM等组件给行业划了两样东西:

一份“最值得做的算子”,和一把“人类顶尖专家手工打磨”的标尺。

不久后,面壁智能ForgeKernel宣布:它用 AI 写的算子,在英伟达和华为两家的算子榜单上同时拿到第一。

一边是专家积累的高性能实现,一边是AI写出的榜首成绩。然而,工程师对这场变化的感受,比外界更早。

就在一个多月前,DeepSeek V4.1发布后,参与其核心算子开发的刘胜与写下了一篇长文:《我不得不把才华埋葬在昨天》。

曾经,亲手调优算子、跑赢公认实现,是他享受的工作。

但AI的进步,让这位工程师开始重新思考自己的角色:

与其继续亲手打磨每一段代码,不如研究怎样让AI把算子写得更好。


在随后发布的补充说明中,他表示,会继续写算子,也会不断引入AI Agent参与。

他反复打磨的,正是模型运行中被大量调用的底层计算代码。

FA3(FlashAttention-3)当年靠重写算子在 H100 上快了近两倍,但这种成果往往半年才兑现一次。

而今年9月发布的GPT-6 Astra,在D2K-Bench的26项算子任务中,无需额外的专家设计指导,整体性能就超过了人类专家参考实现。

从受控实验走向公开打榜,ForgeKernel则把AI写算子的能力带到了英伟达GPU和华为昇腾两类硬件上。

英伟达侧,FlashInfer-Bench在B200上评测26项任务,覆盖DeepSeek-V3/R1等模型的推理计算。提交实现需通过正确性与作弊检查,最终按SOL Score排名。

ForgeKernel完成全部26项任务,以 0.814011的SOL Score登顶,排在Infinigence AI、Geometric、Amir M. Mir等参评者之前,平均加速比达到 5.49x。


华为侧,CANNBench面向昇腾芯片,official-tasks 2.0评测集包含60个算子。

在此前于10月15日封榜的v1.1中,ForgeKernel已位居A3硬件榜第一,基准加速比达到1.835×,比第二名infynai-kernaut高约24.5%。


10月20日,v2.0开榜后,ForgeKernel持续保持榜首。

截至10月26日,已提交55个算子,以4130.38分领先第二名serialization-swift约6.5%;基准加速比达到1.919×,高于serialization-swift的1.197×和MetaRSI的1.220×。


此外,ForgeKernel也以3077.45分登顶CANNBench v2.0的950DT硬件榜,基准加速比达到1.589×,并进一步拿下950PR硬件榜第一。



目前,ForgeKernel已包揽CANNBench的A3、950DT和950PR三个硬件榜单第一。

除了榜单成绩,ForgeKernel也尝试了针对具体模型的优化。

在DeepSeek V4.1 Flash 的相关算子上,ForgeKernel通过AI从零编写、持续迭代,实现了对DeepSeek开源英伟达版本的性能超越。

ForgeKernel的价值,也在榜单之外:让AI承担更多算子开发与优化,降低跨硬件适配成本,让新模型更快在国产芯片上跑好。

对大规模训练而言,已有硬件多发挥一些性能,省下的就是真金白银。

01

专家来不及做的优化,

AI能否补上

ForgeKernel采用AI编写算子,针对的是高性能软件开发中的一项现实难题:

模型和硬件不断更新,依靠专家逐项适配、调优,投入越来越大。

新模型可能引入新的计算结构,同一模型在不同输入长度、批量大小和精度下,也可能需要不同实现。

换一种芯片,工具链和优化方式还要随之调整。

这些工作通常需要工程师反复测试,方案写出来之后,要检查计算结果、分析运行耗时,再决定继续修改还是更换路径。

ForgeKernel让AI承担更多实现与试验工作,尝试降低这部分投入。

模型根据目标自主调用工具,完成代码编写、编译调试、精度检查和性能优化。

能否编译、结果是否正确、运行有没有变快,都能在实际环境中验证,并成为下一轮修改的依据。

这种方式也为按需定制提供了空间。

围绕具体芯片和负载尝试更多方案,过去因专家投入过高而不划算的优化,可能变得值得做。

对国产算力而言,模型更新后能否及时适配、跑通后能否充分调优,都影响芯片的实际利用。

评价这类方案,需要把性能收益、开发周期、人工投入和测试开销放在一起看。

02

模型已经会自己写算子了,

还需要人来教吗?

让AI完成算子优化,需要给它多少指导?

Justgoal做了一个直接的尝试:给基础模型明确的优化目标,配上通用工具,让它自行决定怎样写代码、调试和优化性能。

不预先规定每一步使用什么优化方法,也不要求模型沿着专家设计的固定路线执行。

编译、测试、分析结果,再修改实现,由模型自主推进。

在论文所测任务中,较强基础模型配合极简Goal模式,已能在任务组平均表现上超过人类专家参考实现。

研究还将它与MIT韩松团队参与开发的算子优化智能体系统KDA-0.5进行了对比。

在公开的赛后测试中,KDA-0.5生成的算子在FlashInfer竞赛三类任务上均超过了人类最佳参赛实现。

而在双方共同测试的任务组中,极简Goal模式取得了更高的平均加速比。

这组实验提出了一个值得讨论的问题:当模型已经能够自主寻找优化方案,专用Harness还应该补什么?

Harness是支撑模型执行任务的运行系统。

专用Kernel Agent通常会把领域经验组织成流程,帮助模型缩小搜索范围,减少无效尝试。

但流程需要随着模型能力调整。

ForgeKernel的设计判断是,当模型已经具备足够的算子知识和优化能力,部分固定指导可能限制探索;系统更需要补足的,是模型持续完成复杂任务的能力。

Justgoal与ForgeKernel的关系,也由此清晰起来。

Justgoal采用极简通用Harness,用于探索基础模型自主优化算子的能力。

ForgeKernel则在此基础上加入对算子任务的理解,重点增强长程规划与自进化能力。

模型知道某种优化方法,并不等于它能在较长的开发过程中始终作出合适判断。

面对多轮试验,它还需要决定哪些方向值得继续、何时调整思路,以及怎样把局部改进落实到最终实现。

当前Agent更擅长细节调优,在独立设计高效的流水线级架构方面仍有不足。

因此,ForgeKernel的一个关键动作,是帮助模型找到更好的优化起点,再持续挖掘性能空间。

在此基础上,ForgeKernel通过自进化机制积累并利用优化经验,支持模型持续推进任务。

ForgeKernel 在 Justgoal 之上引入了面向算子任务的系统性设计,核心机制将随后续技术论文公布。

目标Prompt负责说明任务,Harness则支撑模型持续执行与改进。这也是ForgeKernel不能简单归为提示词工程的原因。

跨硬件适配,是检验这套系统设计的一项实际场景。

03

省下真金白银:

国产芯片从“能跑”到“跑好”的距离

ForgeKernel在英伟达GPU与华为昇腾上采用同一套优化流程,所需知识由模型在任务过程中构建。

具体代码仍要适应各自硬件,但组织优化任务的流程可以复用。

这为减少跨平台开发中的重复投入,提供了一条可探索的路径。

生成的Kernel也已接入两侧训练框架,完成MiniCPM 5 0.9B长训测试。

相比榜单测试,这一步让AI写出的代码进入了实际训练流程。

国产模型也参与了其中的任务,在ForgeKernel的实验中,DeepSeek V4 Flash 0831已被用于部分子Agent任务,并成功完成任务。

这意味着,被超越者的模型,也在参与超越者的工作。

在模型训练等高投入场景中,这种开发方式尤其值得关注。

训练中的核心计算会反复执行,一段算子节省的时间,可以随着大量训练步骤持续累积。

围绕确定的模型、芯片和输入配置深度定制,进一步逼近硬件性能上限,就可能减少整轮训练的时间和算力开销。

当开发与试验成本下降,更多针对特定负载的优化也可能变得划算。

这种优化背后,是万亿美元级的投入。Gartner预计,2026年全球AI支出约2.7万亿美元,其中AI基础设施支出约1.48万亿美元。

以这一支出规模作量级参照,1%就对应约148亿美元。

这并不意味着算子提速1%就能直接省下这笔钱,但足以说明:

当优化覆盖大量实际负载,即使幅度不大,也可能产生可观的经济价值。

双榜第一展示了AI写算子的性能,长训测试则让这些代码进入实际训练验证。

一个多月前,刘胜与写下了对手写算子时光的感慨。

如今,ForgeKernel的双榜成绩,让AI接手更多底层开发有了具体的性能证明。

关注+星标,获取AI前沿进展与开源一线动态

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
安倍晋三是日本人,可为啥他遇刺后,墓碑上却刻的是中国汉字呢?

安倍晋三是日本人,可为啥他遇刺后,墓碑上却刻的是中国汉字呢?

陌上桃花开的
2026-10-28 23:01:06
宣萱首度剖白父親突離世 90多歲家中猝逝:其實係祝福

宣萱首度剖白父親突離世 90多歲家中猝逝:其實係祝福

粤睇先生
2026-10-27 23:38:44
1967年贺敏学被抓捕,李敏向父亲报告后,伟人指示马上将其放出

1967年贺敏学被抓捕,李敏向父亲报告后,伟人指示马上将其放出

唠叨说历史
2026-01-20 16:51:16
彩票出现断崖式暴跌,“2元中500万”为何没人信了?原因引人深思

彩票出现断崖式暴跌,“2元中500万”为何没人信了?原因引人深思

阿振观点
2026-08-21 05:38:20
乌克兰真到末路了?北约当场拒绝泽连求援:我们也没爱国者了

乌克兰真到末路了?北约当场拒绝泽连求援:我们也没爱国者了

贵州霖霖
2026-10-27 21:31:48
央视拒接受皇马解释,姆巴佩言论致西甲转播停摆

央视拒接受皇马解释,姆巴佩言论致西甲转播停摆

日常碎碎念啊
2026-10-29 01:45:12
去公园遛弯时你注意没,北京这些老头老太太一身“年轻装”

去公园遛弯时你注意没,北京这些老头老太太一身“年轻装”

三农老历
2026-10-28 01:37:49
央媒怒批、坑害老百姓!臭名昭著的四大相声名人,个个难以宽恕

央媒怒批、坑害老百姓!臭名昭著的四大相声名人,个个难以宽恕

三石记
2026-10-21 10:42:12
大案纪实:胆大妄为!开封前银行副行长杀人分尸,竟将尸块抛在包公祠旁,民众直呼苍天有眼

大案纪实:胆大妄为!开封前银行副行长杀人分尸,竟将尸块抛在包公祠旁,民众直呼苍天有眼

大案纪实故事
2026-10-10 15:48:16
英媒:不同寻常!英国三艘战略核潜艇同时出海

英媒:不同寻常!英国三艘战略核潜艇同时出海

参考消息
2026-10-28 21:33:08
新华社快讯:西班牙高速铁路列车相撞造成至少39人死亡

新华社快讯:西班牙高速铁路列车相撞造成至少39人死亡

新华社
2026-10-28 17:39:25
拿恐吓换来谅解书、高血压写进量刑考量,徽剑案网友“呼吁抗诉”值得被看见!

拿恐吓换来谅解书、高血压写进量刑考量,徽剑案网友“呼吁抗诉”值得被看见!

迷世书童
2026-10-28 19:50:20
辽宁女孩考公务员差1分,政审时考官看见档案备注,当场愣住了

辽宁女孩考公务员差1分,政审时考官看见档案备注,当场愣住了

第四思维
2025-08-04 23:05:53
很扎心!妻子感冒想让丈夫回来照顾孩子,丈夫要应酬转账10000,妻子:该不该离婚?网友吵翻了!

很扎心!妻子感冒想让丈夫回来照顾孩子,丈夫要应酬转账10000,妻子:该不该离婚?网友吵翻了!

丫头舫
2026-10-26 18:43:28
中国与越南激烈交锋:中国借给越南的白龙尾岛,要不回来了?

中国与越南激烈交锋:中国借给越南的白龙尾岛,要不回来了?

军情观察家
2026-10-22 15:02:51
就业噩梦才刚开始:今年是未来十年就业最轻松的一年

就业噩梦才刚开始:今年是未来十年就业最轻松的一年

黑噪音
2026-10-28 21:15:24
iOS27.2分梯队推送,机型功能支持差异对照

iOS27.2分梯队推送,机型功能支持差异对照

小柱解说游戏
2026-10-27 03:11:08
国务院定调,减缓人工智能的应用。全都是机器人了,大量人工下岗,没有工作,没有收入,谁去消费?

国务院定调,减缓人工智能的应用。全都是机器人了,大量人工下岗,没有工作,没有收入,谁去消费?

扶苏聊历史
2026-10-26 16:14:21
中期选举前夕民调出炉:特朗普支持率脱离低点,民主党大概率赢下众议院

中期选举前夕民调出炉:特朗普支持率脱离低点,民主党大概率赢下众议院

财联社
2026-10-27 11:54:19
山姆悄悄布局了四家"县级"店

山姆悄悄布局了四家"县级"店

新零售财经
2026-10-28 11:55:47
2026-10-29 03:32:49
智猩猩
智猩猩
AI 技术内容与服务平台
283文章数 15关注度
往期回顾 全部

科技要闻

宇树科技前三季净利暴增270%

头条要闻

仅退款后冰柜仍插着电在使用 买家回应

头条要闻

仅退款后冰柜仍插着电在使用 买家回应

体育要闻

0胜4负与篮网国王一桌 热火字母哥赌错了?

娱乐要闻

巾帼不让须眉!章子怡拿下重磅荣誉

财经要闻

iPhone Duo七年等来一场吐槽大会

汽车要闻

首销期权益价23.98万起 2026款理想i6上市

态度原创

教育
手机
健康
艺术
本地

教育要闻

一场错位的需求拉锯!“没钱,干陪有什么用?”初中女生的话戳破无数全职妈妈的幻想!

手机要闻

华为Pura90 Plus、畅享100系列发布!多款新品亮相

查不出猝死风险?可能是项目没做对

艺术要闻

23张人性与爱的照片,第1张就让人破防了……

本地新闻

十月奔赴迪庆,解锁梅里日照金山

无障碍浏览 进入关怀版