Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
ci: aarch64 linux: fix torch performance issue with conda openblas pa…
Browse files Browse the repository at this point in the history
…ckage

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.
snadampal committed Feb 3, 2024
1 parent da779da commit 6fafe6c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion aarch64_linux/aarch64_ci_setup.sh
Original file line number Diff line number Diff line change
@@ -30,7 +30,7 @@ if [[ "$DESIRED_PYTHON" == "3.8" ]]; then
else
NUMPY_VERSION="1.26.2"
fi
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
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

python --version
conda --version

0 comments on commit 6fafe6c

Please sign in to comment.