• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1name: torchbench
2
3on:
4  push:
5    tags:
6      - ciflow/torchbench/*
7  workflow_dispatch:
8
9concurrency:
10  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-${{ github.ref_type == 'branch' && github.sha }}-${{ github.event_name == 'workflow_dispatch' }}-${{ github.event_name == 'schedule' }}
11  cancel-in-progress: true
12
13jobs:
14  linux-focal-cuda12_1-py3_10-gcc9-torchbench-build-gcp:
15    name: cuda12.1-py3.10-gcc9-sm80
16    uses: ./.github/workflows/_linux-build.yml
17    with:
18      build-environment: linux-focal-cuda12.1-py3.10-gcc9-sm80
19      docker-image-name: pytorch-linux-focal-cuda12.1-cudnn9-py3-gcc9-inductor-benchmarks
20      cuda-arch-list: '8.0'
21      test-matrix: |
22        { include: [
23          { config: "torchbench_gcp_smoketest", shard: 1, num_shards: 1, runner: "linux.gcp.a100" },
24        ]}
25    secrets:
26      HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}
27
28  linux-focal-cuda12_1-py3_10-gcc9-torchbench-test-gcp:
29    name: cuda12.1-py3.10-gcc9-sm80
30    uses: ./.github/workflows/_linux-test.yml
31    needs: linux-focal-cuda12_1-py3_10-gcc9-torchbench-build-gcp
32    with:
33      build-environment: linux-focal-cuda12.1-py3.10-gcc9-sm80
34      docker-image: ${{ needs.linux-focal-cuda12_1-py3_10-gcc9-torchbench-build-gcp.outputs.docker-image }}
35      test-matrix: ${{ needs.linux-focal-cuda12_1-py3_10-gcc9-torchbench-build-gcp.outputs.test-matrix }}
36      use-gha: anything-non-empty-to-use-gha
37    secrets:
38      HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}
39