kernels-community
/

flash-attn2

Model card Files Files and versions

flash-attn2 / tests

115 kB

Ctrl+K

Ctrl+K

6 contributors

History: 3 commits

drbh

fix: refactor bindings and update py library for correctness

876ac68 9 months ago

__init__.py

0 Bytes
feat: include source and enable build about 1 year ago
test_flash_attn.py

98.2 kB
fix: refactor bindings and update py library for correctness 9 months ago
test_flash_attn_raw.py

6.42 kB
fix: refactor bindings and update py library for correctness 9 months ago
test_util.py

10.6 kB
fix: refactor bindings and update py library for correctness 9 months ago