inductor.yml

name: Test build/test linux gpu

on:
  pull_request:
    branches: [ main, "*"] #will remove this once we see everything is working fine
  workflow_dispatch:
    inputs:
      triton_pin:
        description: 'Triton branch or commit to pin'
        default: 'main'
        required: false
      pytorch_pin:
        description: 'PyTorch branch or commit to pin'
        default: 'main'
        required: false

jobs:
  build-test:
    continue-on-error: true
    uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
    with:
      runner: linux.g5.48xlarge.nvidia.gpu
      gpu-arch-type: cuda
      gpu-arch-version: "12.1"
      timeout: 360
      # docker-image: nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
      script: |
        set -x
        pushd ..
        echo "Installing triton"
        git clone https://github.com/triton-lang/triton.git
        pushd triton
        echo "Checking out triton branch or commit"
        git checkout ${{ github.event.inputs.triton_pin || 'main' }}
        sudo yum install -y zlib-devel
        echo "Installing build-time dependencies"
        pip install ninja==1.11.1.1 cmake==3.30.2 wheel==0.44.0
        export llvm_hash=$(cat cmake/llvm-hash.txt)
        echo "llvm_hash: $llvm_hash"
        pip install -e python
        pushd ..
        echo "Cloning pytorch"
        git clone https://github.com/pytorch/pytorch.git
        pushd pytorch
        echo "Checking out pytorch branch or commit"
        git checkout ${{ github.event.inputs.pytorch_pin || 'main' }}
        git submodule sync
        git submodule update --init --recursive
        pip install -r requirements.txt
        pip install mkl-static mkl-include pytest pytest-xdist
        echo "Installing magma-cuda121"
        conda install -y -c pytorch magma-cuda121
        python setup.py install
        pip freeze
        pytest -n 1 test/inductor/test_torchinductor.py