kernels-community
/

flash-attn

Model card Files Files and versions

flash-attn / tests

Ctrl+K

Ctrl+K

3 contributors

History: 3 commits

drbh

fix: refactor bindings and update py library for correctness

876ac68 about 2 months ago

__init__.py

0 Bytes

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

98.2 kB

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

6.42 kB

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

10.6 kB

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