Skip to content

Commit bc25dcc

Browse files
committed
Add dsr1 and gpt-oss test cases
Signed-off-by: Chenfei Zhang <[email protected]>
1 parent 7a05ee2 commit bc25dcc

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

tests/integration/test_lists/test-db/perf_sanity_l0_dgx_b200.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ perf_sanity_l0_dgx_b200:
1515
backend: pytorch
1616
orchestrator: mpi
1717
tests:
18-
- perf/test_perf.py::test_perf[perf_sanity_upload-l0_dgx_b200-r1_fp8_dep4_mtp1,r1_fp8_tep4_mtp3,gpt_oss_fp4_eagle3_tp8]
18+
- perf/test_perf.py::test_perf[perf_sanity_upload-l0_dgx_b200-r1_fp8_dep8_mtp1,r1_fp8_tep8_mtp3,gpt_oss_fp4_tp8_eagle3]
1919

2020
- condition:
2121
ranges:
@@ -32,7 +32,7 @@ perf_sanity_l0_dgx_b200:
3232
backend: pytorch
3333
orchestrator: mpi
3434
tests:
35-
- perf/test_perf.py::test_perf[perf_sanity_upload-l0_dgx_b200-r1_fp8_dep4_mtp1,r1_fp8_tep4_mtp3,gpt_oss_fp4_eagle3_tp8]
35+
- perf/test_perf.py::test_perf[perf_sanity_upload-l0_dgx_b200-r1_fp8_dep8_mtp1,r1_fp8_tep8_mtp3,gpt_oss_fp4_tp8_eagle3]
3636

3737
- condition:
3838
ranges:

tests/scripts/perf-sanity/l0_dgx_b200.yaml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
server_configs:
2-
- name: "r1_fp8_dep4_mtp1"
2+
- name: "r1_fp8_dep8_mtp1"
33
model_name: "deepseek_r1_0528_fp8"
44
gpus: 8
55
tp: 8
@@ -29,11 +29,11 @@ server_configs:
2929
backend: "openai"
3030
use_chat_template: true
3131

32-
- name: "r1_fp8_tep4_mtp3"
32+
- name: "r1_fp8_tep8_mtp3"
3333
model_name: "deepseek_r1_0528_fp8"
3434
gpus: 8
3535
tp: 8
36-
ep: 1
36+
ep: 8
3737
pp: 1
3838
attention_backend: "TRTLLM"
3939
moe_backend: "TRTLLM"
@@ -90,7 +90,7 @@ server_configs:
9090
model_name: "deepseek_r1_0528_fp4_v2"
9191
gpus: 4
9292
tp: 4
93-
ep: 1
93+
ep: 4
9494
pp: 1
9595
attention_backend: "TRTLLM"
9696
moe_backend: "TRTLLM"
@@ -113,7 +113,7 @@ server_configs:
113113
backend: "openai"
114114
use_chat_template: true
115115

116-
- name: "gpt_oss_fp4_eagle3_tp8"
116+
- name: "gpt_oss_fp4_tp8_eagle3"
117117
model_name: "gpt_oss_120b_fp4"
118118
gpus: 8
119119
tp: 8

0 commit comments

Comments
 (0)