摩尔线程发布Torch-MUSA 2.7.0：大幅增强AI模型训练与推理能力

11月28日最新消息，摩尔线程日前正式发布基于PyTorch深度学习框架的MUSA扩展库Torch-MUSA v2.7.0。新版本在功能集成、性能优化与硬件支持方面实现重要突破，值得一提的是，这已是Torch-MUSA在一个月内连续完成的第二次重大版本更新，此前v2.5.0版本也刚刚发布。

据了解，自v2.5.0版本开始，Torch-MUSA已实现与PyTorch主版本号同步，极大方便开发者的版本识别与管理。最新v2.7.0版本进一步集成了muSolver与muFFT等计算加速库，显著提升了复杂计算任务的执行效率；同时新增对统一内存设备的UMM支持，有效优化内存使用效率。

此外，新版本继续保持与最新MUSA SDK的兼容性，支持使用MUSA SDK 4.2.0至4.3.0及更高版本进行编译。目前Torch-MUSA专属支持的算子总数已突破1050个，系统在性能表现与运行稳定性方面均实现显著提升。

v2.7.0版本主要更新内容：

新增特性

全新集成muFFT与muSolver库集合，大幅扩展计算能力；

在面向边缘计算的SoC设备中支持统一内存管理，基于Arm架构的UMA设计，实现GPU与CPU共享同一物理内存空间，显著降低模型运行过程中的内存开销，具体包括：消除GPU端重复内存分配；减少主机与设备间的内存拷贝；GPU可直接访问由CPU分配器申请的内存空间。

算子扩展与性能优化

新增支持包括ilshift、irshift、replication_pad1d_bwd、angle、ctcLossTensor、ctcLossTensorBwd、logit、amin/amax/prod.dim_int、glu_bwd等多个算子；

新增基础稀疏矩阵(CSR)操作支持；

扩展量化算子支持范围；

修复torch.norm形状错误问题；

支持reduce_sum的uint8输入与int64输出；

C++扩展新增支持tensor.is_musa()方法；

修复空输入下argmax/argmin的异常行为；

优化var/std、pad、convolution3d、layer_norm等操作的执行效率。

系统功能增强

开放torch.musa.mccl.version()接口；

支持getCurrentMUSABlasHandle与getCurrentMUSABlasLtHandle；

优化FSDP2流水线并行策略，降低训练内存占用。

根据最新消息，Torch-MUSA将继续跟进PyTorch的版本更新，计划下一版本支持PyTorch 2.9.0，并进一步优化性能与功能。

Torch-MUSA开源地址：https://github.com/MooreThreads/torch_musa