Skip to content

[Build][NVIDIA] Update public Rubin dependencies and MSA compatibility #56545

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
ywang96 merged 6 commits into from
Sep 15, 2026
Merged

[Build][NVIDIA] Update public Rubin dependencies and MSA compatibility #56545

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
f5aa71d
[Build][NVIDIA] Update public Rubin dependencies and MSA compatibility
wangshangsam Sep 12, 2026
21116fb
Merge branch 'main' into wangshangsam/rubin-public-dependencies-msa-c…
wangshangsam Sep 12, 2026
89c9325
[Build][NVIDIA] Advance compatible Rubin PyTorch nightlies
wangshangsam Sep 12, 2026
f2015ad
Merge branch 'main' into wangshangsam/rubin-public-dependencies-msa-c…
wangshangsam Sep 12, 2026
f50bcda
Merge branch 'main' into wangshangsam/rubin-public-dependencies-msa-c…
wangshangsam Sep 14, 2026
e009068
Merge branch 'main' into wangshangsam/rubin-public-dependencies-msa-c…
wangshangsam Sep 14, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion cmake/external_projects/fmha_sm100.cmake
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ else()
FetchContent_Declare(
fmha_sm100
GIT_REPOSITORY https://github.com/vllm-project/MSA.git
GIT_TAG 087c161814d4d9c735b46c21212a09e5f8eb92fa
GIT_TAG f355c37eb4e1413f21ee2ad8bbad25079e6bef9d
GIT_PROGRESS TRUE
CONFIGURE_COMMAND ""
BUILD_COMMAND ""
Expand Down
2 changes: 1 addition & 1 deletion docker/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -640,7 +640,7 @@ RUN --mount=type=cache,target=/opt/uv/cache \
python3 use_existing_torch.py --prefix; \
fi \
&& if [ "${INSTALL_RUBIN_PRERELEASE}" = "true" ]; then \
sed -Ei '/^[[:space:]]*(flashinfer-python|nvidia-cutlass-dsl\[cu13\])[[:space:]]*==/Id' \
sed -Ei '/^[[:space:]]*(flashinfer-python|nvidia-cutlass-dsl\[cu13\]|quack-kernels)[[:space:]]*==/Id' \
requirements/cuda.txt; \
fi

Expand Down
22 changes: 16 additions & 6 deletions requirements/rubin-prerelease.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,15 +2,25 @@
# Installed when INSTALL_RUBIN_PRERELEASE=true (see docker/Dockerfile).
# TODO(rubin-bringup): delete once Rubin APIs ship in regular PyTorch wheels
# (vllm-project/vllm#49735).
--pre
--extra-index-url https://download.pytorch.org/whl/nightly/cu134
--extra-index-url https://flashinfer.ai/whl/cu134
--extra-index-url https://flashinfer.ai/whl

torch==2.15.0.dev20260903+cu134
torchvision==0.30.0.dev20260903+cu134
torchaudio==2.11.0.dev20260903+cu134
# Match each architecture's torch pin to the latest torchvision wheel.
torch==2.15.0.dev20260909+cu134; platform_machine == "x86_64"
torch==2.15.0.dev20260908+cu134; platform_machine == "aarch64"
torchvision==0.30.0.dev20260909+cu134
torchaudio==2.11.0.dev20260911+cu134

# docker/Dockerfile strips these two from requirements/cuda.txt whenever
# docker/Dockerfile strips these packages from requirements/cuda.txt whenever
# INSTALL_RUBIN_PRERELEASE=true, so they must be pinned here instead to
# still get installed.
nvidia-cutlass-dsl[cu13]==4.8.0.dev0
flashinfer-python==0.6.18
flashinfer-python==0.6.18.post1
quack-kernels==0.6.5

cuda-python==13.4.1
cuda-bindings==13.4.1
flashinfer-jit-cache==0.6.18.post1+cu134
flashinfer-cubin==0.6.18.post1
nixl==1.4.1
Loading