Skip to content

Commit 5e16b8c

Browse files
fadara01yewentao256
authored andcommitted
[fix] Update torch version in cpu-build.txt for AArch64/ppc64le and Darwin (#25579)
Signed-off-by: Fadi Arafeh <fadi.arafeh@arm.com> Signed-off-by: yewentao256 <zhyanwentao@126.com>
1 parent 6c6e553 commit 5e16b8c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

requirements/cpu-build.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,8 @@ packaging>=24.2
66
setuptools>=77.0.3,<80.0.0
77
setuptools-scm>=8
88
--extra-index-url https://download.pytorch.org/whl/cpu
9-
torch==2.6.0+cpu
9+
torch==2.6.0+cpu; platform_machine == "x86_64" # torch>2.6.0+cpu has performance regression on x86 platform, see https://github.com/pytorch/pytorch/pull/151218
10+
torch==2.8.0; platform_machine == "ppc64le" or platform_machine == "aarch64" or platform_system == "Darwin"
1011
wheel
1112
jinja2>=3.1.6
1213
regex

0 commit comments

Comments
 (0)