Skip to content

Commit 97b791d

Browse files
committed
pin the same as ET repro
1 parent 411578a commit 97b791d

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

.ci/scripts/test_wheel_package_qnn.sh

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -139,8 +139,13 @@ run_core_tests () {
139139
echo "=== [$LABEL] Installing wheel & deps ==="
140140
"$PIPBIN" install --upgrade pip
141141
"$PIPBIN" install "$WHEEL_FILE"
142-
"$PIPBIN" install torch=="2.9.0.dev20250906" --index-url "https://download.pytorch.org/whl/nightly/cpu"
143-
"$PIPBIN" install --pre torchao --index-url "https://download.pytorch.org/whl/nightly/cpu"
142+
143+
NIGHTLY_VERSION=$(python -c 'import re; txt=open("install_requirements.py").read(); print(re.search(r"NIGHTLY_VERSION\s*=\s*[\"\']([^\"\']+)[\"\']", txt).group(1))')
144+
145+
"$PIPBIN" install torch=="2.10.0.${NIGHTLY_VERSION}" --index-url "https://download.pytorch.org/whl/nightly/cpu"
146+
pushd "$REPO_ROOT" / third-party/ao > /dev/null
147+
USE_CPP=0 python setup.py develop
148+
popd > /dev/null
144149
145150
echo "=== [$LABEL] Import smoke tests ==="
146151
"$PYBIN" -c "import executorch; print('executorch imported successfully')"

0 commit comments

Comments
 (0)