From 0d4d06fe2fe248dbd5c4a360e8c4ffa3b386279c Mon Sep 17 00:00:00 2001 From: Roger Wang <136131678+ywang96@users.noreply.github.com> Date: Wed, 9 Apr 2025 21:35:00 -0700 Subject: [PATCH] [CI][Bugfix] Pin triton version for CPU (#16384) Signed-off-by: Roger Wang --- requirements/cpu.txt | 3 +++ 1 file changed, 3 insertions(+) diff --git a/requirements/cpu.txt b/requirements/cpu.txt index fc09083781e6f..1205c098e6f06 100644 --- a/requirements/cpu.txt +++ b/requirements/cpu.txt @@ -15,3 +15,6 @@ torchaudio==2.6.0; platform_machine == "ppc64le" torchvision; platform_machine != "ppc64le" and platform_machine != "s390x" torchvision==0.21.0; platform_machine == "ppc64le" datasets # for benchmark scripts + +# cpu cannot use triton 3.3.0 +triton==3.2.0