Skip to content

Commit 0d4d06f

Browse files
authored
[CI][Bugfix] Pin triton version for CPU (#16384)
Signed-off-by: Roger Wang <[email protected]>
1 parent 4aed0ca commit 0d4d06f

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

requirements/cpu.txt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,3 +15,6 @@ torchaudio==2.6.0; platform_machine == "ppc64le"
1515
torchvision; platform_machine != "ppc64le" and platform_machine != "s390x"
1616
torchvision==0.21.0; platform_machine == "ppc64le"
1717
datasets # for benchmark scripts
18+
19+
# cpu cannot use triton 3.3.0
20+
triton==3.2.0

0 commit comments

Comments
 (0)