摩尔线程发布 Torch-MUSA v2.7.0,增强 AI 模型训练与推理支持

IT之家 11 月 28 日消息,近日,摩尔线程正式发布 PyTorch 深度学习框架的 MUSA 扩展库 ——Torch-MUSA v2.7.0,新版本在功能集成、性能优化与硬件支持方面实现进一步突破。Torch-MUSA 在短短一个月内,连续完成 v2.5.0 和 v2.7.0 两次版本更新。

据介绍,自 v2.5.0 起,Torch-MUSA 版本号已与 PyTorch 主版本号保持同步,便于开发者进行版本识别与管理。新版本进一步集成 muSolver 与 muFFT 等计算加速库,显著提升复杂计算任务的执行效率;同时新增对统一内存设备(Unified Memory)的 UMM 支持,有效优化内存使用效率。

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

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

新增特性

算子扩展与性能优化

系统功能增强

IT之家从官方获悉,Torch-MUSA 将继续跟进 PyTorch 的版本更新,计划下一版本支持 PyTorch 2.9.0,并进一步优化性能与功能。

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