8000 [CD] [aarch64] Add CUDA 12.6 and 12.8 to build matrix, remove 12.9 build by tinglvv · Pull Request #162364 · pytorch/pytorch · GitHub
[go: up one dir, main page]
Skip to content

[CD] [aarch64] Add CUDA 12.6 and 12.8 to build matrix, remove 12.9 build #162364

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

Jump to bottom
Closed
wants to merge 4 commits into from
Closed

[CD] [aarch64] Add CUDA 12.6 and 12.8 to build matrix, remove 12.9 build #162364

wants to merge 4 commits into from

Conversation

@tinglvv
Copy link
Collaborator
@tinglvv tinglvv commented Sep 8, 2025 •
edited
Loading

#159779

Add the full CUDA support matrix to sbsa build (12.6, 12.8)
Same arch support as x86 build
Remove 12.9 sbsa build

cc @ptrblck @nWEIdia @atalman @malfet

@tinglvv tinglvv requested a review from a team as a code owner September 8, 2025 05:41
@pytorch-bot
Copy link
pytorch-bot bot commented Sep 8, 2025 •
edited
Loading

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/162364

Note: Links to docs will display an error until the docs builds have been completed.

❌ 8 New Failures, 1 Unrelated Failure

As of commit fb0379c with merge base c9ac8c2 (image):

NEW FAILURES - The following jobs have failed:

BROKEN TRUNK - The following job failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@pytorch-bot pytorch-bot bot added the release notes: releng release notes category label Sep 8, 2025
@tinglvv tinglvv added the ciflow/binaries_wheel Trigger binary build and upload jobs for wheel on the PR label Sep 8, 2025
atalman
atalman approved these changes Sep 8, 2025
View reviewed changes
Copy link
Contributor
@atalman atalman left a comment

Choose a reason for hiding this comment

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

Thank you.

@tinglvv
Copy link
Collaborator Author
tinglvv commented Sep 8, 2025

@pytorchbot merge -i

@pytorch-bot pytorch-bot bot added the ciflow/trunk Trigger trunk jobs on your pull request label Sep 8, 2025
@pytorchmergebot
Copy link
Collaborator
pytorchmergebot commented Sep 8, 2025

Merge started

Your change will be merged while ignoring the following 8 checks: linux-s390x-binary-manywheel / manywheel-py3_11-cpu-s390x-build / build, linux-s390x-binary-manywheel / manywheel-py3_10-cpu-s390x-build / build, linux-s390x-binary-manywheel / manywheel-py3_13-cpu-s390x-build / build, linux-s390x-binary-manywheel / manywheel-py3_14t-cpu-s390x-build / build, linux-s390x-binary-manywheel / manywheel-py3_13t-cpu-s390x-build / build, linux-s390x-binary-manywheel / manywheel-py3_14-cpu-s390x-build / build, linux-s390x-binary-manywheel / manywheel-py3_12-cpu-s390x-build / build, linux-binary-manywheel / manywheel-py3_14-rocm6_4-test

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging
Check the merge workflow status
here

@tinglvv
Copy link
Collaborator Author
tinglvv commented Sep 8, 2025

Should remove sm50-70 for sbsa, since the sbsa support only started from sm80 (Ampere).

