From 51baa9c333a348915dca5d6a2eca0d6037751874 Mon Sep 17 00:00:00 2001 From: Harry Mellor <19981378+hmellor@users.noreply.github.com> Date: Fri, 11 Apr 2025 11:11:00 +0100 Subject: [PATCH] Don't install triton on `ppc64le` platform (#16470) Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> --- requirements/cpu.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements/cpu.txt b/requirements/cpu.txt index 1205c098e6f06..d845fb201ceff 100644 --- a/requirements/cpu.txt +++ b/requirements/cpu.txt @@ -17,4 +17,4 @@ torchvision==0.21.0; platform_machine == "ppc64le" datasets # for benchmark scripts # cpu cannot use triton 3.3.0 -triton==3.2.0 +triton==3.2.0; platform_machine != "ppc64le"