graph-compiler-playground/.github/workflows/test_amd.yml at main · intel/graph-compiler-playground · GitHub

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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
name: AMD full run

on:
    workflow_dispatch:
        inputs:
            tag:
                description: tag to label this run in DB
                required: true
                default: "test"
            runner_type:
                description: Type of runner to use
                required: true
                default: genoa60c
                type: choice
                options:
                    - amd32c
                    - amd64c
                    - genoa60c
            shutdown_cloud_runner:
                description: Shutdown AWS cluster
                default: true
                type: boolean
            test_scripts:
                description: Test scripts to execute
                required: false
                default: '["./mlp.sh", "./cnn.sh", "./llm.sh"]'
                type: string

jobs:
    print_inputs:
        runs-on: Linux
        steps:
            - name: Print Inputs
              run: echo "${{ toJSON(github.event.inputs) }}"

    mlp_test:
        strategy:
            matrix:
                type: [
                  {device: 'cpu', compiler: 'torch'},
                  {device: 'cpu', compiler: 'dynamo'},
                  {device: 'cpu', compiler: 'ipex'},
                ]
                test_script: ${{ fromJson(inputs.test_scripts) }}
            fail-fast: false
        uses: ./.github/workflows/execute-test-script.yml
        with:
            compiler: ${{ matrix.type.compiler }}
            device: ${{ matrix.type.device }}
            tag: ${{ inputs.tag }}
            torch_mlir_repo: ${{ inputs.torch_mlir_repo }}
            torch_mlir_branch: ${{ inputs.torch_mlir_branch }}
            runner_type: ${{ inputs.runner_type }}
            test_script: ${{ matrix.test_script }}
        secrets:
            DB_URL: ${{ secrets.DB_URL }}
            HF_TOKEN: ${{ secrets.HF_TOKEN }}

    shutdown:
        needs: mlp_test
        if: inputs.shutdown_cloud_runner
        runs-on: ${{ inputs.runner_type }}
        steps:
            - name: shutdown
              shell: bash -el {0}
              run: sudo shutdown -h +2