diff --git a/.github/workflows/build_wheels_linux.yml b/.github/workflows/build_wheels_linux.yml index b789b169e6..037214eff6 100644 --- a/.github/workflows/build_wheels_linux.yml +++ b/.github/workflows/build_wheels_linux.yml @@ -220,11 +220,14 @@ jobs: PACKAGE_NAME: ${{ inputs.package-name }} SMOKE_TEST_SCRIPT: ${{ inputs.smoke-test-script }} run: | - set -euxo pipefail + set uxo pipefail + set +e source "${BUILD_ENV_FILE}" WHEEL_NAME=$(ls "${{ inputs.repository }}/dist/") echo "$WHEEL_NAME" + export TORCHINDUCTOR_COMPILE_THREADS=1 + export TORCH_LOGS=output_code ${CONDA_RUN} pip install "${{ inputs.repository }}/dist/$WHEEL_NAME" # Checking that we have a pinned version of torch in our dependency tree ( @@ -248,6 +251,7 @@ jobs: echo "${{ inputs.repository }}/${SMOKE_TEST_SCRIPT} found" ${CONDA_RUN} python "${{ inputs.repository }}/${SMOKE_TEST_SCRIPT}" fi + cat /tmp/torchinductor_root/tt/cttmh7vyic6svkvlu2hcuw4luorlu3r2nantzcwl4y23rlv3oqsp.py # NB: Only upload to GitHub after passing smoke tests - name: Upload wheel to GitHub continue-on-error: true diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index f6f222f1ac..f4e14918e7 100644 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -489,7 +489,7 @@ def generate_wheels_matrix( arches += mod.ROCM_ARCHES if limit_pr_builds: - python_versions = [python_versions[0]] + python_versions = ["3.11"] ret: List[Dict[str, str]] = [] for python_version in python_versions: