kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-02-16 07:53:55 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[https://nvbugs/5839569][test] update test constraint (#11054)

Browse Source 
Signed-off-by: Ivy Zhang <25222398+crazydemo@users.noreply.github.com>
Signed-off-by: Wangshanshan <30051912+dominicshanshan@users.noreply.github.com>
This commit is contained in:
Ivy Zhang 2026-01-28 17:57:30 +08:00 committed by Yanchao Lu
parent 03b635bb08
commit 9384cf8458
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/integration/defs/accuracy/test_llm_api_pytorch.py
View File

@ -3133,6 +3133,7 @@ class TestKimiK2(LlmapiAccuracyTestHarness):
@pytest.mark.skip_less_mpi_world_size(8)
@skip_post_blackwell
@skip_pre_hopper
@pytest.mark.skip_less_device_memory(180000)
@pytest.mark.parametrize(
"tp_size,pp_size,ep_size,fp8kv,attention_dp,cuda_graph,overlap_scheduler,max_batch_size",
[(8, 1, 8, False, False, True, True, 16)],