Spaces:
Running on Zero
Running on Zero
File size: 750 Bytes
df252a6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | """Vendored FastVideo block-sparse attention Triton kernels.
Copied verbatim from `fastvideo-kernel` (https://github.com/hao-ai-lab/FastVideo,
`fastvideo-kernel/python/fastvideo_kernel/triton_kernels/`), Apache License 2.0.
Only the two pure-Triton modules are vendored: the package's default route is a
CUDA extension that has to be compiled per architecture (and whose fastest entry,
`block_sparse_attn_sm100a`, is GB200-only), while FastVideo's own
`--vsa-kernel triton` route -- these two files -- runs anywhere Triton does and
computes exactly the same masked attention.
"""
from .block_sparse_attn_triton import triton_block_sparse_attn_forward
from .index import map_to_index
__all__ = ["triton_block_sparse_attn_forward", "map_to_index"]
|