From 573c075fef769908952bee069a9b1aa80474501d Mon Sep 17 00:00:00 2001 From: Mick Date: Sat, 18 Jul 2026 18:48:05 +0800 Subject: [PATCH] CI: synchronize prefill graph test fixtures (#31665) --- .../unit/configs/test_multimodal_piecewise_cuda_graph.py | 1 + .../model_executor/runner/test_prefill_cuda_graph_padding.py | 3 +++ 2 files changed, 4 insertions(+) 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 11371393e..b770ecd09 100644 --- a/test/registered/unit/configs/test_multimodal_piecewise_cuda_graph.py +++ b/test/registered/unit/configs/test_multimodal_piecewise_cuda_graph.py @@ -33,6 +33,7 @@ class TestMultimodalPiecewiseCudaGraph(CustomTestCase): runner.prefill_backend_name = backend runner.has_mha_companion_layers = backend == Backend.BREAKABLE runner.capture_hidden_mode = CaptureHiddenMode.NULL + runner.capture_num_tokens = [4, 16] runner.max_num_tokens = 16 return runner diff --git a/test/registered/unit/model_executor/runner/test_prefill_cuda_graph_padding.py b/test/registered/unit/model_executor/runner/test_prefill_cuda_graph_padding.py index 8f8d80fa9..2e0baefd4 100644 --- a/test/registered/unit/model_executor/runner/test_prefill_cuda_graph_padding.py +++ b/test/registered/unit/model_executor/runner/test_prefill_cuda_graph_padding.py @@ -1,6 +1,7 @@ import unittest from types import SimpleNamespace +from sglang.srt.model_executor.cuda_graph_config import Backend from sglang.srt.model_executor.forward_batch_info import ( CaptureHiddenMode, ForwardMode, @@ -18,6 +19,8 @@ class TestPrefillCudaGraphPadding(CustomTestCase): def _make_runner(self): runner = PrefillCudaGraphRunner.__new__(PrefillCudaGraphRunner) runner._is_full_backend = False + runner.prefill_backend_name = Backend.TC_PIECEWISE + runner.has_mha_companion_layers = False runner.capture_hidden_mode = CaptureHiddenMode.NULL runner.capture_num_tokens = [4, 16] runner.max_num_tokens = 16