Skip to content

Commit 8e8e754

Browse files
saiendurilifuhuang
authored andcommitted
Enable MI325X AMD CI. (sgl-project#6259)
1 parent c95a83f commit 8e8e754

File tree

2 files changed

+28
-7
lines changed

2 files changed

+28
-7
lines changed

.github/workflows/nightly-test-amd.yml

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,10 @@ concurrency:
1717
jobs:
1818
nightly-test:
1919
if: github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request'
20-
runs-on: linux-mi300-gpu-2
20+
strategy:
21+
matrix:
22+
runner: [linux-mi300-gpu-2, linux-mi325-gpu-2-nightly]
23+
runs-on: ${{matrix.runner}}
2124
steps:
2225
- name: Checkout code
2326
uses: actions/checkout@v4

.github/workflows/pr-test-amd.yml

Lines changed: 24 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,10 @@ jobs:
2525
accuracy-test-1-gpu-amd:
2626
if: (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') &&
2727
github.event.pull_request.draft == false
28-
runs-on: linux-mi300-gpu-1
28+
strategy:
29+
matrix:
30+
runner: [linux-mi300-gpu-1, linux-mi325-gpu-1]
31+
runs-on: ${{matrix.runner}}
2932
steps:
3033
- name: Checkout code
3134
uses: actions/checkout@v4
@@ -65,7 +68,10 @@ jobs:
6568
accuracy-test-2-gpu-amd:
6669
if: (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') &&
6770
github.event.pull_request.draft == false
68-
runs-on: linux-mi300-gpu-2
71+
strategy:
72+
matrix:
73+
runner: [linux-mi300-gpu-2, linux-mi325-gpu-2]
74+
runs-on: ${{matrix.runner}}
6975
steps:
7076
- name: Checkout code
7177
uses: actions/checkout@v4
@@ -103,7 +109,10 @@ jobs:
103109
mla-test-1-gpu-amd:
104110
if: (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') &&
105111
github.event.pull_request.draft == false
106-
runs-on: linux-mi300-gpu-1
112+
strategy:
113+
matrix:
114+
runner: [linux-mi300-gpu-1, linux-mi325-gpu-1]
115+
runs-on: ${{matrix.runner}}
107116
steps:
108117
- name: Checkout code
109118
uses: actions/checkout@v4
@@ -141,7 +150,10 @@ jobs:
141150
performance-test-1-gpu-part-1-amd:
142151
if: (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') &&
143152
github.event.pull_request.draft == false
144-
runs-on: linux-mi300-gpu-1
153+
strategy:
154+
matrix:
155+
runner: [linux-mi300-gpu-1, linux-mi325-gpu-1]
156+
runs-on: ${{matrix.runner}}
145157
steps:
146158
- name: Checkout code
147159
uses: actions/checkout@v4
@@ -200,7 +212,10 @@ jobs:
200212
performance-test-1-gpu-part-2-amd:
201213
if: (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') &&
202214
github.event.pull_request.draft == false
203-
runs-on: linux-mi300-gpu-1
215+
strategy:
216+
matrix:
217+
runner: [linux-mi300-gpu-1, linux-mi325-gpu-1]
218+
runs-on: ${{matrix.runner}}
204219
steps:
205220
- name: Checkout code
206221
uses: actions/checkout@v4
@@ -248,7 +263,10 @@ jobs:
248263
bench-test-2-gpu-amd:
249264
if: (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') &&
250265
github.event.pull_request.draft == false
251-
runs-on: linux-mi300-gpu-2
266+
strategy:
267+
matrix:
268+
runner: [linux-mi300-gpu-2, linux-mi325-gpu-2]
269+
runs-on: ${{matrix.runner}}
252270
steps:
253271
- name: Checkout code
254272
uses: actions/checkout@v4

0 commit comments

Comments
 (0)