kernels-community
/

flash-attn2

Model card Files Files and versions

flash-attn2 / torch-ext /flash_attn /layers

20.6 kB

3 contributors

History: 1 commit

drbh

fix: refactor bindings and update py library for correctness

876ac68 3 months ago

__init__.py

0 Bytes

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

2.14 kB

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

18.5 kB

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