kernels-community
/

flash-attn2

Model card Files Files and versions

flash-attn2 / torch-ext /flash_attn /ops

152 kB

3 contributors

History: 1 commit

drbh

fix: refactor bindings and update py library for correctness

876ac68 3 months ago

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

0 Bytes

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

3.94 kB

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

27.9 kB

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

22.4 kB

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

3.99 kB

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