kernels-community
/

flash-attn2

Model card Files Files and versions

flash-attn2 / tests

115 kB

5 contributors

History: 3 commits

drbh

fix: refactor bindings and update py library for correctness

876ac68 8 months ago

__init__.py

0 Bytes

feat: include source and enable build 11 months ago
test_flash_attn.py

98.2 kB

fix: refactor bindings and update py library for correctness 8 months ago
test_flash_attn_raw.py

6.42 kB

fix: refactor bindings and update py library for correctness 8 months ago
test_util.py

10.6 kB

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