编辑:桃子 好困

【新智元导读】 在 Claude Code 30 分钟迁移 CUDA 引发热议的背后,众智 FlagOS 社区已经给出了一条更系统的答案:通过 KernelGen 与统一编译器 FlagTree,让算子在多种 AI 芯片上实现自动生成、验证与优化,真正降低跨芯片软件迁移成本。

围绕「CUDA 护城河是否松动」的讨论,本质指向一个更现实的问题:

算子能否在不同芯片上被自动生成、正确运行,并具备可用性能?

算子自动生成,已经跑在多芯片上了

对此,众智 FlagOS 社区在2026年1月初推出了升级版 KernelGen——一个支持多种 AI 芯片的高性能 Triton 算子生成自动化工具,并在真实多芯片环境下完成了系统性评测。

从实际数据来看,KernelGen 已不再停留在概念或 Demo 阶段:

生成可编译运行算子的成功率(生成成功率):82%

整体算子数值准确性通过(执行正确率):62%

覆盖英伟达,以及华为、摩尔、海光、天数等多款国产 AI 芯片

这里所说的「生成成功率」,是指代码能编译、能运行。但与普通代码生成不同,算子生成对数学精度提出了更为严苛的要求:需要在多种输入变化条件下,始终保持高精度、可复现的一致数值结果,这也正是文中所强调的「执行正确率」。

如果说「生成成功率」关注的是算子是否能编译、能运行,那么真正的技术门槛在于是否能跑得对。在多种芯片架构并存的场景下,这一挑战被进一步放大——当同一份算子代码需要同时适配华为、摩尔线程、海光、天数等架构各异的芯片时,数值精度、舍入策略、指令调度顺序、缓存层级等细微差异,都可能导致结果偏差,甚至引发正确性失效。

一次生成、一次编译,在多芯片平台上实现数值一致、结果可验证,才是算子自动生成真正需要跨越的核心门槛。

这些结果表明,算子自动生成在多芯片环境下已经具备可行性与工程实用价值

实验与实现细节:KernelGen是如何工作的?

实现原理:从「写算子」到「生产算子」

KernelGen 的目标并非辅助开发者写代码,而是覆盖算子从需求到落地的完整生命周期

输入层:用户可通过自然语言、数学公式或已有实现描述算子需求

生成层:基于大模型与智能体技术,理解算子语义并自动生成 Triton 内核

验证层:自动构建测试用例,在目标芯片上与 PyTorch reference 实现进行严格的数值一致性校验

评估与优化层:对生成算子进行性能评测,量化加速比,并通过自动化调优持续优化执行效率

KernelGen 用户在对话框里仅需输入自然语言、数学公式或是已有实现表达算子开发需求

这一流程的核心目标是:

将算子开发从「专家手工活」,转变为可复制、可扩展的工程流程。

为什么必须与 FlagOS / FlagTree 协同?

在多芯片场景下,仅生成算子代码并不足以解决工程问题,不同 AI 芯片在以下方面差异显著:

并行模型与计算单元组织方式

内存层级与访存语义

指令集与编译稳定性

因此,KernelGen 从设计之初就被纳入 FlagOS 生态,并与统一 AI 编译器 FlagTree 深度协同:

由 FlagTree 提供统一的硬件抽象与编译基础

将硬件差异尽可能收敛在编译器的中间表示层

提升算子在多芯片环境下的编译性能与正确性

这也是算子自动生成首次具备跨芯片工程可行性的关键前提。

FlagTree:支撑算子自动生成的统一AI编译器底座

KernelGen 能够在多芯片环境下实现算子自动生成与验证,并不只是模型能力的结果,其关键支撑来自统一 AI 编译器 FlagTree

FlagTree 是众智 FlagOS 社区长期推进的统一编译器项目。从 2025 年 3 月发布 v0.1,到 2026 年 1 月 5 日发布 v0.4,已逐步发展为面向异构 AI 计算的通用编译基础设施:

已支持 12 家厂商、近 20 款 AI 芯片,覆盖芯片包括华为昇腾、寒武纪、沐曦、摩尔线程、海光

支持架构从 DSA、GPGPU,扩展到 RISC-V AI 芯片、ARM 等多种体系

在技术设计上,FlagTree 主要解决两类问题:

硬件差异隔离:通过统一的硬件中间表示(计算单元、内存层次、原子操作等),将芯片差异最大程度收敛在编译器内部,而不是暴露给算子生成逻辑。

性能与易用性的平衡:在保留 Triton 高级语法的同时,引入硬件感知优化,避免「一套代码跑所有芯片」带来的性能和稳定性问题。

FlagTree v0.4 中,FlagOS 社区进一步引入 TLE(Triton Language Extensions),以分层方式扩展 Triton 的跨芯片表达能力:

TLE-Lite:一次编写,多后端运行,适用于快速验证与轻量级优化

TLE-Struct:面向算子开发者的架构感知调优接口

