SIGN IN SIGN UP

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