ltqin 92653168c2 flashattention fwd add (80, 96) instance (#3415) ...

* add hdim (96,96) instance

* change to (80,96)

* format py

* remove 96 in optdim

* when N=6 change to llvm_amdgcn_raw_buffer_load_i32x3

2025-12-17 09:16:11 -08:00

..

__init__.py

chore(copyright): update copyright header for example directory (#3273)

2025-11-24 18:02:41 -08:00

fmha_batch_prefill.py

chore(copyright): update copyright header for example directory (#3273)

2025-11-24 18:02:41 -08:00

fmha_bwd.py

chore(copyright): update copyright header for example directory (#3273)

2025-11-24 18:02:41 -08:00

fmha_fwd_appendkv.py

chore(copyright): update copyright header for example directory (#3273)

2025-11-24 18:02:41 -08:00

fmha_fwd_splitkv.py

Add attention sink support for FMHA FWD (#3368)

2025-12-15 12:21:59 +08:00

fmha_fwd.py

flashattention fwd add (80, 96) instance (#3415)

2025-12-17 09:16:11 -08:00

fmha_pagedkv_prefill.py

Add attention sink support for FMHA FWD (#3368)

2025-12-15 12:21:59 +08:00