Skip to content

Commit aeabaf7

Browse files
authoredJan 31, 2025
bugfix: Fix block-sparse attention API (#767)
This pull request includes a small change to the `flashinfer/sparse.py` file. The change reorders the `lse` parameter to ensure it is placed correctly in the argument list. * [`flashinfer/sparse.py`](diffhunk://#diff-de586226b1037125dbf7d04b1037c052a7651cd335e01fabd39400ee877054fdL586-R587): Reordered the `lse` parameter in the `run` method to follow the `out` parameter.
1 parent e5a3bef commit aeabaf7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed
 

‎flashinfer/sparse.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -583,8 +583,8 @@ def run(
583583
self._paged_kv_indptr_buf,
584584
self._paged_kv_indices_buf,
585585
self._paged_kv_last_page_len,
586-
lse,
587586
out,
587+
lse,
588588
TensorLayout[self._kv_layout].value,
589589
-1, # window_left
590590
_get_cache_alibi_slopes_buf(q.shape[1], self.device),

0 commit comments

Comments
 (0)