FilesExpand file tree

master

/

inductor-perf-compare.yml

Copy path

View runs

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

37 lines (33 loc) · 1.62 KB

master

/

inductor-perf-compare.yml

Copy path

Top

File metadata and controls

• Code
• Blame

37 lines (33 loc) · 1.62 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

name: inductor-A100-perf-compare

on:

push:

tags:

- ciflow/inductor-perf-compare/*

workflow_dispatch:

concurrency:

group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-${{ github.ref_type == 'branch' && github.sha }}-${{ github.event_name == 'workflow_dispatch' }}

cancel-in-progress: true

jobs:

linux-bionic-cuda11_8-py3_10-gcc7-inductor-build:

name: cuda11.8-py3.10-gcc7-sm80

uses: ./.github/workflows/_linux-build.yml

with:

build-environment: linux-bionic-cuda11.8-py3.10-gcc7-sm80

docker-image-name: pytorch-linux-bionic-cuda11.8-cudnn8-py3-gcc7

cuda-arch-list: '8.0'

test-matrix: |

{ include: [

{ config: "inductor_huggingface_perf_compare", shard: 1, num_shards: 1, runner: "linux.gcp.a100" },

{ config: "inductor_timm_perf_compare", shard: 1, num_shards: 2, runner: "linux.gcp.a100" },

{ config: "inductor_timm_perf_compare", shard: 2, num_shards: 2, runner: "linux.gcp.a100" },

{ config: "inductor_torchbench_perf_compare", shard: 1, num_shards: 1, runner: "linux.gcp.a100" },

]}

linux-bionic-cuda11_8-py3_10-gcc7-inductor-test:

name: cuda11.8-py3.10-gcc7-sm80

uses: ./.github/workflows/_linux-test.yml

needs: linux-bionic-cuda11_8-py3_10-gcc7-inductor-build

with:

build-environment: linux-bionic-cuda11.8-py3.10-gcc7-sm80

docker-image: ${{ needs.linux-bionic-cuda11_8-py3_10-gcc7-inductor-build.outputs.docker-image }}

test-matrix: ${{ needs.linux-bionic-cuda11_8-py3_10-gcc7-inductor-build.outputs.test-matrix }}

use-gha: anything-non-empty-to-use-gha