forked from triton-lang/triton
-
Notifications
You must be signed in to change notification settings - Fork 0
39 lines (36 loc) · 1.1 KB
/
torch-inductor-tests.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
name: Torchinductor
on:
workflow_dispatch:
jobs:
Runner-Preparation:
runs-on: ubuntu-latest
outputs:
matrix: ${{ steps.set-matrix.outputs.matrix }}
steps:
- name: Prepare runner matrix
id: set-matrix
run: |
echo '::set-output name=matrix::[["self-hosted", "A100"]]'
Integration-Tests:
needs: Runner-Preparation
runs-on: ${{ matrix.runner }}
strategy:
matrix:
runner: ${{fromJson(needs.Runner-Preparation.outputs.matrix)}}
steps:
- name: Checkout
uses: actions/checkout@v2
#- name: Packages
# run: |
# ./.github/workflows/torchinductor/scripts/install_torchinductor.sh
- name: Environment
run: |
source /opt/torchinductor_venv/bin/activate
./.github/workflows/torchinductor/scripts/install_triton.sh
- name: Performance
run: |
./.github/workflows/torchinductor/scripts/run_torchinductor_perf.sh
# Runs too long time
#- name: Accuracy
# run: |
# ./.github/workflows/torchinductor/scripts/run_torchinductor_acc.sh