[Perf] Skip bool-mask round-trip in block-sparse VSA attention (#1243)

2026-04-26 15:14:37 -07:00
parent 37418946c8
commit 7b872cc41e
4 changed files with 323 additions and 98 deletions
@@ -5,6 +5,11 @@ from fastvideo_kernel.ops import (
    video_sparse_attn,
 )

+from fastvideo_kernel.block_sparse_attn import (
+    block_sparse_attn,
+    block_sparse_attn_from_indices,
+)
+
 from fastvideo_kernel.vmoba import (
    moba_attn_varlen,
    process_moba_input,
@@ -22,6 +27,8 @@ from fastvideo_kernel.turbodiffusion_ops import (
 __all__ = [
    "sliding_tile_attention",
    "video_sparse_attn",
+    "block_sparse_attn",
+    "block_sparse_attn_from_indices",
    "moba_attn_varlen",
    "process_moba_input",
    "process_moba_output",