mirrors/uv
1
0
Fork 0
mirror of https://github.com/astral-sh/uv.git synced 2025-11-03 05:03:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Add triton to torch-backend manifest (#15405)

Browse source 
## 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.
This commit is contained in:
Charlie Marsh 2025-08-21 14:23:12 +01:00 • committed by GitHub
parent e31f000da7
commit 61e2308806
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
crates/uv-torch/src/backend.rs
View file

@ -267,6 +267,7 @@ impl TorchStrategy {
| "torchserve"
| "torchtext"
| "torchvision"
| "triton"
| "pytorch-triton"
| "pytorch-triton-rocm"
| "pytorch-triton-xpu"