智东西

作家 | 程茜

裁剪 | 云鹏

智东西8月4日音信，本日，面壁智能集结开源社区OpenBMB开源环球首个救援Stencil（模板诡计）自动策动、自动部署的AI优化系统ForgeStencil。

字据官方长远，ForgeStencil一周内完成了超100个真确工业和科学诡计软件的自动优化，这突出于每年参加约4-8个工程师，等效研发参加为300万~1000万元。

一般而言，众人每东谈主每年能精调的运用软件通常不越过20个，ForgeStencil将单个运用所需的优化时长压缩到了小时级别，研发浑沌普及约1–2个数目级，研发遵循普及约100倍，且这一遵循不错跟着算力范围加大并行放大。

当代工业与前沿科研的紧闭高度依赖高性能诡计（HPC）软件，但这类软件底层弥远存在一种算力密集的中枢诡计步地Stencil，其中枢是对限定网格上每个点偏握固定邻域践诺相同局部运算。

Stencil属于访存密集型诡计，性能受内存带宽拘谨，通常会占据运用绝大部分耗时，其优化水平也平直决定了工业与科研仿真软件的运转遵循。此前，Stencil优化高度依赖稀缺的HPC众人，这也导致其难以范围化，ForgeStencil即是为解决这一辛苦而生。

ForgeStencil基于面壁智能提议的全新软件开发想路Forge Engineering打造，是其继5月AI制造AI死心ForgeTrain之后取得的又一技能紧闭。

开源地址：https://github.com/OpenBMB/ForgeStencil

一、两大Agent协同调优，无需东谈主工介入

ForgeStencil初度达成了从提议优化目标到运用无缝部署的全自动闭环。

该设施中，东谈主类提供待优化的运用源码，之后ForgeStencil接办从自动分析真确运用、定位热门函数、铸造Kernel，到完成算子替换、正确性考据与集成回原运用，中间不需要东谈主类众人介入优化决策。

ForgeStencil系统由Kernel Agent（表面所在）与App Agent（工程落地）构成。

Kernel Agent专注于自主策动并合成高性能Kernel，不错针对主流Stencil类型、多种Shape与精度条款，自主构建专用算子矩阵，将Stencil算子的数学抒发写成迫临硬件物理极限的代码。

官方长远，与现在市面上Halide、Devito、EBISU、DRStencil、FlashFFTStencil等开源算法比较，ForgeStencil在同等硬件下的算子测试中进展如下：

同精度（fp32）对比下，ForgeStencil得到几何平均2.35倍的加快比；混杂精度（fp16）读写时，ForgeStencil拿下畸形的1.95倍提速；在更难的变系数Stencil沿路Shape上，比较最优基线ForgeStencil取得几何平均下1.34倍的加快。

传统的静态算子库通常无法平直融入真确软件，因为践诺的科学诡权衡法不是法式Stencil轮回，不同的运用会对应不同的输入输出模样、诡计进程等。

基于此，App Agent不错为每个运用单独打造有野心，其会定位性能热门、确立运用自己的GPU基线、铸造候选优化、用要领自带的校验与计时考据，再集成回原运用。

在框架层，App Agent会寻找算子交融契机，并把待优化的Stencil算子索取出来。Kernel Agent厚爱优化具体的算子，其在之前确立的算子矩阵被手脚后续开发的学问库，最终App Agent会使用运用自带的测例作念端到端评测，以确保优化能面向真确场景。

基于此，ForgeStencil通过Kernel Agent与App Agent的协同，达成从自动生成代码向自动策动优化、从局部算子提速向真确运用部署的进化。

二、研发遵循普及百倍，随算力范围扩大并行提速

此前，HPC众人要优化一个真确的工业软件或科学诡计运用，需要进行性能瓶颈分析、软件架构清爽等，通盘进程历时数日到数周，因此每东谈主每年能精调的运用软件通常不越过20个。

基于大模子，这一策动限定的遵循普及了。

面壁智能的数据显现，ForgeStencil用时1周，就完成了超100个真确运用软件的自动优化，单个运用所需的优化时长在小时级别，研发浑沌普及约1–2个数目级，研发遵循普及约100倍，且不错跟着算力范围的加大并行放大。

这意味着Stencil算子的优化偏握所决定的工业软件和科学诡计运用优化初度达成范围化的可能性。

现在，ForgeStencil已在一批主流科学软件与基准上完成自动优化，并在运用自带的GPU达成之上达成端到端加快，其中42%平直对应真确工业分娩软件厂家。

具体的运用包括：

Hypre（LLNL结构化多重网格求解器库，加快3.86倍）：环球工业仿真最庸俗依赖的分娩级数值库之一，ForgeStencil措置了高难度的红黑GS光滑子访存归拢优化；

Minisweep（Sn闹翻纵标输运，核反映堆中子学，加快5.78倍）：联想核反映堆中枢诡计的必需品，ForgeStencil重构了极其复杂的KBA波前扫掠结构；

gprMax/FDTD（Yee网格Maxwell电磁仿真，加快2.47倍）：雷达和芯片电磁仿真实主干，ForgeStencil完成了电磁装备与探地雷达的中枢Stencil核替换；

RTM逆时偏移（油气地震成像，加快1.81倍）；

TOTAL E&P minimod（谈达尔油气地震mini-app，加快1.22倍）：石油勘察的主力算法；

QuantLib债券订价（量化金融，加快1.82倍）：金融机构所使用的订价库；

FHd非笛卡尔MRI重建（加快2.45倍）：医学影像软件；

DBT数字乳腺断层成像反投影（加快1.63倍）：医疗拓荒中的重建与成像负载。

与此同期，与东谈主类众人比较，ForgeStencil还有一大上风即是其领有大齐并行的Agent分享学问库，不错分享教育。

结语：智能优化Stencil算子，小时级完成深度优化

玄虚来看，ForgeStencil不错普及工业软件和科学诡计运用的研发遵循。短期内，存量软件的自动优化不错带来平直的收益，已有的以Stencil为热门的工业或科学软件，能在小时级生成接近硬件极限性能的算子达成，这将进一步从简算力、压缩研发时候。

在更永恒的角度欧洲杯体育，Stencil算子背后还包括CAE仿真、地震成像、电磁与流体诡计等工业软件，翌日这些软件性能优化达成自动化，将进一步加快国内制造业的发展。