/block-sparse-attn-ck

Block sparse attention layer for PyTorch from Composable Kernels

Primary LanguageC++

Stargazers

No one’s star this repository yet.