test

pytorch · May 28, 2024 · c705173 · c705173
1 parent bab2ee3
commit c705173
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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: