Prhub

#29236 [MUSA][diffusion] Bump torchada version to 0.1.68

原始 PR 作者 yeahdongcn 合并时间 2026-06-26 07:53 文件变更 3 提交数 1 评论 1 代码增减 +3 / -3

执行摘要

MUSA 平台 torchada 依赖升级

解决 MUSA diffusion CI 问题,PR body 引用了上游 torchada#85 作为修复依据。

属于常规依赖维护,无需精读。

讨论亮点

无 review 评论或讨论。

实现拆解

  1. 3rdparty/amd/wheel/sglang/pyproject.tomlsrt_musa extra 中将 torchada>=0.1.59 改为 torchada>=0.1.68
  2. python/pyproject_other.tomlsrt_musa extra 中做相同修改。
  3. sgl-kernel/pyproject_musa.toml[build-system] requires 中做相同修改。
    所有修改均为版本号最低要求上调,保持 >= 约束。
文件 模块 状态 重要度
3rdparty/amd/wheel/sglang/pyproject.toml AMD 构建 modified 2.9
python/pyproject_other.toml Python 核心 modified 2.9
sgl-kernel/pyproject_musa.toml Kernel 库 modified 2.9

关键源码片段

3rdparty/amd/wheel/sglang/pyproject.toml configuration

定义 MUSA 平台 SRT 运行时依赖,升级 torchada 版本下限

# 3rdparty/amd/wheel/sglang/pyproject.toml
# MUSA 平台 SRT 依赖组,torchada 版本下限从 0.1.59 提升到 0.1.68
srt_musa = [
  "sglang[runtime_common]",
  "torch",
  "torch_musa",
  "torchada>=0.1.68", # 升级点
  "mthreads-ml-py",
  "mate>=0.2.0",
  "deep-gemm>=0.1.3",
  "flash_attn_3>=0.1.4",
  "numpy<2.0",
]
python/pyproject_other.toml configuration

覆盖 MUSA 平台 SRT 依赖,需同步升级

# python/pyproject_other.toml
# MUSA 平台 SRT 依赖组,与 amd 目录下定义同步升级
srt_musa = [
  "deep-gemm>=0.1.3",
  "flash_attn_3>=0.1.4",
  "mate>=0.2.0",
  "mthreads-ml-py",
  "numpy<2.0",
  "sglang[runtime_common]",
  "torch",
  "torch_musa",
  "torchada>=0.1.68", # 升级点
]
sgl-kernel/pyproject_musa.toml configuration

MUSA kernel 构建依赖,升级 torchada 确保编译环境一致

# sgl-kernel/pyproject_musa.toml
# 构建时系统依赖,torchada 作为构建依赖需与运行时版本下限对齐
[build-system]
requires = [
  "setuptools>=75.0",
  "scikit-build-core>=0.10",
  "torch",
  "torchada>=0.1.68", # 升级点
  "wheel",
]
build-backend = "setuptools.build_meta"

评论区精华

没有提炼出高价值讨论线程

当前评论区没有形成足够清晰的争议点或结论,后续有更多讨论时会体现在这里。

风险与影响

风险极低。版本升级为非破坏性更新(>= 约束),不影响已有安装;但若新版本引入兼容性问题,可能影响 MUSA 平台 SRT 和 kernel 构建。

影响范围仅限于 MThreads MUSA 平台用户。升级后需重新构建 sgl-kernel 和安装 sglang,对已有环境无破坏性影响。

依赖升级 平台特定

关联 Issue

未识别关联 Issue

当前没有检测到明确关联的 Issue 链接,后续同步到相关引用后会出现在这里。

完整报告

参与讨论