Skip to content

Add Tritonbench CI #541

Add Tritonbench CI

Add Tritonbench CI #541

Workflow file for this run

name: TorchBench PR Test
on:
pull_request:
# ignore tritonbench paths
paths-ignore:
- 'torchbenchmark/operators/*'
- 'torchbenchmark/util/kernels/*'
- 'torchbenchmark/util/triton_op.py'
- 'userbenchmark/triton/*'
- '.ci/tritonbench/*'
workflow_dispatch:
push:
branches:
- main
# ignore tritonbench paths
paths-ignore:
- 'torchbenchmark/operators/*'
- 'torchbenchmark/util/kernels/*'
- 'torchbenchmark/util/triton_op.py'
- 'userbenchmark/triton/*'
- '.ci/tritonbench/*'
jobs:
cpu-test:
uses: ./.github/workflows/_linux-test-cpu.yml
secrets:
HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}
cuda-test:
uses: ./.github/workflows/_linux-test-cuda.yml
secrets:
HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}
concurrency:
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}-${{ github.event_name == 'workflow_dispatch' }}
cancel-in-progress: true