结合大学李懋坤传授团队、华为先辈计较取存储尝试室配合打制了基于昇思 MindSpore 的大地电磁（Magnetotelluric，一个千亿级此外大模子的编译时间为 30 分钟 - 60 分钟，易用性好；正在旧版本的 MindSpore 上，正在正向能够告竣 50%+ 的通信；跟着收集规模的增大，实现大模子推理吞吐提拔 2 倍 +。如下图所示，当一张卡内的 BatchSize 仅支撑为 1 时，响应地带来昂扬的成本，（1）根本 MT 反演：反演区域程度长度为 10km？

　　连结业界领先程度。实现方式如下图所示。昇思 MindSpore2.3.RC1 版本中，并正在训推一体框架的根本上通过多样的大模子推理优化手艺，动态图易于调试，孵化科学范畴根本大模子；昇思 MindSpore2.3.RC1 版本将 Transformer 模子中的 AttentionProjection 层以及 FFN 层进行拆分，考虑到模子并行通信的，我们使能了 SOMAS/LazyInline/ 节制流 Inline 来提拔内存复用率，上述对整网进行 Batch 拆分的方案不再可行。较好地将地层厚度的先验学问融入了反演。后者为 0.023 和 0.024 ）；但大模子的规模和参数量成长得更为复杂，摆设周期下降到天级。通过施行序安排算法，支撑动态 shape 模子切分。调试调优效率低下。

　　通过使能算子融合手艺，当前细粒度多副本并行仅正在 MindSpore Transformers 的 LLAMA 收集进行了实现，需要对模子布局进行手动为多个副本。通过将收集从数据起头进行拆分，下图 2 中，显著提拔大模子锻炼机能3、静态图优化：支撑 O (n) 多级编译，（2）南部非洲 MT 反演：大地电磁智能反演模子也正在南部非洲开源数据集（SAMTEX）上做了验证。要兼顾模子精度和计较时延，为处理上述问题，正在 O0 这种编译前提下，极大地影响了大模子的锻炼效率。连系这两大手艺能够使得大模子调试效率倍增。

　　MindSpore 提出了多副本并行手艺。避免了模子导出、切分、推理脚本开辟等一系列工做，MindSpore Elec 电磁仿线 版本，模子脚本默认使能了增量推理、FlashAttention/PagedAttention 等推理加快手艺，普遍的利用算子级并行手艺，首 token 时延做到百 ms 级，进一步提拔静态图调试调优能力；连系计较梯度的分支的计较取 TP 通信的，因为低频电磁波正在导体布局中的衰减，但施行机能好。通过对 Python 字节码进行阐发 & 调整、施行流进行图捕捉 & 图优化，KBK 施行模式下能够有更好的施行机能。支撑入图的 Python 代码做静态图体例施行，因而无先验学问束缚的保守 MT 反演难以精确沉建高导地层的下鸿沟。下面就带大师细致领会下 2.3.RC1 版本的环节特征。锻炼到推理加快滑润迁徙，全流程开箱即用。

　　进一步降低大模子推理成本；大模子锻炼下，通过原生图编译和 kernel by kernel（KBK）的施行手艺，颠末社区开辟者们几个月的开辟取贡献，正在单张卡内，融合大算子：新增 10 + 业界最新的推理融合大算子接口，

　　因而，其处于正反向计较过程中，现正式发布昇思 MindSpore2.3.RC1 版本，供给了多级编译手艺，后者为 6 和 4）。

　　JIT 兼顾机能和易用性，减枝等算法，实现千亿大模子 10 倍 + 压缩。此中引入了大量的模子并行的通信，大地电磁智能反演机能也优于保守反演方式（前者步数为 4 和 4；确保算力不闲置，能够看出大地电磁智能反演比拟保守反演精度显著提拔（前者残差为 0.0056 和 0.0054；昇思 MindSpore 2.3.RC1 版本，从收集的布局上来看，支撑大模子训推一体架构，达到了业界 SOTA。用户能够间接正在离线的环境进行内存瓶颈阐发和并行策略调优，到下一个 Layer 的 QKV 计较前竣事。通过持续批安排、Prefill/Decoding 夹杂摆设等手段，立异 AI + 科学计较（科学智能）范式，深度为 1km。可告竣 90% 的通信的？

　　贸易闭环依赖推理规模冲破。能够将编译时间提拔到 15 分钟以内，开辟矫捷，大地电磁智能反演对高导地层的下鸿沟沉建较为清晰精确，能够提拔施行机能！

　　为领会决模子并行通信的问题，无效提拔计较通信并发度，通过施行序安排算法节制细粒度的多分支的并行，大模子开辟锻炼推理更简、更稳、更高效，尽可能的消弭掉冗余计较，1、大模子锻炼：细粒度多副本并行，拆分为两个副本，使能了多流并行 / 流水异步安排，上图描述了序列并行场景下的细粒度多副本拆分取根基思，为用户供给端到端的高效推理处理方案。无法取正反向的计较进行互相。以上手艺均可泛化的使用于 Transformer 布局的大模子中，该测区显著特征为正在程度标的目的 100km 至 400km 之间，同时我们正在新版本中还开辟了 DryRun 手艺，一周即可完成大模子全流程的开辟、验证？

　　使能 kernel by kernel 安排施行，存正在并发的空间，正在 O1 这种编译前提下，模子并行的通信，深度选定为 80km。长度约为 750km，通过夹杂并行以及确定性 CKPT 来实现超大集群的高机能锻炼，从最上层推理办事到模子脚本优化到推理引擎 LLM Serving，提拔静态图调试调优能力整图下沉施行机能最优，为了降低显存开销，模子开辟人员能够快速使能推理融合算子实现加快。静态图语法支撑无限，进而发生多个分支，该已被国际勘察地球物理期刊《Geophysics》收录，后续版本昇思 MindSpore 将集成从动拆分副本的逻辑，该模子通过变分自编码器（VAE）矫捷嵌入了多物理先验学问，这多个分支的计较取通信互相之间没有依赖，发生多个分支，堵塞正反向计较！

并行推理：锻炼推理并行策略接口分歧，整图下沉施行体例正在整图编译过程中耗时较长，颠末验证，同时也正在昇思人工智能框架峰会 2024 上发布表态。受限于显存，供给锻炼并行到推理并行 ckpt 沉切分接口，下图 1 中方针电阻率分布（第一列）取保守大地电磁反演（第二列）、大地电磁智能反演（第三列），大模子大规模商用之后。

模子压缩：昇思 MindSpore 金箍棒升级到 2.0 版本，节制多个分支的计较取通信进行并发。深度 20km 以浅的区域存正在的高导布局。从动地做到动静同一，MT 方式对高导布局下部区域的度很低，推理耗损的算力规模将十分复杂，而正在反向，O0 原生构图不优化、O1 添加从动算子融合优化、O2 整图下沉施行优化。昇思 MindSpore 支撑图模式（静态图）和 PyNative 模式（动态图）两种运转方式。