@huydhn huydhn mentioned this pull request Sep 10, 2025
Closed
pytorchmergebot pushed a commit that referenced this pull request Sep 13, 2025
@huydhn @pytorchmergebot
Build vLLM aarch64 nightly wheels (#162664)
66133b1 
PyTorch has published its aarch64 nightly wheels for all CUDA version after #162364
Pull Request resolved: #162664
Approved by: https://github.com/atalman
huydhn added a commit to huydhn/pytorch that referenced this pull request Sep 17, 2025
@huydhn
Another tweak from pytorch#162364
cc72948 
Signed-off-by: Huy Do <huydhn@gmail.com>
markc-614 pushed a commit to markc-614/pytorch that referenced this pull request Sep 17, 2025
@tinglvv @markc-614
[CD] [aarch64] Add CUDA 12.6 and 12.8 to build matrix, remove 12.9 bu…
371c85e 
…ild (pytorch#162364)

pytorch#159779

Add the full CUDA support matrix to sbsa build (12.6, 12.8)
Same arch support as x86 build
Remove 12.9 sbsa build

Pull Request resolved: pytorch#162364
Approved by: https://github.com/atalman
< 8000 /div>
markc-614 pushed a commit to markc-614/pytorch that referenced this pull request Sep 17, 2025
@huydhn @markc-614
Build vLLM aarch64 nightly wheels (pytorch#162664)
28fe36e 
PyTorch has published its aarch64 nightly wheels for all CUDA version after pytorch#162364
Pull Request resolved: pytorch#162664
Approved by: https://github.com/atalman
mansiag05 pushed a commit to mansiag05/pytorch that referenced this pull request Sep 22, 2025
@tinglvv @mansiag05
[CD] [aarch64] Add CUDA 12.6 and 12.8 to build matrix, remove 12.9 bu…
4cc00fe 
…ild (pytorch#162364)

pytorch#159779

Add the full CUDA support matrix to sbsa build (12.6, 12.8)
Same arch support as x86 build
Remove 12.9 sbsa build

Pull Request resolved: pytorch#162364
Approved by: https://github.com/atalman
mansiag05 pushed a commit to mansiag05/pytorch that referenced this pull request Sep 22, 2025
@huydhn @mansiag05
Build vLLM aarch64 nightly wheels (pytorch#162664)
290b5d3 
PyTorch has published its aarch64 nightly wheels for all CUDA version after pytorch#162364
Pull Request resolved: pytorch#162664
Approved by: https://github.com/atalman
cleonard530 pushed a commit to cleonard530/pytorch that referenced this pull request Sep 22, 2025
@tinglvv @cleonard530
[CD] [aarch64] Add CUDA 12.6 and 12.8 to build matrix, remove 12.9 bu…
2f35b97 
…ild (pytorch#162364)

pytorch#159779

Add the full CUDA support matrix to sbsa build (12.6, 12.8)
Same arch support as x86 build
Remove 12.9 sbsa build

Pull Request resolved: pytorch#162364
Approved by: https://github.com/atalman
cleonard530 pushed a commit to cleonard530/pytorch that referenced this pull request Sep 22, 2025
@huydhn @cleonard530
Build vLLM aarch64 nightly wheels (pytorch#162664)
e697101 
PyTorch has published its aarch64 nightly wheels for all CUDA version after pytorch#162364
Pull Request resolved: pytorch#162664
Approved by: https://github.com/atalman
dsashidh pushed a commit to dsashidh/pytorch that referenced this pull request Sep 26, 2025
@tinglvv @dsashidh
[CD] [aarch64] Add CUDA 12.6 and 12.8 to build matrix, remove 12.9 bu…
ce724b2 
…ild (pytorch#162364)

pytorch#159779

Add the full CUDA support matrix to sbsa build (12.6, 12.8)
Same arch support as x86 build
Remove 12.9 sbsa build

Pull Request resolved: pytorch#162364
Approved by: https://github.com/atalman
dsashidh pushed a commit to dsashidh/pytorch that referenced this pull request Sep 26, 2025
@huydhn @dsashidh
Build vLLM aarch64 nightly wheels (pytorch#162664)
7cf1e4f 
PyTorch has published its aarch64 nightly wheels for all CUDA version after pytorch#162364
Pull Request resolved: pytorch#162664
Approved by: https://github.com/atalman
@atalman atalman mentioned this pull request Oct 8, 2025
Closed
pytorchmergebot pushed a commit that referenced this pull request Oct 9, 2025
@atalman @pytorchmergebot
CUDA aarch64 12.6 and 12.8 builds fix triton constraints (#165013)
81dbeb0 
Since we have introduced CUDA aarch64 builds for all cuda versions we need to remove this constraint.
This was missed by #162364

Proper constraint on triton should be:
```
Requires-Dist: triton==3.5.0; platform_system == "Linux"
```

not:
```
Requires-Dist: triton==3.5.0; platform_system == "Linux" and platform_machine == "x86_64"
```

Pull Request resolved: #165013
Approved by: https://github.com/Camyll, https://github.com/nWEIdia, https://github.com/tinglvv
pytorchbot pushed a commit that referenced this pull request Oct 9, 2025
@atalman @pytorchbot
CUDA aarch64 12.6 and 12.8 builds fix triton constraints (#165013)
79961ae 
Since we have introduced CUDA aarch64 builds for all cuda versions we need to remove this constraint.
This was missed by #162364

Proper const
8000
raint on triton should be:
```
Requires-Dist: triton==3.5.0; platform_system == "Linux"
```

not:
```
Requires-Dist: triton==3.5.0; platform_system == "Linux" and platform_machine == "x86_64"
```

Pull Request resolved: #165013
Approved by: https://github.com/Camyll, https://github.com/nWEIdia, https://github.com/tinglvv

(cherry picked from commit 81dbeb0)
@pytorchbot pytorchbot mentioned this pull request Oct 9, 2025
Merged
atalman added a commit that referenced this pull request Oct 9, 2025
@pytorchbot @atalman
CUDA aarch64 12.6 and 12.8 builds fix triton constraints (#165022)
0fabc3b 
CUDA aarch64 12.6 and 12.8 builds fix triton constraints (#165013)

Since we have introduced CUDA aarch64 builds for all cuda versions we need to remove this constraint.
This was missed by #162364

Proper constraint on triton should be:
```
Requires-Dist: triton==3.5.0; platform_system == "Linux"
```

not:
```
Requires-Dist: triton==3.5.0; platform_system == "Linux" and platform_machine == "x86_64"
```

Pull Request resolved: #165013
Approved by: https://github.com/Camyll, https://github.com/nWEIdia, https://github.com/tinglvv

(cherry picked from commit 81dbeb0)

Co-authored-by: atalman <atalman@fb.com>
pytorchmergebot pushed a commit to huydhn/pytorch that referenced this pull request Oct 11, 2025
@huydhn @pytorchmergebot
Another tweak from pytorch#162364
b1f78ae 
Signed-off-by: Huy Do <huydhn@gmail.com>
Chao1Han pushed a commit to Chao1Han/pytorch that referenced this pull request Oct 21, 2025
@atalman @Chao1Han
CUDA aarch64 12.6 and 12.8 builds fix triton constraints (pytorch#165013
7de9579 
)

Since we have introduced CUDA aarch64 builds for all cuda versions we need to remove this constraint.
This was missed by pytorch#162364

Proper constraint on triton should be:
```
Requires-Dist: triton==3.5.0; platform_system == "Linux"
```

not:
```
Requires-Dist: triton==3.5.0; platform_system == "Linux" and platform_machine == "x86_64"
```

Pull Request resolved: pytorch#165013
Approved by: https://github.com/Camyll, https://github.com/nWEIdia, https://github.com/tinglvv
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@atalman atalman atalman approved these changes

Assignees

No one assigned

Labels

ciflow/binaries_wheel Trigger binary build and upload jobs for wheel on the PR ciflow/trunk Trigger trunk jobs on your pull request Merged open source release notes: releng release notes category

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

4 participants

@tinglvv @pytorchmergebot @atalman @pytorchbot
0