Skip to content

Test CUDA Builds #17178

Test CUDA Builds

Test CUDA Builds #17178

Workflow file for this run

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