[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:
co-authored by
llilian73
hnyls2002
Liangsheng Yin
parent
0f18d389b4
commit
b24c8f10e7
@@ -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 = (
|
||||
(
|
||||
|
||||
Reference in New Issue
Block a user