Skip to content

Commit 8d35333

Browse files
atalmanpytorchmergebot
authored andcommitted
[CD] Aarch64 builds should not override OVERRIDE_PACKAGE_VERSION envvar (pytorch#144285)
Currently our nightly aarch64 binaries have correct suffixes +cpu or +cu126. But release binaries are missing these suffixes. Hence to correct this, make sure are nightly and release binaries are consistent, I propose this change. I see that override is already set correctly in release workflow: https://github.com/pytorch/pytorch/actions/runs/12383179841/job/34565381200 For CPU: ``` OVERRIDE_PACKAGE_VERSION="2.6.0+cpu" ``` For CUDA: ``` OVERRIDE_PACKAGE_VERSION="2.6.0+cu126" ``` The removed code will set : OVERRIDE_PACKAGE_VERSION="2.6.0" for both cuda and cpu builds for release binaries. Pull Request resolved: pytorch#144285 Approved by: https://github.com/malfet, https://github.com/tinglvv
1 parent 12fdb93 commit 8d35333

File tree

1 file changed

+0
-13
lines changed

1 file changed

+0
-13
lines changed

.ci/aarch64_linux/aarch64_ci_build.sh

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -6,19 +6,6 @@ GPU_ARCH_VERSION=${GPU_ARCH_VERSION:-}
66
SCRIPTPATH="$( cd -- "$(dirname "$0")" >/dev/null 2>&1 ; pwd -P )"
77
source $SCRIPTPATH/aarch64_ci_setup.sh
88

9-
tagged_version() {
10-
GIT_DESCRIBE="git --git-dir /pytorch/.git describe --tags --match v[0-9]*.[0-9]*.[0-9]*"
11-
if ${GIT_DESCRIBE} --exact >/dev/null; then
12-
${GIT_DESCRIBE}
13-
else
14-
return 1
15-
fi
16-
}
17-
18-
if tagged_version >/dev/null; then
19-
export OVERRIDE_PACKAGE_VERSION="$(tagged_version | sed -e 's/^v//' -e 's/-.*$//')"
20-
fi
21-
229
###############################################################################
2310
# Run aarch64 builder python
2411
###############################################################################

0 commit comments

Comments
 (0)