Skip to content

Commit bd9b7cf

Browse files
committed
requirements/cpu.txt: set platform_machine
fixes CPU build for s390x/ppc64le Signed-off-by: Daniele Trifirò <[email protected]>
1 parent a6d8c3d commit bd9b7cf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

requirements/cpu.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,4 +17,4 @@ torchvision==0.21.0; platform_machine == "ppc64le"
1717
datasets # for benchmark scripts
1818

1919
# cpu cannot use triton 3.3.0
20-
triton==3.2.0; platform_machine != "ppc64le"
20+
triton==3.2.0; platform_machine == "x86_64"

0 commit comments

Comments
 (0)