pioduino/uv
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

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

Browse Source 
## 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
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/uv-torch/src/backend.rs
+6
View File
@@ -314,6 +314,8 @@ impl TorchStrategy {
| "pytorch-triton-rocm" | "pytorch-triton-rocm"
| "pytorch-triton-xpu" | "pytorch-triton-xpu"
| "torch" | "torch"
| "torch-tensorrt"
| "torchao"
| "torcharrow" | "torcharrow"
| "torchaudio" | "torchaudio"
| "torchcsprng" | "torchcsprng"
@@ -342,6 +344,8 @@ impl TorchStrategy {
| "torch-scatter" | "torch-scatter"
| "torch-sparse" | "torch-sparse"
| "torch-spline-conv" | "torch-spline-conv"
| "torch-tensorrt"
| "torchao"
| "torcharrow" | "torcharrow"
| "torchaudio" | "torchaudio"
| "torchcsprng" | "torchcsprng"
@@ -371,6 +375,8 @@ impl TorchStrategy {
| "megablocks" | "megablocks"
| "natten" | "natten"
| "torch" | "torch"
| "torch-tensorrt"
| "torchao"
| "torcharrow" | "torcharrow"
| "torchaudio" | "torchaudio"
| "torchcsprng" | "torchcsprng"