-
Notifications
You must be signed in to change notification settings - Fork 3
39 lines (37 loc) · 1.01 KB
/
pr.yaml
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: TritonBench PR Test
on:
pull_request:
paths:
- .ci/*
- tritonbench/*
- .github/workflows/pr.yaml
push:
branches:
- main
jobs:
h100-pytorch-test:
# Don't run on forked repos
if: github.repository_owner == 'pytorch-labs'
runs-on: [gcp-h100-runner]
timeout-minutes: 240
environment: docker-s3-upload
env:
CONDA_ENV: "pytorch"
SETUP_SCRIPT: "/workspace/setup_instance.sh"
steps:
- name: Checkout Tritonbench
uses: actions/checkout@v3
with:
# no need to checkout submodules recursively
submodules: true
- name: Tune Nvidia GPU
run: |
sudo nvidia-smi -pm 1
sudo ldconfig
nvidia-smi
- name: Test Tritonbench operators on H100 GPU
run: |
bash ./.ci/tritonbench/test-gpu.sh
concurrency:
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}-${{ github.event_name == 'workflow_dispatch' }}
cancel-in-progress: true