pytorch-bisect.yaml

name: PyTorch Bisect

on:
  workflow_dispatch:
    inputs:
      benchmark:
        description: Benchmark to bisect
        required: true
        type: choice
        default: inductor
        options:
          - inductor
      type:
        description: Bisect type
        required: true
        type: choice
        default: performance
        options:
          - functional
          - performance
      runner:
        description: Hardware runner
        required: true
        type: choice
        default: linux.dgx.b200
        options:
          - linux.dgx.b200
          - mt-l-x86iamx-22-225-h100
      repro_cmdline:
        description: The command line to reproduce
        required: true
        type: string
      good_commit:
        description: Last known good PyTorch commit
        required: true
        type: string
      bad_commit:
        description: First known bad PyTorch commit
        required: true
        type: string
      regression_threshold:
        type: number
        default: 10
        description: |
          Performance regression threshold in %

jobs:
  bisect:
    name: Run PyTorch Bisect
    runs-on: ${{ inputs.runner }}
    # mt- runners are ephemeral OSDC pods with no host CUDA toolchain, so build
    # PyTorch inside a CUDA devel container. linux.dgx.b200 is a DGX host and
    # keeps the bare-host path (fromJSON('null') => no container).
    container: ${{ startsWith(inputs.runner, 'mt-') && fromJSON('{"image":"pytorch/pytorch:2.12.0-cuda13.0-cudnn9-devel","options":"--gpus all"}') || fromJSON('null') }}
    env:
      USE_UV: '1'
      WORKSPACE_DIR: ${{ github.workspace }}
      GOOD_COMMIT: ${{ inputs.good_commit }}
      BAD_COMMIT: ${{ inputs.bad_commit }}
      REPRO_CMDLINE: ${{ inputs.repro_cmdline }}
      REGRESSION_THRESHOLD: ${{ inputs.regression_threshold }}
      PYTORCH_REPO: pytorch/pytorch
      FUNCTIONAL: ${{ inputs.type == 'functional' && '1' || '0' }}
      # Inside the OSDC CUDA devel container CUDA lives at /usr/local/cuda; on the
      # bare-host b200 path keep the workspace-local toolchain. run.sh requires
      # CUDA_HOME to be non-empty.
      CUDA_HOME: ${{ startsWith(inputs.runner, 'mt-') && '/usr/local/cuda' || format('{0}/torch-benchmarks/cuda', github.workspace) }}
      CUDA_VERSION: "13.0"
    permissions:
      id-token: write
      contents: read
    steps:
      - name: Configure git safe.directory
        run: git config --global --add safe.directory '*' || true

      - name: Checkout repository
        uses: actions/checkout@v4

      - name: Checkout tritonparse
        uses: actions/checkout@v4
        with:
          repository: pytorch-labs/tritonparse
          path: torch-benchmarks/tritonparse
          ref: xz9/fix-bisect
          fetch-depth: 0

      - name: Checkout PyTorch
        uses: actions/checkout@v4
        with:
          repository: pytorch/pytorch
          path: torch-benchmarks/pytorch
          submodules: recursive
          fetch-depth: 0

      - name: Checkout torchbench
        uses: actions/checkout@v4
        with:
          repository: pytorch/benchmark
          path: torch-benchmarks/benchmark
          fetch-depth: 0

      - name: Checkout torchvision
        uses: actions/checkout@v4
        with:
          repository: pytorch/vision
          path: torch-benchmarks/torchvision
          fetch-depth: 0

      - name: Install uv environment
        uses: pytorch/test-infra/.github/actions/setup-uv@main

      - name: Prepare bisect environment
        shell: bash
        run: |
          set -eux
          uv venv "${GITHUB_WORKSPACE}/torch-benchmarks/venv"
          export VIRTUAL_ENV="${GITHUB_WORKSPACE}/torch-benchmarks/venv"
          echo "VIRTUAL_ENV=${VIRTUAL_ENV}" >> $GITHUB_ENV
          echo "PYTORCH_SRC_DIR=$(realpath torch-benchmarks/pytorch)" >> $GITHUB_ENV
          echo "VISION_SRC_DIR=$(realpath torch-benchmarks/torchvision)" >> $GITHUB_ENV
          uv pip install -e torch-benchmarks/tritonparse

      - name: Run PyTorch bisect
        shell: bash
        run: |
          uv run .github/scripts/bisect/run.sh

      - uses: actions/upload-artifact@v4
        if: always()
        with:
          name: pytorch-bisect-logs
          path: ${{ github.workspace }}/bisect_logs
          retention-days: 30