Python
/
uv
mirror of https://github.com/astral-sh/uv
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add `torch-tensorrt` and `torchao` to the PyTorch list (#17053) 

## Summary

Closes https://github.com/astral-sh/uv/issues/17050.
This commit is contained in:
Charlie Marsh 2025-12-09 15:29:32 -05:00 committed by GitHub
parent 2502577c9d
commit 4af2d2b922
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -314,6 +314,8 @@ impl TorchStrategy {
| "pytorch-triton-rocm"
| "pytorch-triton-xpu"
| "torch"
| "torch-tensorrt"
| "torchao"
| "torcharrow"
| "torchaudio"
| "torchcsprng"
@ -342,6 +344,8 @@ impl TorchStrategy {
| "torch-scatter"
| "torch-sparse"
| "torch-spline-conv"
| "torch-tensorrt"
| "torchao"
| "torcharrow"
| "torchaudio"
| "torchcsprng"
@ -371,6 +375,8 @@ impl TorchStrategy {
| "megablocks"
| "natten"
| "torch"
| "torch-tensorrt"
| "torchao"
| "torcharrow"
| "torchaudio"
| "torchcsprng"