From c7051732e92ebff238867b5ed314f6b5ba59d516 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 28 May 2024 13:32:26 -0700 Subject: [PATCH 1/3] test --- .github/workflows/build_wheels_linux.yml | 1 + tools/scripts/generate_binary_build_matrix.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/build_wheels_linux.yml b/.github/workflows/build_wheels_linux.yml index b789b169e6..ac337aa449 100644 --- a/.github/workflows/build_wheels_linux.yml +++ b/.github/workflows/build_wheels_linux.yml @@ -225,6 +225,7 @@ jobs: WHEEL_NAME=$(ls "${{ inputs.repository }}/dist/") echo "$WHEEL_NAME" + export TORCHINDUCTOR_COMPILE_THREADS=1 ${CONDA_RUN} pip install "${{ inputs.repository }}/dist/$WHEEL_NAME" # Checking that we have a pinned version of torch in our dependency tree ( 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: From 0873e5762d263c8f99de449b37e23c970a941369 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 28 May 2024 14:35:34 -0700 Subject: [PATCH 2/3] test --- .github/workflows/build_wheels_linux.yml | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/.github/workflows/build_wheels_linux.yml b/.github/workflows/build_wheels_linux.yml index ac337aa449..41845df083 100644 --- a/.github/workflows/build_wheels_linux.yml +++ b/.github/workflows/build_wheels_linux.yml @@ -220,12 +220,13 @@ jobs: PACKAGE_NAME: ${{ inputs.package-name }} SMOKE_TEST_SCRIPT: ${{ inputs.smoke-test-script }} run: | - set -euxo pipefail + set uxo pipefail 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 ( @@ -249,6 +250,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 From 27e966052c73f86ea222ae252677bd045de16a1b Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 28 May 2024 14:49:49 -0700 Subject: [PATCH 3/3] test --- .github/workflows/build_wheels_linux.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/workflows/build_wheels_linux.yml b/.github/workflows/build_wheels_linux.yml index 41845df083..037214eff6 100644 --- a/.github/workflows/build_wheels_linux.yml +++ b/.github/workflows/build_wheels_linux.yml @@ -221,6 +221,7 @@ jobs: SMOKE_TEST_SCRIPT: ${{ inputs.smoke-test-script }} run: | set uxo pipefail + set +e source "${BUILD_ENV_FILE}" WHEEL_NAME=$(ls "${{ inputs.repository }}/dist/") echo "$WHEEL_NAME"