From 4b72a065eebb16684993215df3b974c51ca2f070 Mon Sep 17 00:00:00 2001 From: Ilya Lavrenov Date: Thu, 23 Jan 2025 15:50:28 +0400 Subject: [PATCH] Pin triton version (#28627) ### Details: - *item1* - *...* ### Tickets: - *ticket-id* --- tests/requirements_pytorch | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/requirements_pytorch b/tests/requirements_pytorch index f42deb81839883..87c57fb7465167 100644 --- a/tests/requirements_pytorch +++ b/tests/requirements_pytorch @@ -20,6 +20,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"