bugfix: fix sliding window attention tests of ragged attention api (#633

) The unittests of sliding window atteniton didn't pass because we forget to set `causal` in `plan` function. This PR fixes the issue.
flashinfer-ai · Nov 24, 2024 · ae501ed · ae501ed
1 parent a059586
commit ae501ed
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_sliding_window.py b/tests/test_sliding_window.py
@@ -352,6 +352,7 @@ def test_batch_ragged_prefill_sliding_window(
         num_kv_heads,
         head_dim,
         window_left=window_left,
+        causal=True,
     )
     o = wrapper.run(q, k, v)