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"