kernels-community
/

flash-attn

Model card Files Files and versions Community

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 8 days ago

triton
fix: refactor bindings and update py library for correctness 8 days ago
__init__.py

0 Bytes

fix: refactor bindings and update py library for correctness 8 days ago
activations.py

3.94 kB

fix: refactor bindings and update py library for correctness 8 days ago
fused_dense.py

27.9 kB

fix: refactor bindings and update py library for correctness 8 days ago
layer_norm.py

22.4 kB

fix: refactor bindings and update py library for correctness 8 days ago
rms_norm.py

3.99 kB

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