pr-test.yml

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