kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-13 22:18:36 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[None][test] update test case constraint (#10381)

Browse Source 
Signed-off-by: Ivy Zhang <25222398+crazydemo@users.noreply.github.com>
This commit is contained in:
Ivy Zhang 2026-01-06 12:28:59 +08:00 committed by GitHub
parent 1b1058279c
commit 22a1d31a27
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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

@ -2269,6 +2269,7 @@ class TestDeepSeekR1(LlmapiAccuracyTestHarness):
torch.cuda.empty_cache()
@skip_pre_blackwell
@pytest.mark.skip_less_device_memory(95000)
@pytest.mark.parametrize(
"tp_size,pp_size,ep_size,mtp_nextn,fp8kv,attention_dp,cuda_graph,overlap_scheduler,max_batch_size,moe_backend",
[