chore(deps): bump torch to 2.14 and unify CUDA wheels on cu130 - #1647
Merged
Merged
Conversation
This was referenced Sep 26, 2026
- pyproject: collapse the platform-split torch pins into torch>=2.9,<2.15 - uv.sources: route linux/win torch to the cu130 index (cu128 tops out at torch 2.11); drop the pytorch-cu128 index - uv.lock: torch 2.8.0/2.8.0+cu128/2.9.0+cu130 -> 2.14.0/2.14.0+cu130, triton 3.8.0, nvidia deps cu12 -> cu13 - rocm: torch==2.14.0 + triton-rocm==3.8.0 (torch 2.14 pairs with triton~=3.8); relax the setuptools<70 typo-era pin which conflicts with torch 2.14+rocm7.2 (requires setuptools>=77); regenerate uv.rocm.lock - update the torch CUDA source contract tests and cu128 doc references Measured on Apple Silicon (M5 Max, g1_walk_flat, mujoco): FlashSAC end-to-end 16.1k -> 21.7k steps/s (+35%), learner 213ms -> 147ms/iter; FastSAC unaffected (GEMM-bound).
TATP-233
force-pushed
the
chore/torch-2.14-cu130
branch
from
September 26, 2026 17:59
d6e4a9e to
da1514e
Compare
Merged
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
torch>=2.9,<2.15;tool.uv.sources统一走r2-cu130(cu128 索引最高只有 torch 2.11),删除pytorch-cu128索引uv.lock:torch 2.8.0 / 2.8.0+cu128 / 2.9.0+cu130 → 2.14.0(darwin/win 之外 PyPI)+ 2.14.0+cu130(linux/win),triton 3.8.0,nvidia 依赖 cu12 → cu13pyproject.rocm.tomltorch==2.14.0 + triton-rocm==3.8.0(torch 2.14 官方配套 triton~=3.8);放宽setuptools<70遗留 pin(与 torch 2.14+rocm7.2 的 setuptools>=77 冲突);重新生成uv.rocm.locktests/scripts/test_torch_cuda_source.py)与文档中 cu128 表述(中英 installation/contributing、CONTRIBUTING.md)scripts/tools/setup_isaacsim_env.sh的 cu128 是 IsaacSim 独立 pin 的环境,不在本次范围Linked Work
Validation
make test-allpassed on the final local head before this PR was created or updatedCommands actually run:
make test-all(slow 套件)本地未跑(M 系列 Mac 无 CUDA,慢速训练矩阵不适用);CUDA cu130 / ROCm rocm7.2 实机验证依赖 CI 与 Linux 机器。
Remote CI route:
main: https://github.com/Motphys/UniLab/actions/runs/36234368359 — 全绿(ruff-lint / ruff-format / mypy / pyright / test (ubuntu-slim) / benchmark-smoke;docs 构建 pass)。Impact
Artifacts
Rebase update
origin/main@d72f7f07,新 head 为da1514e6,并已 force-with-lease 推送。