Skip to content

Commit b9a0569

Browse files
authored
[RELEASE ONLY CHANGES] Pin qnn wheel build job to torch 2.10.0 (#16874)
### Summary This change pins the QNN wheel build job to torch 2.10 for 1.1 release. See #15664 as example from 1.0 release.
1 parent 29b0432 commit b9a0569

1 file changed

Lines changed: 3 additions & 16 deletions

File tree

.ci/scripts/test_wheel_package_qnn.sh

Lines changed: 3 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -139,25 +139,12 @@ run_core_tests () {
139139
echo "=== [$LABEL] Installing wheel & deps ==="
140140
"$PIPBIN" install --upgrade pip
141141
"$PIPBIN" install "$WHEEL_FILE"
142-
TORCH_VERSION=$(
143-
"$PYBIN" - <<'PY'
144-
import runpy
145-
module_vars = runpy.run_path("torch_pin.py")
146-
print(module_vars["TORCH_VERSION"])
147-
PY
148-
)
142+
TORCH_VERSION=2.10.0
149143

150-
NIGHTLY_VERSION=$(
151-
"$PYBIN" - <<'PY'
152-
import runpy
153-
module_vars = runpy.run_path("torch_pin.py")
154-
print(module_vars["NIGHTLY_VERSION"])
155-
PY
156-
)
157-
echo "=== [$LABEL] Install torch==${TORCH_VERSION}.${NIGHTLY_VERSION} ==="
144+
echo "=== [$LABEL] Install torch==${TORCH_VERSION} ==="
158145

159146
# Install torchao based on the pinned PyTorch version
160-
"$PIPBIN" install torch=="${TORCH_VERSION}.${NIGHTLY_VERSION}" --index-url "https://download.pytorch.org/whl/nightly/cpu"
147+
"$PIPBIN" install torch=="${TORCH_VERSION}"
161148
"$PIPBIN" install wheel
162149

163150
# Install torchao based on the pinned commit from third-party/ao submodule

0 commit comments

Comments
 (0)