Skip to content

Commit f439163

Browse files
committed
adding more workflows
1 parent 8510c0a commit f439163

3 files changed

Lines changed: 250 additions & 134 deletions

File tree

.github/configs/runners.yaml

Lines changed: 48 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,48 @@
1+
h100:
2+
- 'h100-cr_0'
3+
- 'h100-cr_1'
4+
- 'h100-cw_0'
5+
- 'h100-cw_1'
6+
h200:
7+
- 'h200-cw_0'
8+
- 'h200-cw_1'
9+
- 'h200-nb_0'
10+
- 'h200-nb_1'
11+
- 'h200-nb_2'
12+
- 'h200-nb_3'
13+
- 'h200-nv_0'
14+
- 'h200-nv_1'
15+
- 'h200-nv_2'
16+
- 'h200-nv_3'
17+
b200-trt:
18+
- 'b200-nv_0'
19+
- 'b200-nv_1'
20+
b200:
21+
- 'b200-nb_0'
22+
- 'b200-nb_1'
23+
- 'b200-nvd_0'
24+
- 'b200-nvd_1'
25+
- 'b200-nvd_2'
26+
- 'b200-nvd_3'
27+
- 'b200-tg_0'
28+
mi300x:
29+
- 'mi300x-amd_0'
30+
- 'mi300x-amd_1'
31+
- 'mi300x-amd_2'
32+
- 'mi300x-amd_3'
33+
- 'mi300x-amd_4'
34+
- 'mi300x-cr_0'
35+
- 'mi300x-oci_0'
36+
mi325x:
37+
- 'mi325x-amd_0'
38+
- 'mi325x-tw_0'
39+
- 'mi325x-tw_1'
40+
- 'mi325x-tw_2'
41+
- 'mi325x-tw_3'
42+
mi355x:
43+
- 'mi355x-amd_0'
44+
- 'mi355x-amd_1'
45+
- 'mi355x-amd_2'
46+
- 'mi355x-amd_3'
47+
gb200:
48+
- gb200-nv_0

.github/workflows/test.yml

Lines changed: 2 additions & 68 deletions
Original file line numberDiff line numberDiff line change
@@ -12,75 +12,9 @@ on:
1212
required: true
1313
type: string
1414

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-
6015
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-
8016
get-jobs:
8117
runs-on: ubuntu-latest
82-
# needs: verify-compatible-runner
83-
# if: ${{ always() && (needs.verify-compatible-runner.result == 'success' || needs.verify-compatible-runner.result == 'skipped') }}
8418
outputs:
8519
search-space-config: ${{ steps.get-jobs.outputs.search-space-config }}
8620
steps:
@@ -89,7 +23,7 @@ jobs:
8923

9024
- id: get-jobs
9125
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 }})
9327
echo "search-space-config=$CONFIG_JSON" >> $GITHUB_OUTPUT
9428
9529
test-sweep:
@@ -106,7 +40,7 @@ jobs:
10640
isl: ${{ matrix.config.isl }}
10741
osl: ${{ matrix.config.osl }}
10842
max-model-len: ${{ matrix.config.max-model-len }}
109-
runner: ${{ inputs.runner != '' && inputs.runner || matrix.config.runner }}
43+
runner: ${{ matrix.config.runner }}
11044
image: ${{ matrix.config.image }}
11145
model: ${{ matrix.config.model }}
11246
framework: ${{ matrix.config.framework }}

0 commit comments

Comments
 (0)