[FlashInfer] Avoid D2H sync for sliding-window lengths (#32218)

Co-authored-by: llilian73 <204300658+llilian73@users.noreply.github.com>
Co-authored-by: hnyls2002 <lsyincs@gmail.com>
Co-authored-by: Liangsheng Yin <hnyls2002@gmail.com>
This commit is contained in:
lynn.lin
2026-09-01 14:08:22 -07:00
committed by GitHub
co-authored by llilian73 hnyls2002 Liangsheng Yin
parent 0f18d389b4
commit b24c8f10e7
2 changed files with 30 additions and 2 deletions
@@ -120,6 +120,22 @@ class TestFlashInferSWAAttentionBackendCorrectness(CustomTestCase):
1,
"dflash",
),
(
DenseAttentionCase(
name="runner_dflash_verify_swa_window_edges",
backend="flashinfer",
forward_mode=ForwardMode.TARGET_VERIFY,
num_heads=4,
num_kv_heads=4,
page_size=16,
# Straddle the window: one request below, one at, one above.
prefix_lens=(1, 4, 9),
extend_lens=(3, 3, 3),
sliding_window_size=4,
),
1,
"dflash",
),
)
SPEC_VERIFY_CUDA_GRAPH_CASES = (
(