File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -131,6 +131,7 @@ git submodule update --init --recursive
131131
132132python -m pip install pyyaml
133133COMMIT_TORCH=$( python tools/yaml_utils.py -f dependency_version.yml -d pytorch -k commit)
134+ VERSION_TORCH=$( python tools/yaml_utils.py -f dependency_version.yml -d pytorch -k version)
134135COMMIT_TORCHVISION=$( python tools/yaml_utils.py -f dependency_version.yml -d torchvision -k commit)
135136COMMIT_TORCHAUDIO=$( python tools/yaml_utils.py -f dependency_version.yml -d torchaudio -k commit)
136137REPO_TORCHCCL=$( python tools/yaml_utils.py -f dependency_version.yml -d torch-ccl -k repo)
@@ -226,7 +227,7 @@ cd pytorch
226227git apply ../intel-extension-for-pytorch/torch_patches/* .patch
227228python -m pip install -r requirements.txt
228229conda install --force-reinstall intel::mkl-static intel::mkl-include -y
229- export PYTORCH_BUILD_VERSION=" ${COMMIT_TORCH : 1} .post0+cxx11.abi "
230+ export PYTORCH_BUILD_VERSION=${VERSION_TORCH}
230231export PYTORCH_BUILD_NUMBER=0
231232# Ensure cmake can find python packages when using conda or virtualenv
232233if [ -n " ${CONDA_PREFIX-} " ]; then
You can’t perform that action at this time.
0 commit comments