reorganizing inductor test for triton #67
This file contains 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 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 | ||
|
Check failure on line 20 in .github/workflows/inductor.yml
|
||
| 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 | ||
