Skip to content

Commit 6fafe6c

Browse files
committed
ci: aarch64 linux: fix torch performance issue with conda openblas package
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 da779da commit 6fafe6c

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)