From d4806ee921f59eeab6834d62c1c5bd0b368fa804 Mon Sep 17 00:00:00 2001 From: Charlie Marsh Date: Wed, 17 Sep 2025 10:04:50 -0400 Subject: [PATCH] Re-add `triton` as a torch backend package (#15910) ## Summary This accidentally regressed in https://github.com/astral-sh/uv/pull/15769/files#diff-fcd4a516243929cdb086b7b79af9865a6ed432a0386765b0436392edc17a5a4eL260. --- 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 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" ) } }