Skip to content

Commit 5e72651

Browse files
xinhe-nvdominicshanshan
authored andcommitted
[TRTLLM-8366][feat] add kimi multi nodes case (NVIDIA#8025)
Signed-off-by: Xin He (SW-GPU) <[email protected]>
1 parent 228ad93 commit 5e72651

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

tests/integration/defs/test_e2e.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3406,7 +3406,7 @@ def test_ptp_quickstart_advanced_llama_multi_nodes(llm_root, llm_venv,
34063406
check_call(" ".join(run_cmd), shell=True, env=llm_venv._new_env)
34073407

34083408

3409-
@pytest.mark.timeout(5400)
3409+
@pytest.mark.timeout(7200)
34103410
@pytest.mark.skip_less_device_memory(80000)
34113411
@pytest.mark.skip_less_device(4)
34123412
@pytest.mark.parametrize("eval_task", ["mmlu"])
@@ -3423,6 +3423,8 @@ def test_ptp_quickstart_advanced_llama_multi_nodes(llm_root, llm_venv,
34233423
pytest.param('Qwen3/saved_models_Qwen3-235B-A22B_nvfp4_hf',
34243424
marks=skip_pre_blackwell),
34253425
pytest.param('DeepSeek-R1/DeepSeek-R1-0528-FP4', marks=skip_pre_blackwell),
3426+
pytest.param('Kimi-K2-Instruct',
3427+
marks=(skip_pre_hopper, skip_post_blackwell)),
34263428
])
34273429
def test_multi_nodes_eval(llm_venv, model_path, tp_size, pp_size, ep_size,
34283430
eval_task):

tests/integration/test_lists/qa/llm_function_multinode.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,5 +7,6 @@ test_e2e.py::test_multi_nodes_eval[llama4-models/nvidia/Llama-4-Maverick-17B-128
77
test_e2e.py::test_multi_nodes_eval[Qwen3/Qwen3-235B-A22B-tp16-mmlu]
88
test_e2e.py::test_multi_nodes_eval[Qwen3/saved_models_Qwen3-235B-A22B_nvfp4_hf-tp16-mmlu]
99
test_e2e.py::test_multi_nodes_eval[DeepSeek-R1/DeepSeek-R1-0528-FP4-tp16-mmlu]
10+
test_e2e.py::test_multi_nodes_eval[Kimi-K2-Instruct-tp16-mmlu]
1011
test_e2e.py::test_openai_disagg_multi_nodes_completion[ctx_tp2pp1-gen_tp2pp1]
1112
test_e2e.py::test_openai_disagg_multi_nodes_completion[ctx_tp1pp2-gen_tp1pp2]

0 commit comments

Comments
 (0)