From eb892ae14e204fda269c9a37c6d78578ffd9a0df Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?oliver=20k=C3=B6nig?= <okoenig@nvidia.com>
Date: Thu, 23 Jan 2025 16:57:23 +0100
Subject: [PATCH] build: Pin down `triton` (#11938)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: oliver könig <okoenig@nvidia.com>
---
 reinstall.sh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/reinstall.sh b/reinstall.sh
index 2c59dd7eb9e3..030e6252a0c6 100755
--- a/reinstall.sh
+++ b/reinstall.sh
@@ -65,6 +65,7 @@ if [[ "$HEAVY_DEPS" == "TRUE" ]]; then
     "unstructured==0.14.9"
     "git+https://github.com/Dao-AILab/causal-conv1d.git@${CAUSAL_CONV_TAG}"
     "git+https://github.com/state-spaces/mamba.git@${MAMBA_TAG}"
+    "triton==3.1.0"
   )
 
   pip install --no-cache-dir -r tools/ctc_segmentation/requirements.txt