|
12 | 12 | required: true |
13 | 13 | type: string |
14 | 14 |
|
15 | | - runner: |
16 | | - description: "Specific runner node to run on" |
17 | | - required: false |
18 | | - type: choice |
19 | | - options: |
20 | | - - "h100-cr_0" |
21 | | - - "h100-cr_1" |
22 | | - - "h100-cw_0" |
23 | | - - "h100-cw_1" |
24 | | - - "h200-cw_0" |
25 | | - - "h200-cw_1" |
26 | | - - "h200-nb_0" |
27 | | - - "h200-nb_1" |
28 | | - - "h200-nb_2" |
29 | | - - "h200-nb_3" |
30 | | - - "h200-nv_0" |
31 | | - - "h200-nv_1" |
32 | | - - "h200-nv_2" |
33 | | - - "h200-nv_3" |
34 | | - - "b200-nv_0" |
35 | | - - "b200-nv_1" |
36 | | - - "b200-nb_0" |
37 | | - - "b200-nb_1" |
38 | | - - "b200-nvd_0" |
39 | | - - "b200-nvd_1" |
40 | | - - "b200-nvd_2" |
41 | | - - "b200-nvd_3" |
42 | | - - "b200-tg_0" |
43 | | - - "mi300x-amd_0" |
44 | | - - "mi300x-amd_1" |
45 | | - - "mi300x-amd_2" |
46 | | - - "mi300x-amd_3" |
47 | | - - "mi300x-amd_4" |
48 | | - - "mi300x-cr_0" |
49 | | - - "mi300x-oci_0" |
50 | | - - "mi325x-amd_0" |
51 | | - - "mi325x-tw_0" |
52 | | - - "mi325x-tw_1" |
53 | | - - "mi325x-tw_2" |
54 | | - - "mi325x-tw_3" |
55 | | - - "mi355x-amd_0" |
56 | | - - "mi355x-amd_1" |
57 | | - - "mi355x-amd_2" |
58 | | - - "mi355x-amd_3" |
59 | | - |
60 | 15 | jobs: |
61 | | - # verify-compatible-runner: |
62 | | - # runs-on: ubuntu-latest |
63 | | - # if: ${{ inputs.runner != '' }} |
64 | | - # steps: |
65 | | - # - name: Verify runner compatible |
66 | | - # shell: python |
67 | | - # run: | |
68 | | - # import re |
69 | | - |
70 | | - # inputs_name_re = re.match(r'^[^-]+-[^-]+-([^-]+)', '${{ inputs.name }}') |
71 | | - # if inputs_name_re: |
72 | | - # config_gpu = inputs_name_re.group(1) |
73 | | - # inputs_runner_re = re.match(r'^([^-]+)', '${{ inputs.runner }}') |
74 | | - # if inputs_runner_re: |
75 | | - # runner_gpu = inputs_runner_re.group(1) |
76 | | - |
77 | | - # assert config_gpu == runner_gpu, f"Specified runner '${{ inputs.runner }})' is not compatible with config '${{ inputs.name }}'" |
78 | | - |
79 | | - |
80 | 16 | get-jobs: |
81 | 17 | runs-on: ubuntu-latest |
82 | | - # needs: verify-compatible-runner |
83 | | - # if: ${{ always() && (needs.verify-compatible-runner.result == 'success' || needs.verify-compatible-runner.result == 'skipped') }} |
84 | 18 | outputs: |
85 | 19 | search-space-config: ${{ steps.get-jobs.outputs.search-space-config }} |
86 | 20 | steps: |
|
89 | 23 |
|
90 | 24 | - id: get-jobs |
91 | 25 | run: | |
92 | | - CONFIG_JSON=$(python3 ${GITHUB_WORKSPACE}/utils/matrix-logic/generate_sweep_configs.py test-config --test-mode ${{ inputs.generate-cli-command }}) |
| 26 | + CONFIG_JSON=$(python3 ${GITHUB_WORKSPACE}/utils/matrix-logic/generate_sweep_configs.py ${{ inputs.generate-cli-command }}) |
93 | 27 | echo "search-space-config=$CONFIG_JSON" >> $GITHUB_OUTPUT |
94 | 28 |
|
95 | 29 | test-sweep: |
|
106 | 40 | isl: ${{ matrix.config.isl }} |
107 | 41 | osl: ${{ matrix.config.osl }} |
108 | 42 | max-model-len: ${{ matrix.config.max-model-len }} |
109 | | - runner: ${{ inputs.runner != '' && inputs.runner || matrix.config.runner }} |
| 43 | + runner: ${{ matrix.config.runner }} |
110 | 44 | image: ${{ matrix.config.image }} |
111 | 45 | model: ${{ matrix.config.model }} |
112 | 46 | framework: ${{ matrix.config.framework }} |
|
0 commit comments