Skip to content

use the same pytorch/torchao version same as ET repro in the wheel package test #14560

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 8 commits into from
Sep 27, 2025
Merged

use the same pytorch/torchao version same as ET repro in the wheel package test #14560

Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
97b791d
pin the same as ET repro
cccclai Sep 24, 2025
11719a5
changes
cccclai Sep 25, 2025
94c24e7
debug
cccclai Sep 25, 2025
d22dd4a
fix
cccclai Sep 25, 2025
3d5c962
restore
cccclai Sep 25, 2025
fa2e70f
fix
cccclai Sep 26, 2025
de25986
fix
cccclai Sep 26, 2025
9500c91
execute python code to get version
cccclai Sep 27, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 26 additions & 4 deletions .ci/scripts/test_wheel_package_qnn.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -98,7 +98,7 @@ PYTHON_VERSION=$1
# Check wheel does NOT contain qualcomm/sdk
# ----------------------------
echo "Checking wheel does not contain qualcomm/sdk..."
SDK_FILES=$(unzip -l "$WHEEL_FILE" | awk '{print $4}' | grep "executorch/backends/qualcomm/sdk" || true)
SDK_FILES=$(unzip -l "$WHEEL_FILE" | awk '{print $4}' | grep -E "executorch/backends/qualcomm/sdk" || true)
if [ -n "$SDK_FILES" ]; then
echo "ERROR: Wheel package contains unexpected qualcomm/sdk files:"
echo "$SDK_FILES"
Expand All @@ -111,7 +111,7 @@ fi
# Check .so files in the wheel
# ----------------------------
echo "Checking for .so files inside the wheel..."
WHEEL_SO_FILES=$(unzip -l "$WHEEL_FILE" | awk '{print $4}' | grep "executorch/backends/qualcomm/python" || true)
WHEEL_SO_FILES=$(unzip -l "$WHEEL_FILE" | awk '{print $4}' | grep -E "executorch/backends/qualcomm/python" || true)
if [ -z "$WHEEL_SO_FILES" ]; then
echo "ERROR: No .so files found in wheel under executorch/backends/qualcomm/python"
exit 1
Expand Down Expand Up @@ -139,8 +139,30 @@ run_core_tests () {
echo "=== [$LABEL] Installing wheel & deps ==="
"$PIPBIN" install --upgrade pip
"$PIPBIN" install "$WHEEL_FILE"
"$PIPBIN" install torch=="2.9.0.dev20250906" --index-url "https://download.pytorch.org/whl/nightly/cpu"
"$PIPBIN" install --pre torchao --index-url "https://download.pytorch.org/whl/nightly/cpu"
TORCH_VERSION=$(
"$PYBIN" - <<'PY'
import runpy
module_vars = runpy.run_path("torch_pin.py")
print(module_vars["TORCH_VERSION"])
PY
)

NIGHTLY_VERSION=$(
"$PYBIN" - <<'PY'
import runpy
module_vars = runpy.run_path("torch_pin.py")
print(module_vars["NIGHTLY_VERSION"])
PY
)
echo "=== [$LABEL] Install torch==${TORCH_VERSION}.${NIGHTLY_VERSION} ==="

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

# Install torchao based on the pinned commit from third-party/ao submodule
pushd "$REPO_ROOT/third-party/ao" > /dev/null
USE_CPP=0 "$PYBIN" setup.py develop
popd > /dev/null

echo "=== [$LABEL] Import smoke tests ==="
"$PYBIN" -c "import executorch; print('executorch imported successfully')"
Expand Down
Loading