Skip to content

Commit e1c9aa7

Browse files
[None][chore] AutoDeploy: Add the Nemotron MOE to CI (#9328)
Signed-off-by: Chenghao Zhang <[email protected]> Co-authored-by: Suyog Gupta <[email protected]>
1 parent 0582e54 commit e1c9aa7

File tree

3 files changed

+4
-0
lines changed

3 files changed

+4
-0
lines changed

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,7 @@ l0_b200:
8383
# ------------- AutoDeploy tests ---------------
8484
- unittest/_torch/auto_deploy/unit/singlegpu -k "not test_trtllm_bench_backend_comparison"
8585
- accuracy/test_llm_api_autodeploy.py::TestLlama3_1_8B::test_auto_dtype[False-1]
86+
- accuracy/test_llm_api_autodeploy.py::TestNemotronMOE::test_fp8
8687
- unittest/_torch/auto_deploy/unit/singlegpu
8788
- condition:
8889
ranges:

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -127,6 +127,7 @@ l0_dgx_h200:
127127
- disaggregated/test_disaggregated.py::test_disaggregated_benchmark_on_diff_backends[DeepSeek-V3-Lite-fp8]
128128
# ------------- AutoDeploy tests ---------------
129129
- accuracy/test_llm_api_autodeploy.py::TestLlama3_1_8B::test_auto_dtype[False-4]
130+
- accuracy/test_llm_api_autodeploy.py::TestNemotronMOE::test_bf16
130131
- condition:
131132
ranges:
132133
system_gpu_count:

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

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -110,6 +110,8 @@ l0_h100:
110110
- accuracy/test_llm_api_autodeploy.py::TestLlama3_1_8B::test_auto_dtype[True-1]
111111
- accuracy/test_llm_api_autodeploy.py::TestNemotronH::test_auto_dtype[False]
112112
- accuracy/test_llm_api_autodeploy.py::TestNemotronH::test_auto_dtype[True]
113+
- accuracy/test_llm_api_autodeploy.py::TestNemotronMOE::test_fp8
114+
- accuracy/test_llm_api_autodeploy.py::TestNemotronMOE::test_bf16
113115
- condition:
114116
ranges:
115117
system_gpu_count:

0 commit comments

Comments
 (0)