Fix hybrid attention graph hook test fixture (#30790)

This commit is contained in:
Mohammad Miadh Angkad
2026-07-10 23:37:04 +05:30
committed by GitHub
parent 7998fecfd1
commit 7b99900980
@@ -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),