From 53d698fb4873a9fb848891ffed3082fe21a5fdd4 Mon Sep 17 00:00:00 2001 From: Ilya Lavrenov Date: Thu, 23 Jan 2025 16:08:09 +0400 Subject: [PATCH] Pin triton version (#28638) ### Details: - Ported https://github.com/openvinotoolkit/openvino/pull/28627 - We can see here https://github.com/openvinotoolkit/openvino/pull/28630, that releases/2025/0 is affected as well. --- 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"