Skip to content

GitLab

Unverified Commit 0d4d06fe authored by Roger Wang's avatar Roger Wang Committed by GitHub
Browse files

[CI][Bugfix] Pin triton version for CPU (#16384) 


Signed-off-by: default avatarRoger Wang <ywang@roblox.com>
parent 4aed0ca6
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
requirements/cpu.txt
View file @ 0d4d06fe
......@@ -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
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment