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

梁圣不再独行!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-11 16:35:06
湖北省最宜居城市有哪些?

湖北省最宜居城市有哪些?

小影的娱乐
2026-10-28 13:37:09
英国球手炮轰斯诺克南京国锦赛:一天双赛不公平,中国球员获优待

英国球手炮轰斯诺克南京国锦赛:一天双赛不公平,中国球员获优待

杨华评论
2026-10-28 22:01:35
媒体就该起到监督作用,孙玉良:需要唱赞歌,请把这个权利给人民

媒体就该起到监督作用,孙玉良:需要唱赞歌,请把这个权利给人民

孙玉良
2026-09-05 08:29:19
基于一中原则,巴基斯坦禁止台湾航空飞越领空,民进党反应亮了

基于一中原则,巴基斯坦禁止台湾航空飞越领空,民进党反应亮了

DS北风
2026-10-29 03:26:07
美联储维持利率不变后 美股小幅回吐涨幅

美联储维持利率不变后 美股小幅回吐涨幅

证券时报
2026-10-29 02:25:01
不出意外,2027年开始,按摩店、足疗店、洗浴店将迎来“大变局”

不出意外,2027年开始,按摩店、足疗店、洗浴店将迎来“大变局”

云霄纪史观
2026-10-25 02:08:15
预测二零二六中超赛季成都蓉城可能会客场零比七惨败山东泰山并以最耻辱的方式夺得冠军

预测二零二六中超赛季成都蓉城可能会客场零比七惨败山东泰山并以最耻辱的方式夺得冠军

体育全天候
2026-10-28 17:08:36
泰勒·斯威夫特红色紧身衣卖断货,亚马逊12美元同款被抢疯

泰勒·斯威夫特红色紧身衣卖断货,亚马逊12美元同款被抢疯

浅遇时光
2026-10-28 03:41:36
上海警方:全市严查!一区配备5台警用无人机,近期多人被罚款

上海警方:全市严查!一区配备5台警用无人机,近期多人被罚款

环球网资讯
2026-10-28 10:13:36
她曾睡遍才子大佬,前半生阅男无数,晚年无人陪伴,如今独居广州

她曾睡遍才子大佬,前半生阅男无数,晚年无人陪伴,如今独居广州

铜臭的历史味
2026-10-13 11:28:57
继手伸进裤子后,48岁刘涛被当众摸胸,全程赔笑不敢怒,多人遭殃

继手伸进裤子后,48岁刘涛被当众摸胸,全程赔笑不敢怒,多人遭殃

小鋭有话说
2026-10-28 21:37:50
德约科维奇2-1逆转安贝尔,晋级ATP500巴塞尔站次轮

德约科维奇2-1逆转安贝尔,晋级ATP500巴塞尔站次轮

懂球帝
2026-10-29 02:05:07
苏联最后元帅的临终前劝诫普京:俄罗斯的未来,答案在中国

苏联最后元帅的临终前劝诫普京:俄罗斯的未来,答案在中国

磊子讲史
2026-10-22 14:31:36
姆巴佩不当言论事件始末

姆巴佩不当言论事件始末

干货收并购
2026-10-27 12:21:16
车厢内人员被甩飞:西班牙列车相撞死亡人数升至50人

车厢内人员被甩飞:西班牙列车相撞死亡人数升至50人

快科技
2026-10-28 19:48:11
关键时刻,蓝营一人倒戈绿营,蒋万安没说“我是中国人”遭反噬?

关键时刻,蓝营一人倒戈绿营,蒋万安没说“我是中国人”遭反噬?

DS北风
2026-10-28 17:10:03
前曼城球星开炮:凯恩拿金球奖?他连德甲最佳都不是

前曼城球星开炮:凯恩拿金球奖?他连德甲最佳都不是

晚风知我意21
2026-10-29 04:03:20
一次交付23架运20B?德专家没想到,美媒:全球最强预警机已量产

一次交付23架运20B?德专家没想到,美媒:全球最强预警机已量产

影孖看世界
2026-10-27 23:30:08
内娱审美再下降,赵丽颖复出造型被群嘲,网友让她体面退休,太扎心

内娱审美再下降,赵丽颖复出造型被群嘲,网友让她体面退休,太扎心

蹲坑看世界
2026-10-26 12:33:13
2026-10-29 04:40:49
智猩猩
智猩猩
AI 技术内容与服务平台
283文章数 15关注度
往期回顾 全部

科技要闻

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

头条要闻

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

头条要闻

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

体育要闻

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

娱乐要闻

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

财经要闻

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

汽车要闻

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

态度原创

健康
数码
旅游
艺术
公开课

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

数码要闻

苹果公布其首款触控MacBook Pro官方设计素材

旅游要闻

忻州古城“古城焕新·产城共荣”入选2026中国旅游产业影响力案例

艺术要闻

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

公开课

李玫瑾:为什么性格比能力更重要?

无障碍浏览 进入关怀版