kernels-community
/

flash-attn

Model card Files Files and versions Community

flash-attn / tests

Ctrl+K

Ctrl+K

3 contributors

History: 3 commits

drbh

fix: refactor bindings and update py library for correctness

876ac68 3 days ago

__init__.py

0 Bytes

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

98.2 kB

fix: refactor bindings and update py library for correctness 3 days ago
test_flash_attn_raw.py

6.42 kB

fix: refactor bindings and update py library for correctness 3 days ago
test_util.py

10.6 kB

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