[CuTe,Flex] varlen blocksparsity#2224

Merged

drisspg merged 2 commits into

Dao-AILab:mainfrom

reubenconducts:rstern/varlen-blocksparsity

May 7, 2026

Commits on May 6, 2026

varlen block-sparsity for forward
reubenconducts
committed
rename cu_total_n_blocks to cu_block_idx_offsets; move cu_total_m_blocks/cu_block_idx_offsets into BlockSparseTensors instead of threading them as standalone parameters; drop the two <tensor>[-1].i…
reubenconducts
committed