Block Sparsity and Flex Attention mask mod support (#1942)
* clean up and rebase for PR * add mask mod tests * add benchmarking files * refactor for better style * remove extraneous csrc * type hint buffers * refactor: order of non/overlap and modify blocksparse producer to agree with dense * change variable name back to buffers * remove unnecessary variable in first_half_block * restore erroneous packgqa deletion * add blocksparsity and mask_mod asserts to interface.py * fix rebase issues * Restore submodule and reset pointer to upstream/main * rename cutlass.const_expr to const_expr * support fully masked m blocks (i.e. skipped tiles) * remove outdated commented code
R
Reuben Stern committed
143b0ba20df0aca7d968d8ef5852ed10fe09caab
Parent: a098f98
Committed by GitHub <noreply@github.com>
on 10/21/2025, 10:11:37 PM