Dan Yao 1da802bdf2 Fix FA bwd alibi+causal NaN errors (#1352) ...

* fix bwd alibi nan error

* fix datatype

---------

Co-authored-by: danyao12 <danyao12>

2024-06-20 09:50:53 -05:00

..

block_attention_bias_enum.hpp

[CK_TILE] support alibi (#1269)

2024-05-07 22:32:54 +08:00

block_dropout.hpp

Hacking ck_tile fmha Dropout facility (#1344)

2024-06-19 10:37:22 +08:00

block_masking.hpp

Fix FA bwd alibi+causal NaN errors (#1352)

2024-06-20 09:50:53 -05:00

block_position_encoding.hpp

[CK_TILE] support group from cmdline (#1295)

2024-05-28 11:13:21 +08:00