kernels-community
/

flash-attn

Model card Files Files and versions

flash-attn / torch-ext

268 kB

3 contributors

History: 9 commits

drbh

fix: refactor bindings and update py library for correctness

876ac68 2 months ago

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

3.54 kB

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

6.99 kB

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