Add support for CUDA 13.0+ in ptx_get_version #8925
Closed
+46
−0
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.
This commit adds support for CUDA 13.x versions in the ptx_get_version function. Previously, CUDA 13.0 would raise a RuntimeError saying 'Triton only support CUDA 10.0 or higher', even though 13.0 > 10.0.
The fix maps CUDA 13.0+ to PTX version 8.9+, following the existing pattern for CUDA 10.x (PTX 6.3+), 11.x (PTX 7.0+), and 12.x (PTX 8.0+).
Note: A separate fix is needed for PTX version mismatch where the .version directive should match the actual PTX features used by LLVM (capped at 8.3). This may need to be applied in C++ code or Python wrapper depending on repository structure.
Fixes Windows build issues with CUDA 13.0.
New contributor declaration
I am not making a trivial change, such as fixing a typo in a comment.
I have written a PR description following these
rules.
I have run
pre-commit run --from-ref origin/main --to-ref HEAD.Select one of the following.
/testforlittests/unittestfor C++ tests/python/testfor end-to-end testsFILL THIS IN.Select one of the following.
littests.littests I have added follow these best practices,including the "tests should be minimal" section. (Usually running Python code
and using the instructions it generates is not minimal.)