Skip to content

Commit 3957ad2

Browse files
authored
update compile_bundle.sh pytorch version (#4188)
1 parent 1219fa6 commit 3957ad2

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

scripts/compile_bundle.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,7 @@ git submodule update --init --recursive
131131

132132
python -m pip install pyyaml
133133
COMMIT_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)
134135
COMMIT_TORCHVISION=$(python tools/yaml_utils.py -f dependency_version.yml -d torchvision -k commit)
135136
COMMIT_TORCHAUDIO=$(python tools/yaml_utils.py -f dependency_version.yml -d torchaudio -k commit)
136137
REPO_TORCHCCL=$(python tools/yaml_utils.py -f dependency_version.yml -d torch-ccl -k repo)
@@ -226,7 +227,7 @@ cd pytorch
226227
git apply ../intel-extension-for-pytorch/torch_patches/*.patch
227228
python -m pip install -r requirements.txt
228229
conda 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}
230231
export PYTORCH_BUILD_NUMBER=0
231232
# Ensure cmake can find python packages when using conda or virtualenv
232233
if [ -n "${CONDA_PREFIX-}" ]; then

0 commit comments

Comments
 (0)