Skip to content

Commit 61e2308

Browse files
Add triton to torch-backend manifest (#15405)
## Summary The PyTorch team publishes ARM Linux wheels for `triton` to the PyTorch index, which aren't available on PyPI. ## Test Plan ``` echo "torch" | cargo run pip compile - --torch-backend=cu128 --python-platform aarch64-unknown-linux-gnu --python-version 3.13 ``` Previously failed because it couldn't find a compatible `triton` wheel.
1 parent e31f000 commit 61e2308

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

crates/uv-torch/src/backend.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -267,6 +267,7 @@ impl TorchStrategy {
267267
| "torchserve"
268268
| "torchtext"
269269
| "torchvision"
270+
| "triton"
270271
| "pytorch-triton"
271272
| "pytorch-triton-rocm"
272273
| "pytorch-triton-xpu"

0 commit comments

Comments
 (0)