Skip to content
This repository was archived by the owner on Jan 27, 2026. It is now read-only.

Add Torch 2.9 build variants#242

Merged
danieldk merged 17 commits intomainfrom
torch-2.9
Oct 3, 2025
Merged

Add Torch 2.9 build variants#242
danieldk merged 17 commits intomainfrom
torch-2.9

Conversation

@danieldk
Copy link
Member

@danieldk danieldk commented Oct 3, 2025

This change adds Torch 2.9 build variants to kernel-builder. Until the final release of Torch 2.9, the Torch 2.7 variants are also still built. The Torch 2.9 support is based on RC5, typically there are no ABI-incompatible changes at this point, but there is no guarantee.

@danieldk danieldk marked this pull request as ready for review October 3, 2025 11:53
Copy link
Contributor

@MekkCyber MekkCyber left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Sounds great !

Comment on lines +158 to +159
# sm_101 has been renamed to sm_110 in CUDA 13.
"7.5;8.0;8.6;8.9;9.0;10.0;11.0;12.0";
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

do torch kernels not compile using cuda 13.0 on sm < 7.5 ? if so why are we keeping 7.0 in CUDA_DEFAULT_KERNEL_ARCHS

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Missed that, removed from CUDA_DEFAULT_KERNEL_ARCHS now 👍.

versions.nix Outdated
Comment on lines +141 to +145
torchVersion = "2.8";
rocmVersion = "7.0.1";
cxx11Abi = true;
systems = [ "x86_64-linux" ];
bundleBuild = false;
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

we can build torch 2.8 for 7.0.1 but not 2.9 ?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I haven't tested it yet. This build variant was a one-off for Abdennacer to test with ROCm 7. I should probably remove it again, since it adds extra build time.

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Removed the ROCm 7 variant. We can add it once upstream PyTorch starts supporting it.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

sounds good

@danieldk danieldk merged commit ed5722d into main Oct 3, 2025
27 checks passed
@danieldk danieldk deleted the torch-2.9 branch October 3, 2025 14:25
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants