From 2c0188cc782236d609f410f0ce4d95ebe54ce95e Mon Sep 17 00:00:00 2001 From: Liangsheng Yin Date: Sat, 8 Aug 2026 02:04:16 -0700 Subject: [PATCH] [Fix] Give the piecewise CUDA graph test stub an `hf_config` (#34100) --- .../unit/configs/test_multimodal_piecewise_cuda_graph.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/test/registered/unit/configs/test_multimodal_piecewise_cuda_graph.py b/test/registered/unit/configs/test_multimodal_piecewise_cuda_graph.py index ff71e1848..77b19500b 100644 --- a/test/registered/unit/configs/test_multimodal_piecewise_cuda_graph.py +++ b/test/registered/unit/configs/test_multimodal_piecewise_cuda_graph.py @@ -95,8 +95,11 @@ class TestMultimodalPiecewiseCudaGraph(CustomTestCase): def test_trtllm_mla_stays_on_breakable_and_is_disabled_by_compatibility(self): args = ServerArgs(model_path="dummy") + # The MLA rule reads hf_config to exempt DSA models, so the stub needs + # an architecture that is MLA but not DSA. args.model_config = SimpleNamespace( - is_multimodal_piecewise_cuda_graph_supported=True + is_multimodal_piecewise_cuda_graph_supported=True, + hf_config=SimpleNamespace(architectures=["DeepseekV2ForCausalLM"]), ) args.cuda_graph_config = CudaGraphConfig( prefill=PhaseConfig(backend=Backend.BREAKABLE)