MUSA SDK v4.3.6 发布版本信息

下载链接

MTT S5000

• 下载链接
  • DEB包（如适用OS）：下载地址
• 支持操作系统：
  • Ubuntu 22.04.x LTS

发布说明

MUSA SDK v4.3.6 版本在 4.3.5 基础上进行功能迭代与问题修复，重点优化了编译器能力、运行时与通信性能，并持续提升深度学习相关算子能力，现已正式发布。

具体支持平台详见下方列表。

支持平台

MTT S5000支持平台

CPU	OS	内核版本	MUSA SDK包
Intel	Ubuntu 22.04.x LTS	5.15.0-105-generic	DEB

产品说明

摩尔线程 MUSA SDK 是一套完整的 GPU 并行计算开发环境，专为利用摩尔线程 GPU 加速程序而设计。通过 MUSA SDK，您能够轻松地利用 GPU 加速各种应用，涵盖企业数据中心、桌面、工作站以及云平台和超级计算机等多个领域。该套件包括了 GPU 加速库、调试和优化工具、C/C++编译器以及用于应用程序部署的运行时库等功能。作为一套完整的开发工具，MUSA SDK 包含了以下主要组件：

• MUSA Toolkits，包含底层编译器、MUSA 运行时库、Musify 工具以及MUSA-X 基础数学计算加速库
• muDNN，MT GPU 深度学习加速库
• MCCL，MT GPU 通信加速库
• MUSA SDK 依赖于摩尔线程 GPU 和摩尔线程 GPU 通用驱动程序
• MT GPU：MTT S5000, MTT S4000，MTT S3000，MTT S80, MTT X300 等
• 通用驱动软件

功能描述

• 支持 SIMT 架构并行编程模型
• 提供 mcc 编译器工具链
• MUSA 驱动与运行时库
• 摩尔线程 MUSA 数学库 (muBLAS, muFFT, muPP, muRAND)
• MCCL 通信库 (支持多 GPU 和多节点通信，优化 PCIe 和 MTLink 互联)
• muDNN 深度学习加速库

主要更新

功能更新（Feature Updates）

• 编译器与工具链增强（mcc）

  • 新增 sqmma commit_group / wait_group 软件能力支持，要求成对匹配使用
  • 增加 fp82f16、f162fp8、f2fp8 与 f162f32 的 burst 1/2/4 版本
  • 扩展 kernel struct-byval 优化策略，支持单成员满足 dword 对齐场景
  • 修复若干编译器实现问题

• 运行时与通信性能优化（Driver/Runtime & MCCL）

  • 优化 graph 后处理延迟和设备资源占用
  • 新增 muGraphExecChildGraphNodeSetParams API 支持
  • 优化 alltoall/sendrecv 双机带宽，以及单机 allreduce 带宽和延迟
  • 升级支持 mccl-tests ace 测试，并修复部分 reducescatter 与初始化问题

• 算子与框架支持增强（muDNN & Triton）

  • GroupedMatMul、BatchMatMul 增强分块能力与策略，并优化性能
  • Unary、Permute、BatchNorm、Pooling 等算子在类型支持和小 shape 场景下进一步优化
  • 优化编译流程以降低运行过程中的显存占用，并修复若干问题
  • Triton 侧 import triton 不再依赖 torch

产品组件说明

具体包名称与版本信息请以开发者官网对应版本下载页为准。