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

Re-add triton as a torch backend package (#15910)

Browse source 
## Summary

This accidentally regressed in
https://github.com/astral-sh/uv/pull/15769/files#diff-fcd4a516243929cdb086b7b79af9865a6ed432a0386765b0436392edc17a5a4eL260.
This commit is contained in:
Charlie Marsh 2025-09-17 10:04:50 -04:00 • committed by GitHub
parent accfb48876
commit d4806ee921
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -313,7 +313,10 @@ impl TorchStrategy {
| "torchserve" | "torchserve"
| "torchtext" | "torchtext"
| "torchvision" | "torchvision"
| "triton"
| "pytorch-triton" | "pytorch-triton"
| "pytorch-triton-rocm"
| "pytorch-triton-xpu"
) )
} }
TorchSource::Pyx => { TorchSource::Pyx => {
@ -336,7 +339,10 @@ impl TorchStrategy {
| "torchserve" | "torchserve"
| "torchtext" | "torchtext"
| "torchvision" | "torchvision"
| "triton"
| "pytorch-triton" | "pytorch-triton"
| "pytorch-triton-rocm"
| "pytorch-triton-xpu"
) )
} }
} }