@@ -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):
575573class 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" ,
0 commit comments