Laurent Mazare 67834119fc Fix the flash-attention function names. (#282)

2023-07-31 10:04:39 +01:00

..

cutlass @ c4f6b8c6bc

Add flash attention (#241)

2023-07-26 07:48:10 +01:00

kernels

Flash attention without padding (varlen). (#281)

2023-07-31 09:45:39 +01:00

src

Fix the flash-attention function names. (#282)

2023-07-31 10:04:39 +01:00

tests

Flash attention without padding (varlen). (#281)

2023-07-31 09:45:39 +01:00

build.rs

Add some flash attn test (#253)

2023-07-26 20:56:00 +01:00

Cargo.toml

Softmax numerical stability. (#267)

2023-07-28 13:13:01 +01:00