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

## Summary Closes https://github.com/astral-sh/uv/issues/17050.
2025-12-09 15:29:32 -05:00
parent 2502577c9d
commit 4af2d2b922
1 changed files with 6 additions and 0 deletions
@@ -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"