TLE-Raw:允许直接内联 CUDA、MLIR 等厂商原生代码,用于极致性能场景

通过 FlagTree 的编译支撑,KernelGen 自动生成的算子才能在不同芯片上保持较高的编译通过率和执行稳定性。

性能指标与验证情况:多芯片、多模型、多轮评测

不同芯片上的生成与执行正确率(用户真实场景)

数据表明:

华为芯片在生成成功率上表现最优

英伟达芯片在执行正确率(数值准确性)上表现最佳

国产多样化架构下仍存在进一步优化空间

整体结果验证了 KernelGen 在降低人工开发成本、提升跨芯片适配效率方面的核心价值。

110 个 Torch 算子的多芯片正确性与性能评测

为验证算子自动生成在真实工程场景下的可用性,评测选取了 110 个代表性 Torch API 算子,每个算子进行 5 轮迭代式自动生成,并在多芯片环境下完成完整对比测试。

评测覆盖华为、海光、天数、摩尔、Nvidia等平台,并在每个平台上分别对比:

芯片原生 Triton 编译器

FlagOS / FlagTree Triton 编译器

评测重点关注两项核心指标:

执行正确率:是否成功编译运行,并在多种输入 shape 下与 Torch reference 数值一致

执行正确率与性能汇总(110 个算子)

主要结论

FlagTree 在多芯片平台上整体执行正确率更高、稳定性更强

在海光、天数、摩尔及 Nvidia 等平台上,FlagTree 的通过率普遍高于原生 Triton,其中 Nvidia 平台达到 70%,为各组合中最高。

更高的「能跑通」比例,为后续性能优化保留了更大空间

对规模化算子生成而言,执行正确率是第一门槛,FlagTree 在这一维度上更具工程优势。

在执行正确率方面,KernelGen 对不同数据类型设定了工业界要求的严格且可量化的精度约束:整型与布尔类型要求完全一致(零误差),而浮点与复数类型则依据其数值表示能力设定明确的误差上限——例如 FP16/FP8 级别控制在 1e-3,BF16 控制在 1e-2 量级,FP32 与 Complex64 达到 1e-6 级别。通过这种按数据类型精细分级的精度标准,KernelGen 在追求极致性能的同时,确保算子替换与跨后端生成具备可验证、可复现的数值正确性。

注:评测基于110 个算子 × 多平台 × 双编译器的完整结果,算子级明细已随 KernelGen / FlagOS 项目提供。

不同大模型在算子生成任务中的差异(以华为 Ascend 为例)

在算子自动生成场景中,大模型并非只承担「代码补全」的角色,其对算子语义理解、shape 泛化以及边界条件处理能力,都会直接影响生成算子的可执行性与稳定性。

在统一使用 FlagTree Triton 编译器、并以华为 Ascend平台为目标硬件的条件下,评测对比了多种主流大模型在 110 个 Torch API 算子生成任务中的表现,重点统计其执行正确率(数值准确性通过)

从评测结果可以看到,不同大模型在算子生成任务中的能力差异较为明显,整体呈现出清晰的分层:

GPT-5 表现最优,在 110 个算子中有 65 个能够成功生成 Triton 实现并通过多种输入 shape 的一致性校验,显示出其在算子语义理解、控制流生成和边界条件处理方面的综合优势。

GLM-4.7 位于第二梯队,在部分复杂算子和 shape 泛化场景中仍存在不稳定情况,但已具备较强的自动算子生成能力。

Qwen3-Max 与 MiniMax-M2.1 在算子生成这一高约束场景下成功率相对较低,主要受限于对算子细粒度语义和底层 Triton 编程模式的理解能力。

这一结果表明,在算子自动生成这一高度结构化、强约束的任务中,模型本身的推理与结构化生成能力,仍然是影响最终工程效果的关键因素之一

专家知识注入与算子性能进化(英伟达)

在引入模型自反思与外部专家知识后,KernelGen 的算子性能持续提升:

算子执行正确率最高可达 75.5%

74.2% 算子加速比 >0.8

68.5% 算子加速比 >1.0

加速比中位数 1.04x,平均 1.07x

Baseline 为基于智能体自反思的迭代方法;IterOpt为面向优化目标的反思迭代方法;v1_KNWL在优化迭代中引入模型轨迹总结知识;v2_KNWL与v3_KNWL分别在不同轮次进一步融合外部专家种子知识

这表明,算子自动生成已从「能跑」迈向「能用、可优化」。

总结

Claude Code 的 30 分钟迁移案例,让行业看到了 AI 正在改变「写代码」的方式

而 KernelGen 与 FlagOS 所展示的,是另一层更工程化的进展:

在硬件高度碎片化的时代,通过算子自动生成、统一编译器与跨芯片生态协同,让算子开发不再成为AI系统落地的核心瓶颈。

这不是一次演示的胜负,而是一条正在被验证、并持续演进的系统软件路线。

本文转自:凤凰网科技

原文地址: https://tech.ifeng.com/c/8qWFm3KcmTP