Skip to content

Commit c4b9fcb

Browse files
authored
[TRTLLM-13613][test] Trim duplicated and dead multimodal accuracy tests from pre-merge CI (#15615)
Signed-off-by: Wanli Jiang <35160485+Wanli-Jiang@users.noreply.github.com>
1 parent c9b6518 commit c4b9fcb

4 files changed

Lines changed: 8 additions & 17 deletions

File tree

tests/integration/defs/accuracy/test_llm_api_pytorch_multimodal.py

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -247,10 +247,9 @@ class TestNemotron_Nano_12B_V2_VL(LlmapiAccuracyTestHarness):
247247
@pytest.mark.parametrize(
248248
"enable_chunked_prefill,max_num_tokens",
249249
[
250-
(False, MAX_NUM_TOKENS),
251250
(True, 1024),
252251
],
253-
ids=["full_budget", "forced_chunked_prefill"],
252+
ids=["forced_chunked_prefill"],
254253
)
255254
def test_auto_dtype(self, enable_chunked_prefill, max_num_tokens):
256255
with LLM(
@@ -494,10 +493,9 @@ class TestQwen3VL(LlmapiAccuracyTestHarness):
494493
@pytest.mark.parametrize(
495494
"enable_chunked_prefill,max_num_tokens",
496495
[
497-
(False, MAX_NUM_TOKENS),
498496
(True, 1024),
499497
],
500-
ids=["full_budget", "forced_chunked_prefill"],
498+
ids=["forced_chunked_prefill"],
501499
)
502500
def test_auto_dtype(self, enable_chunked_prefill, max_num_tokens):
503501
with LLM(
@@ -543,8 +541,8 @@ class TestKimiK25(LlmapiAccuracyTestHarness):
543541
@pytest.mark.skip_less_device_memory(183000)
544542
@pytest.mark.parametrize(
545543
"ep_size,attention_dp",
546-
[(1, False), (1, True), (8, False), (8, True)],
547-
ids=["tp8", "tp8_attn_dp", "ep8", "dep8"],
544+
[(8, True)],
545+
ids=["dep8"],
548546
)
549547
def test_nvfp4(self, ep_size, attention_dp):
550548
"""NVFP4 accuracy on MMMU benchmark (8x B200)."""
@@ -575,7 +573,6 @@ def test_nvfp4(self, ep_size, attention_dp):
575573
class TestMistralSmall24B(LlmapiAccuracyTestHarness):
576574
MODEL_NAME = "mistralai/Mistral-Small-3.1-24B-Instruct-2503"
577575
MODEL_PATH = f"{llm_models_root()}/Mistral-Small-3.1-24B-Instruct-2503"
578-
MAX_NUM_TOKENS = 16384
579576

580577
# NOTE: MMMU adds <|endoftext|> to the stop token.
581578
sampling_params = SamplingParams(
@@ -588,10 +585,9 @@ class TestMistralSmall24B(LlmapiAccuracyTestHarness):
588585
@pytest.mark.parametrize(
589586
"max_num_tokens",
590587
[
591-
MAX_NUM_TOKENS,
592588
1024,
593589
],
594-
ids=["full_budget", "forced_chunked_prefill"],
590+
ids=["forced_chunked_prefill"],
595591
)
596592
def test_auto_dtype(self, max_num_tokens):
597593
kv_cache_config = KvCacheConfig(free_gpu_memory_fraction=0.75)
@@ -742,7 +738,7 @@ class TestNanoV3Omni(LlmapiAccuracyTestHarness):
742738
),
743739
128,
744740
QuantAlgo.MIXED_PRECISION,
745-
(MMMU_TASK_SPEC, VOXPOPULI_TASK_SPEC, VIDEOMME_TASK_SPEC),
741+
(MMMU_TASK_SPEC, VOXPOPULI_TASK_SPEC),
746742
None,
747743
marks=(skip_pre_blackwell,),
748744
id="nvfp4",

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,6 @@ l0_b200:
5454
- accuracy/test_llm_api_pytorch.py::TestQwen3_30B_A3B::test_w4a16_mxfp4[latency-TRTLLM]
5555
- accuracy/test_llm_api_pytorch.py::TestQwen3_30B_A3B_Instruct_2507::test_skip_softmax_attention[target_sparsity_0.9-fp8kv=True]
5656
- accuracy/test_llm_api_pytorch.py::TestQwen3_5_35B_A3B::test_fp8[enable_block_reuse=True]
57-
- accuracy/test_llm_api_pytorch_multimodal.py::TestNanoV3Omni::test_auto_dtype[nvfp4]
5857
- accuracy/test_llm_api_pytorch_multimodal.py::TestNanoV3Omni::test_auto_dtype[fp8_mmmu_encoder_cuda_graph]
5958
- accuracy/test_epd_disagg_multimodal.py::TestVideoMMEEPD::test_disaggregated_videomme[qwen3vl_2b_instruct]
6059
- accuracy/test_epd_disagg_multimodal.py::TestVideoMMEEPD::test_disaggregated_videomme[nemotron_nano_v3_omni_nvfp4]
@@ -355,6 +354,7 @@ l0_b200:
355354
- accuracy/test_llm_api_pytorch.py::TestGPTOSS::test_w4_1gpu[v1_kv_cache-True-True-trtllm-auto]
356355
- accuracy/test_llm_api_pytorch.py::TestGPTOSS::test_w4_1gpu[v2_kv_cache-True-True-trtllm-auto]
357356
- accuracy/test_llm_api_pytorch_multimodal.py::TestNanoV3Omni::test_auto_dtype[bf16]
357+
- accuracy/test_llm_api_pytorch_multimodal.py::TestNanoV3Omni::test_auto_dtype[nvfp4]
358358
# ------------- VisualGen single-GPU tests ---------------
359359
- examples/visual_gen/test_visual_gen.py::test_visual_gen_quickstart
360360
- examples/visual_gen/test_visual_gen.py::test_visual_gen_api_walkthrough

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,6 @@ l0_dgx_b200:
164164
- accuracy/test_llm_api_pytorch.py::TestDeepSeekV32::test_dsa_host_cache_offload[host_cache_offload_mtp3_no_adp] TIMEOUT (60)
165165
- accuracy/test_disaggregated_serving.py::TestDeepSeekV32Exp::test_auto_dtype[False] TIMEOUT (60)
166166
- accuracy/test_llm_api_pytorch.py::TestKimiK25::test_nvfp4[tp8] TIMEOUT (60)
167-
- accuracy/test_llm_api_pytorch_multimodal.py::TestKimiK25::test_nvfp4[dep8] TIMEOUT (120)
168167
- accuracy/test_llm_api_pytorch.py::TestNemotronV3Super::test_nvfp4_8gpus_mtp TIMEOUT (60)
169168
- accuracy/test_llm_api_pytorch.py::TestNemotronV3Super::test_nvfp4_8gpus_mtp_custom_op TIMEOUT (60)
170169
- accuracy/test_llm_api_pytorch.py::TestNemotronV3Super::test_nvfp4_8gpus[attention_dp_on-trtllm] TIMEOUT (60)
@@ -244,6 +243,7 @@ l0_dgx_b200:
244243
- accuracy/test_llm_api_pytorch.py::TestKimiK25::test_nvfp4[tp8_attn_dp] TIMEOUT (60)
245244
- accuracy/test_llm_api_pytorch.py::TestKimiK25::test_nvfp4[ep8] TIMEOUT (60)
246245
- accuracy/test_llm_api_pytorch.py::TestKimiK25::test_nvfp4[dep8] TIMEOUT (60)
246+
- accuracy/test_llm_api_pytorch_multimodal.py::TestKimiK25::test_nvfp4[dep8] TIMEOUT (120)
247247
- accuracy/test_llm_api_pytorch.py::TestMistralLarge3_675B::test_fp8[latency_moe_deepgemm] TIMEOUT (60)
248248
- accuracy/test_llm_api_pytorch.py::TestNemotronV3Super::test_nvfp4_parallelism[TP8_PP1] TIMEOUT (60)
249249
- test_e2e.py::test_deepseek_r1_mtp_bench TIMEOUT(60) # Cover https://nvbugs/5670108

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

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,6 @@ l0_l40s:
2727
- unittest/_torch/modeling/test_modeling_step3p7vl.py
2828
- test_e2e.py::test_ptp_scaffolding[DeepSeek-R1-Distill-Qwen-7B-DeepSeek-R1/DeepSeek-R1-Distill-Qwen-7B]
2929
- unittest/llmapi/apps/_test_openai_chat_multimodal.py::test_single_chat_session_image_embeds -m needs_l40s
30-
# MMMU sanity check
31-
- accuracy/test_llm_api_pytorch_multimodal.py::TestQwen2_5_VL_7B::test_auto_dtype
32-
- accuracy/test_llm_api_pytorch_multimodal.py::TestVILA1_5_3B::test_auto_dtype
3330
# AutoDeploy: Nemotron-Nano-V3 on Ada uses flashinfer; trtllm has no
3431
# (E4M3 input, BF16 output, paged_kv, head_dim=128, sm_89) FMHA cubin.
3532
- accuracy/test_llm_api_autodeploy.py::TestNemotronNanoV3::test_accuracy[fp8-1-attn_dp_off-flashinfer]
@@ -85,8 +82,6 @@ l0_l40s:
8582
tests:
8683
- examples/test_gpt.py::test_llm_gpt2_next_prompt_tuning[use_cpp_session-tp1] # 10 mins
8784
- examples/test_gpt.py::test_llm_gpt2_next_prompt_tuning[use_py_session-tp1]
88-
- examples/test_multimodal.py::test_llm_multimodal_general[video-neva-pp:1-tp:1-bfloat16-bs:1-cpp_e2e:False-nb:1]
89-
- examples/test_multimodal.py::test_llm_multimodal_general[kosmos-2-pp:1-tp:1-float16-bs:1-cpp_e2e:True-nb:1]
9085
- examples/test_nemotron.py::test_llm_nemotron_3_8b_1gpu[bfloat16-fp8] # 18mins
9186
- examples/test_whisper.py::test_llm_whisper_general[large-v3-disable_gemm_plugin-disable_attention_plugin-disable_weight_only-float16-nb:1-use_python_runtime] # 8 mins
9287
- unittest/trt/attention/test_bert_attention.py # 12 mins

0 commit comments

Comments
 (0)