kernels-community
/

flash-attn

Model card Files Files and versions

flash-attn / torch-ext /flash_attn /ops

Ctrl+K

Ctrl+K

3 contributors

History: 1 commit

drbh

fix: refactor bindings and update py library for correctness

876ac68 about 1 month ago

triton
fix: refactor bindings and update py library for correctness about 1 month ago
__init__.py

0 Bytes

fix: refactor bindings and update py library for correctness about 1 month ago
activations.py

3.94 kB

fix: refactor bindings and update py library for correctness about 1 month ago
fused_dense.py

27.9 kB

fix: refactor bindings and update py library for correctness about 1 month ago
layer_norm.py

22.4 kB

fix: refactor bindings and update py library for correctness about 1 month ago
rms_norm.py

3.99 kB

fix: refactor bindings and update py library for correctness about 1 month ago