AUTHORS
LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
csrc/flash_sparse_attn/flash_api.cpp
csrc/flash_sparse_attn/src/block_info.h
csrc/flash_sparse_attn/src/flash.h
csrc/flash_sparse_attn/src/flash_bwd_kernel.h
csrc/flash_sparse_attn/src/flash_bwd_launch_template.h
csrc/flash_sparse_attn/src/flash_bwd_preprocess_kernel.h
csrc/flash_sparse_attn/src/flash_fwd_kernel.h
csrc/flash_sparse_attn/src/flash_fwd_launch_template.h
csrc/flash_sparse_attn/src/generate_kernels.py
csrc/flash_sparse_attn/src/hardware_info.h
csrc/flash_sparse_attn/src/kernel_traits.h
csrc/flash_sparse_attn/src/mask.h
csrc/flash_sparse_attn/src/namespace_config.h
csrc/flash_sparse_attn/src/softmax.h
csrc/flash_sparse_attn/src/static_switch.h
csrc/flash_sparse_attn/src/utils.h
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim128_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim128_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim128_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim128_fp16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim192_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim192_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim192_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim192_fp16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim256_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim256_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim256_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim256_fp16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim32_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim32_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim32_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim32_fp16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim64_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim64_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim64_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim64_fp16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim96_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim96_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim96_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_bwd_hdim96_fp16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim128_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim128_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim128_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim128_fp16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim192_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim192_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim192_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim192_fp16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim256_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim256_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim256_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim256_fp16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim32_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim32_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim32_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim32_fp16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim64_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim64_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim64_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim64_fp16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim96_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim96_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim96_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_hdim96_fp16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim128_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim128_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim128_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim128_fp16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim192_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim192_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim192_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim192_fp16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim256_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim256_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim256_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim256_fp16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim32_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim32_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim32_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim32_fp16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim64_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim64_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim64_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim64_fp16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim96_bf16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim96_bf16_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim96_fp16_causal_sm80.cu
csrc/flash_sparse_attn/src/instantiations/flash_fwd_split_hdim96_fp16_sm80.cu
flash_sparse_attn/__init__.py
flash_sparse_attn.egg-info/PKG-INFO
flash_sparse_attn.egg-info/SOURCES.txt
flash_sparse_attn.egg-info/dependency_links.txt
flash_sparse_attn.egg-info/requires.txt
flash_sparse_attn.egg-info/top_level.txt
flash_sparse_attn/deprecated/__init__.py
flash_sparse_attn/deprecated/flash_dmattn_triton.py
flash_sparse_attn/deprecated/flash_sparse_attn_flex.py
flash_sparse_attn/deprecated/flash_sparse_attn_interface.py
flash_sparse_attn/deprecated/flash_sparse_attn_triton.py
flash_sparse_attn/integrations/flash_sparse_attention.py
flash_sparse_attn/integrations/import_utils.py
flash_sparse_attn/integrations/modeling_flash_sparse_attention_utils.py
flash_sparse_attn/modules/flash_sparse_attention.py
flash_sparse_attn/modules/multi_head_attention.py
flash_sparse_attn/ops/triton/activations.py
flash_sparse_attn/ops/triton/assert_inputs.py
flash_sparse_attn/ops/triton/block_info.py
flash_sparse_attn/ops/triton/flash_bwd_postprocess.py
flash_sparse_attn/ops/triton/flash_bwd_preprocess.py
flash_sparse_attn/ops/triton/flash_dense_bwd.py
flash_sparse_attn/ops/triton/flash_dense_fwd.py
flash_sparse_attn/ops/triton/flash_fwd_combine.py
flash_sparse_attn/ops/triton/flash_gated_bwd.py
flash_sparse_attn/ops/triton/flash_gated_fwd.py
flash_sparse_attn/ops/triton/flash_sparse_bwd.py
flash_sparse_attn/ops/triton/flash_sparse_fwd.py
flash_sparse_attn/ops/triton/interface.py
flash_sparse_attn/ops/triton/launch_grid.py
flash_sparse_attn/ops/triton/launch_template.py
flash_sparse_attn/ops/triton/mask.py
flash_sparse_attn/ops/triton/seqlen_info.py
flash_sparse_attn/ops/triton/utils.py
flash_sparse_attn/utils/mask.py
flash_sparse_attn/utils/padding.py
tests/test_dense_base_backward.py
tests/test_dense_base_forward.py
tests/test_dense_varlen_backward.py
tests/test_dense_varlen_forward.py
tests/test_gated_base_backward.py
tests/test_gated_base_forward.py
tests/test_gated_varlen_backward.py
tests/test_gated_varlen_forward.py
tests/test_sparse_base_backward.py
tests/test_sparse_base_forward.py
tests/test_sparse_varlen_backward.py
tests/test_sparse_varlen_forward.py
tests/test_utils.py