Fix hybrid attention graph hook test fixture (#30790)

2026-07-10 23:37:04 +05:30
parent 7998fecfd1
commit 7b99900980
1 changed files with 1 additions and 0 deletions
@@ -142,6 +142,7 @@ def test_hybrid_wrappers_forward_in_graph_hook():
            kv_cache_dtype=torch.bfloat16,
            token_to_kv_pool=None,
            req_to_token_pool=None,
            server_args=SimpleNamespace(speculative_attention_mode="decode"),
        ),
        prefill_backend=make_fake("prefill", calls),
        decode_backend=make_fake("decode", calls),