cameronshinn/block-sparse-attn-ck
Block sparse attention layer for PyTorch from Composable Kernels
C++
Stargazers
No one’s star this repository yet.
Block sparse attention layer for PyTorch from Composable Kernels
C++
No one’s star this repository yet.