Skip to content

Commit 850d28b

Browse files
authored
ci: aarch64 linux: fix torch performance issue with conda openblas package (#1696)
changing the conda openblas package from pthread version to openmp version to match torch openmp runtime. The pthread version was conflicting with the openmp runtime and causing thread over-subscription and performance degradation.
1 parent 196b77b commit 850d28b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

aarch64_linux/aarch64_ci_setup.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ if [[ "$DESIRED_PYTHON" == "3.8" ]]; then
3030
else
3131
NUMPY_VERSION="1.26.2"
3232
fi
33-
conda install -y -c conda-forge numpy==${NUMPY_VERSION} pyyaml==6.0.1 patchelf==0.17.2 pygit2==1.13.2 openblas==0.3.25 ninja==1.11.1 scons==4.5.2
33+
conda install -y -c conda-forge numpy==${NUMPY_VERSION} pyyaml==6.0.1 patchelf==0.17.2 pygit2==1.13.2 openblas==0.3.25=*openmp* ninja==1.11.1 scons==4.5.2
3434

3535
python --version
3636
conda --version

0 commit comments

Comments
 (0)