From 34cdb8eb94f3c1aefd31ae1c0499bffd9d3de20c Mon Sep 17 00:00:00 2001
From: Svetlana Karslioglu <svekars@meta.com>
Date: Tue, 16 Sep 2025 11:27:32 -0700
Subject: [PATCH] Add triton

---
 .ci/docker/requirements.txt | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt
index b246193ff0..9109b6578d 100644
--- a/.ci/docker/requirements.txt
+++ b/.ci/docker/requirements.txt
@@ -49,6 +49,7 @@ onnxscript>=0.2.2
 onnxruntime
 evaluate
 accelerate>=0.20.1
+triton==3.4.0
 
 importlib-metadata==6.8.0