摩尔线程发布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
游乐网为非赢利性网站,所展示的游戏/软件/文章内容均来自于互联网或第三方用户上传分享,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系youleyoucom@outlook.com。
同类文章
北京推进智能网联新能源车险,支持L2-L4级别统一适配
3月29日,北京已在全国率先启动智能网联新能源汽车商业保险产品开发应用。新产品基本沿用现有的新能源商业车险体系,按照“总体稳定、部分优化”的原则,主要为消费者和汽车企业关心的特定智驾场景、软硬件损失
苹果今年将发布两款新iPhone应用,包含聊天机器人
预计苹果今年将发布两款新的 iPhone 应用,包括 Apple Business 应用和一款具备类似聊天机器人功能的 Siri 应用。借助 Apple Business 应用,使用全新 Apple
苹果聘请前谷歌副总裁分管AI产品营销
据 Axios 报道,苹果公司已聘请前谷歌副总裁 Lilian Rincon 担任人工智能产品营销副总裁。加入苹果之前, Rincon 曾任谷歌购物产品副总裁。在苹果, Rincon 将负责苹果所有
雷军销售心法:一句话卖出一辆车,金牌销售的秘诀
3月29日消息,谁能料到前段时间奥迪车主与雷军之间的那个打赌,竟然还有后续。这到底是咋回事?事情发生在3月25日,网友@单手开吉利 在雷军的微博评论区晒出了自己去年10月刚提的奥迪车,还当场立下一个
跨国工业巨头为何入局中国机器人市场?
近期,上百家来自全球跨国企业的CEO齐聚中国,参加中国发展高层论坛。第一财经记者了解到,不少全球高管利用此次访华之际,参观了中国先进制造企业,包括电动汽车、智能家电以及人形机器人厂商。芬兰电梯巨头通
- 日榜
- 周榜
- 月榜
相关攻略
2015-03-10 11:25
2015-03-10 11:05
2021-08-04 13:30
2015-03-10 11:22
2015-03-10 12:39
2022-05-16 18:57
2025-05-23 13:43
2025-05-23 14:01
热门教程
- 游戏攻略
- 安卓教程
- 苹果教程
- 电脑教程

