Test CUDA Builds #17178
Workflow file for this run
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Test CUDA Builds | |
| on: | |
| workflow_dispatch: | |
| permissions: | |
| contents: read | |
| jobs: | |
| repro-loop-ordering: | |
| name: repro-loop-ordering | |
| uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main | |
| permissions: | |
| id-token: write | |
| contents: read | |
| with: | |
| timeout: 30 | |
| runner: mt-l-x86iavx512-11-125-a100 | |
| gpu-arch-type: cuda | |
| gpu-arch-version: "13.0" | |
| use-custom-docker-registry: false | |
| submodules: recursive | |
| ref: ${{ github.sha }} | |
| script: | | |
| set -ux | |
| export PIP_EXTRA_INDEX_URL= | |
| pip install filelock typing-extensions "setuptools<82" sympy networkx jinja2 "fsspec[http]<=2025.3.0" numpy pillow | |
| ./install_requirements.sh | |
| python -c 'import torch, triton; print(torch.__version__, triton.__version__)' | |
| export TORCHINDUCTOR_COMPILE_THREADS=1 | |
| export TORCH_COMPILE_DEBUG=1 | |
| export TORCHINDUCTOR_CACHE_DIR=/tmp/inductor_loop_ordering_false | |
| python repro_loop_ordering.py --loop-ordering-after-fusion false 2>&1 | tee /tmp/false.log | |
| false_status=${PIPESTATUS[0]} | |
| export TORCHINDUCTOR_CACHE_DIR=/tmp/inductor_loop_ordering_true | |
| python repro_loop_ordering.py --loop-ordering-after-fusion true 2>&1 | tee /tmp/true.log | |
| true_status=${PIPESTATUS[0]} | |
| echo "LOOP_ORDERING_FALSE_STATUS=${false_status}" | |
| echo "LOOP_ORDERING_TRUE_STATUS=${true_status}" | |
| echo "Undefined temporary errors:" | |
| grep -E "NameError.*tmp[0-9]+.*not defined" /tmp/false.log /tmp/true.log || true | |
| exit 0 |