8000 Triton aarch64 and triton sbsa · Issue #147857 · pytorch/pytorch · GitHub
[go: up one dir, main page]
Skip to content

Triton aarch64 and triton sbsa #147857

8000
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
johnnynunez opened this issue Feb 25, 2025 · 1 comment
Closed

Triton aarch64 and triton sbsa #147857

johnnynunez opened this issue Feb 25, 2025 · 1 comment
Labels
oncall: releng In support of CI and Release Engineering
Milestone
2.7.0

Comments

@johnnynunez
Copy link
Contributor
johnnynunez commented Feb 25, 2025 •
edited by pytorch-bot bot
Loading

🚀 The feature, motivation and pitch

Why?

runners:
Github: https://github.blog/changelog/2025-01-16-linux-arm64-hosted-runners-now-available-for-free-in-public-repositories-public-preview/
windows arm q2 2025: github/roadmap#1098

Devices:
GH200 and future devices: Digits, jetson thor, cuda arm laptops are coming
Nvidia is merging SBSA and ARM64 together

Alternatives

alls is in x86_64 https://download.pytorch.org/whl/nightly/pytorch-triton/
alternatives, at this moment for jetson: https://github.com/dusty-nv/jetson-containers/tree/master/packages/ml/triton

Additional context

useful for use in github arm runners like flash attention repository for GH200

cc @ezyang @gchanan @zou3519 @kadeng @msaroufim @malfet @seemethere @snadampal @milpuz01 @bertmaher @int3 @davidberard98 @nmacchioni @chenyang78 @embg @peterbell10 @aakhundov
@malfet malfet added this to the 2.7.0 milestone Mar 3, 2025
@malfet malfet added module: build Build system issues module: arm Related to ARM architectures builds of PyTorch. Includes Apple M1 upstream triton Upstream Triton Issue high priority labels Mar 3, 2025
@malfet malfet added oncall: releng In support of CI and Release Engineering and removed high priority triage review module: build Build system issues module: arm Related to ARM architectures builds of PyTorch. Includes Apple M1 upstream triton Upstream Triton Issue labels Mar 3, 2025
@malfet
Copy link
Contributor
malfet commented Mar 3, 2025

Should it be closed as duplicate of #130558

@atalman atalman mentioned this issue Apr 3, 2025
Closed
65 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
oncall: releng In support of CI and Release Engineering
Projects
PyTorch OSS Dev Infra
Archived in project
Milestone
2.7.0
Development

No branches or pull requests
2 participants
@malfet @johnnynunez
0