kernels-community
/

flash-attn

Model card Files Files and versions Community

flash-attn / torch-ext

Ctrl+K

Ctrl+K

3 contributors

History: 9 commits

drbh

fix: refactor bindings and update py library for correctness

876ac68 3 days ago

flash_attn
fix: refactor bindings and update py library for correctness 3 days ago
torch_binding.cpp

3.54 kB

fix: refactor bindings and update py library for correctness 3 days ago
torch_binding.h

6.99 kB

fix: refactor bindings and update py library for correctness 3 days ago