From b6e2aa748b2bef8d72f2003cd881a77ac47dcb57 Mon Sep 17 00:00:00 2001 From: Lucas Wilkinson Date: Sun, 13 Sep 2026 13:34:09 -0400 Subject: [PATCH] [CI] Initialize warmup registry in V2 QSA runner fixture (#56707) Signed-off-by: Lucas Wilkinson Co-authored-by: OpenAI Codex --- tests/v1/worker/test_gpu_model_runner_v2.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/v1/worker/test_gpu_model_runner_v2.py b/tests/v1/worker/test_gpu_model_runner_v2.py index c9ce409f5c..86ff1a0745 100644 --- a/tests/v1/worker/test_gpu_model_runner_v2.py +++ b/tests/v1/worker/test_gpu_model_runner_v2.py @@ -8,6 +8,7 @@ import pytest import torch import vllm.v1.worker.gpu.model_runner as model_runner_module +from vllm.model_executor.warmup.jit_warmup import JitWarmupRegistry from vllm.v1.kv_cache_interface import ( CircularBufferSpec, FullAttentionSpec, @@ -37,6 +38,7 @@ def test_qsa_circular_group_uses_custom_slot_mapping(monkeypatch): parallel_config=parallel_config, cache_config=SimpleNamespace(mamba_cache_mode="none"), ) + runner.jit_warmup_registry = JitWarmupRegistry(runner.vllm_config) runner.model_state = SimpleNamespace( get_additional_cg_support=lambda: (), num_new_sampled_tokens_per_step=1,