摩尔线程驱动更新:强化AI模型训练与推理支持
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版本主要包含以下更新内容:
新增特性
集成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等多个核心算子;
新增基础Sparse (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
热门专题
最新APP
热门推荐
苹果折叠屏手机 iPhone Fold 最新渲染图曝光:摄像头凸起优化,设计更显精致 有关苹果公司首款折叠屏 iPhone 的传闻持续受到关注。4月5日,知名爆料者 Majin Bu 在社交平台X上再度分享了一组据称是 iPhone Fold 的高清渲染图,从多角度揭示了这款备受期待设备可能的外观设
通用性首选:官府无垢队阵容深度解析 在当前版本中,若要挑选一套兼具强度与广泛适用性的阵容,以官府流派【长孙无垢】为核心的搭配方案无疑是热门之选。这套经典组合通常由长孙无垢(官府)、李一桐、李善德、李光弼,以及关羽或平安组成。其核心战斗逻辑清晰且高效:一方面,依靠长孙无垢与李光弼的技能联动,通过对目标
洛克王国全精灵隐藏进化条件完整攻略大全 在《洛克王国》丰富多彩的冒险世界中,除了常规的等级进化,众多精灵还埋藏着独特的“隐藏进化”路径。这些特殊的进化条件,往往是解锁精灵终极形态、完成图鉴收集的关键所在。与普通进化方式不同,隐藏进化需要触发特定的环境、时间、道具或任务条件,充满了探索与解密的乐趣。你
燕云十六声石震关卡怎么过?高效通关技巧与实战攻略详解 掌握核心机制:石震关卡难点全解析 石震关卡的核心挑战在于敌人配置:不仅数量密集,且拥有高额血量和攻击力。这些敌人并非随机分布,而是依据特定区域、巡逻路线及攻击逻辑进行部署。提前掌握不同敌人的攻击前摇、技能范围与仇恨机制,是制定有效战术的前提,真正
英雄联盟手游安妮符文终极指南:爆发流核心配置与实战策略 在英雄联盟手游的对局中,黑暗之女安妮以其强大的瞬间爆发与控制能力,始终是中单位置的热门选择。虽然操作看似简单易懂,但想要真正掌握这位火焰法师的精髓,打出毁天灭地的效果,一套科学高效的符文搭配是不可或缺的基石。正确的符文选择,能让她从温顺的火苗化





