kernels-community
/

flash-attn2

Model card Files Files and versions

flash-attn2 / torch-ext /flash_attn

257 kB

3 contributors

History: 4 commits

drbh

fix: refactor bindings and update py library for correctness

876ac68 3 months ago

layers
fix: refactor bindings and update py library for correctness 3 months ago
ops
fix: refactor bindings and update py library for correctness 3 months ago
__init__.py

14.2 kB

fix: refactor bindings and update py library for correctness 3 months ago
bert_padding.py

9.93 kB

fix: refactor bindings and update py library for correctness 3 months ago
flash_attn_interface.py

60.7 kB

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