From 4af2d2b922e1c579bc0977dd22d6573cce7026de Mon Sep 17 00:00:00 2001
From: Charlie Marsh <charlie.r.marsh@gmail.com>
Date: Tue, 9 Dec 2025 15:29:32 -0500
Subject: [PATCH] Add `torch-tensorrt` and `torchao` to the PyTorch list
 (#17053)

## Summary

Closes https://github.com/astral-sh/uv/issues/17050.
---
 crates/uv-torch/src/backend.rs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/crates/uv-torch/src/backend.rs b/crates/uv-torch/src/backend.rs
index 2bb8374fa..9455980e7 100644
--- a/crates/uv-torch/src/backend.rs
+++ b/crates/uv-torch/src/backend.rs
@@ -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"