diff --git a/crates/uv-torch/src/backend.rs b/crates/uv-torch/src/backend.rs index 4c44bb730..b9fa72141 100644 --- a/crates/uv-torch/src/backend.rs +++ b/crates/uv-torch/src/backend.rs @@ -313,7 +313,10 @@ impl TorchStrategy { | "torchserve" | "torchtext" | "torchvision" + | "triton" | "pytorch-triton" + | "pytorch-triton-rocm" + | "pytorch-triton-xpu" ) } TorchSource::Pyx => { @@ -336,7 +339,10 @@ impl TorchStrategy { | "torchserve" | "torchtext" | "torchvision" + | "triton" | "pytorch-triton" + | "pytorch-triton-rocm" + | "pytorch-triton-xpu" ) } }