kernels-community
/

flash-attn2

Model card Files Files and versions

flash-attn2 / torch-ext /flash_attn

257 kB

5 contributors

History: 4 commits

drbh

fix: refactor bindings and update py library for correctness

876ac68 8 months ago

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

fix: refactor bindings and update py library for correctness 8 months ago
bert_padding.py
9.93 kB

fix: refactor bindings and update py library for correctness 8 months ago
flash_attn_interface.py
60.7 kB

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