kernels-community
/

flash-attn2

Model card Files Files and versions

flash-attn2 / torch-ext

268 kB

5 contributors

History: 9 commits

drbh

fix: refactor bindings and update py library for correctness

876ac68 8 months ago

flash_attn
fix: refactor bindings and update py library for correctness 8 months ago
torch_binding.cpp

3.54 kB

fix: refactor bindings and update py library for correctness 8 months ago
torch_binding.h

6.99 kB

fix: refactor bindings and update py library for correctness 8 months ago