Skip to content

Commit 1105aea

Browse files
functionstackxcquil11
functionstackx
authored and
cquil11
committed
remove: llama 70b
1 parent 69844b2 commit 1105aea

8 files changed

Lines changed: 805 additions & 375 deletions

‎.github/workflows/full-sweep-tmpl.yml‎

Lines changed: 188 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,188 @@
1+
name: Template - Full Sweep
2+
3+
on:
4+
workflow_call:
5+
inputs:
6+
run_1k1k:
7+
type: boolean
8+
required: true
9+
run_8k1k:
10+
type: boolean
11+
required: true
12+
run_1k8k:
13+
type: boolean
14+
required: true
15+
16+
use_h100:
17+
type: boolean
18+
required: true
19+
use_h200:
20+
type: boolean
21+
required: true
22+
use_b200:
23+
type: boolean
24+
required: true
25+
use_mi300x:
26+
type: boolean
27+
required: true
28+
use_mi325x:
29+
type: boolean
30+
required: true
31+
use_mi355x:
32+
type: boolean
33+
required: true
34+
use_gb200:
35+
type: boolean
36+
required: false
37+
default: false
38+
39+
jobs:
40+
dsr1-1k1k:
41+
if: ${{ inputs.run_1k1k }}
42+
uses: ./.github/workflows/dsr1-tmpl.yml
43+
secrets: inherit
44+
with:
45+
exp-name: 'dsr1_1k1k'
46+
isl: 1024
47+
osl: 1024
48+
max-model-len: 2048
49+
random-range-ratio: 0.8
50+
use_h200: ${{ inputs.use_h200 }}
51+
use_b200: ${{ inputs.use_b200 }}
52+
use_mi300x: ${{ inputs.use_mi300x }}
53+
use_mi325x: ${{ inputs.use_mi325x }}
54+
use_mi355x: ${{ inputs.use_mi355x }}
55+
use_gb200: ${{ inputs.use_gb200 }}
56+
57+
collect-dsr1-1k1k-results:
58+
needs: dsr1-1k1k
59+
if: ${{ inputs.run_1k1k && always() }}
60+
uses: ./.github/workflows/collect-results.yml
61+
secrets: inherit
62+
with:
63+
exp-name: 'dsr1_1k1k'
64+
65+
gptoss-1k1k:
66+
if: ${{ inputs.run_1k1k }}
67+
uses: ./.github/workflows/gptoss-tmpl.yml
68+
secrets: inherit
69+
with:
70+
exp-name: 'gptoss_1k1k'
71+
isl: 1024
72+
osl: 1024
73+
max-model-len: 2048
74+
random-range-ratio: 0.8
75+
use_h100: ${{ inputs.use_h100 }}
76+
use_h200: ${{ inputs.use_h200 }}
77+
use_b200: ${{ inputs.use_b200 }}
78+
use_mi300x: ${{ inputs.use_mi300x }}
79+
use_mi325x: ${{ inputs.use_mi325x }}
80+
use_mi355x: ${{ inputs.use_mi355x }}
81+
82+
collect-gptoss-1k1k-results:
83+
needs: gptoss-1k1k
84+
if: ${{ inputs.run_1k1k && always() }}
85+
uses: ./.github/workflows/collect-results.yml
86+
secrets: inherit
87+
with:
88+
exp-name: 'gptoss_1k1k'
89+
90+
dsr1-8k1k:
91+
if: ${{ inputs.run_8k1k }}
92+
uses: ./.github/workflows/dsr1-tmpl.yml
93+
secrets: inherit
94+
with:
95+
exp-name: 'dsr1_8k1k'
96+
isl: 8192
97+
osl: 1024
98+
max-model-len: 9216
99+
random-range-ratio: 0.8
100+
use_h200: ${{ inputs.use_h200 }}
101+
use_b200: ${{ inputs.use_b200 }}
102+
use_mi300x: ${{ inputs.use_mi300x }}
103+
use_mi325x: ${{ inputs.use_mi325x }}
104+
use_mi355x: ${{ inputs.use_mi355x }}
105+
use_gb200: ${{ inputs.use_gb200 }}
106+
107+
collect-dsr1-8k1k-results:
108+
needs: dsr1-8k1k
109+
if: ${{ inputs.run_8k1k && always() }}
110+
uses: ./.github/workflows/collect-results.yml
111+
secrets: inherit
112+
with:
113+
exp-name: 'dsr1_8k1k'
114+
115+
gptoss-8k1k:
116+
if: ${{ inputs.run_8k1k }}
117+
uses: ./.github/workflows/gptoss-tmpl.yml
118+
secrets: inherit
119+
with:
120+
exp-name: 'gptoss_8k1k'
121+
isl: 8192
122+
osl: 1024
123+
max-model-len: 9216
124+
random-range-ratio: 0.8
125+
use_h100: ${{ inputs.use_h100 }}
126+
use_h200: ${{ inputs.use_h200 }}
127+
use_b200: ${{ inputs.use_b200 }}
128+
use_mi300x: ${{ inputs.use_mi300x }}
129+
use_mi325x: ${{ inputs.use_mi325x }}
130+
use_mi355x: ${{ inputs.use_mi355x }}
131+
132+
collect-gptoss-8k1k-results:
133+
needs: gptoss-8k1k
134+
if: ${{ inputs.run_8k1k && always() }}
135+
uses: ./.github/workflows/collect-results.yml
136+
secrets: inherit
137+
with:
138+
exp-name: 'gptoss_8k1k'
139+
140+
dsr1-1k8k:
141+
if: ${{ inputs.run_1k8k }}
142+
uses: ./.github/workflows/dsr1-tmpl.yml
143+
secrets: inherit
144+
with:
145+
exp-name: 'dsr1_1k8k'
146+
isl: 1024
147+
osl: 8192
148+
max-model-len: 9216
149+
random-range-ratio: 0.8
150+
use_h200: ${{ inputs.use_h200 }}
151+
use_b200: ${{ inputs.use_b200 }}
152+
use_mi300x: ${{ inputs.use_mi300x }}
153+
use_mi325x: ${{ inputs.use_mi325x }}
154+
use_mi355x: ${{ inputs.use_mi355x }}
155+
use_gb200: ${{ inputs.use_gb200 }}
156+
157+
collect-dsr1-1k8k-results:
158+
needs: dsr1-1k8k
159+
if: ${{ inputs.run_1k8k && always() }}
160+
uses: ./.github/workflows/collect-results.yml
161+
secrets: inherit
162+
with:
163+
exp-name: 'dsr1_1k8k'
164+
165+
gptoss-1k8k:
166+
if: ${{ inputs.run_1k8k }}
167+
uses: ./.github/workflows/gptoss-tmpl.yml
168+
secrets: inherit
169+
with:
170+
exp-name: 'gptoss_1k8k'
171+
isl: 1024
172+
osl: 8192
173+
max-model-len: 9216
174+
random-range-ratio: 0.8
175+
use_h100: ${{ inputs.use_h100 }}
176+
use_h200: ${{ inputs.use_h200 }}
177+
use_b200: ${{ inputs.use_b200 }}
178+
use_mi300x: ${{ inputs.use_mi300x }}
179+
use_mi325x: ${{ inputs.use_mi325x }}
180+
use_mi355x: ${{ inputs.use_mi355x }}
181+
182+
collect-gptoss-1k8k-results:
183+
needs: gptoss-1k8k
184+
if: ${{ inputs.run_1k8k && always() }}
185+
uses: ./.github/workflows/collect-results.yml
186+
secrets: inherit
187+
with:
188+
exp-name: 'gptoss_1k8k'

0 commit comments

Comments
 (0)