From 6fa2d882b2020cbd779a9d236efabed3851a943f Mon Sep 17 00:00:00 2001
From: Ilya Lavrenov <ilya.lavrenov@intel.com>
Date: Thu, 23 Jan 2025 10:54:59 +0100
Subject: [PATCH] Pin triton version

---
 tests/requirements_pytorch | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/requirements_pytorch b/tests/requirements_pytorch
index 33907145f7de4b..df8e89db7511fe 100644
--- a/tests/requirements_pytorch
+++ b/tests/requirements_pytorch
@@ -21,6 +21,8 @@ pytest-xdist[psutil]==3.6.1
 defusedxml==0.7.1
 
 autoawq==0.2.7; platform_system == "Linux" and platform_machine == "x86_64"
+# triton is a dependency of autoawq, newer versions lead to TorchFX test failures
+triton==3.1.0; platform_system == "Linux" and platform_machine == "x86_64"
 auto-gptq==0.7.1; platform_system == "Linux" and platform_machine == "x86_64" and python_version < "3.12"
 av==13.0.0
 basicsr==1.4.2; python_version < "3.12"