From 61e2308806f8231f535c12dbdd38494433d4d99a Mon Sep 17 00:00:00 2001
From: Charlie Marsh <charlie.r.marsh@gmail.com>
Date: Thu, 21 Aug 2025 14:23:12 +0100
Subject: [PATCH] Add `triton` to `torch-backend` manifest (#15405)

## Summary

The PyTorch team publishes ARM Linux wheels for `triton` to the PyTorch
index, which aren't available on PyPI.

## Test Plan

```
echo "torch" | cargo run pip compile - --torch-backend=cu128 --python-platform aarch64-unknown-linux-gnu --python-version 3.13
```

Previously failed because it couldn't find a compatible `triton` wheel.
---
 crates/uv-torch/src/backend.rs | 1 +
 1 file changed, 1 insertion(+)

diff --git a/crates/uv-torch/src/backend.rs b/crates/uv-torch/src/backend.rs
index 44658b93f..5407d6fbe 100644
--- a/crates/uv-torch/src/backend.rs
+++ b/crates/uv-torch/src/backend.rs
@@ -267,6 +267,7 @@ impl TorchStrategy {
                 | "torchserve"
                 | "torchtext"
                 | "torchvision"
+                | "triton"
                 | "pytorch-triton"
                 | "pytorch-triton-rocm"
                 | "pytorch-triton-xpu"