vivienfanghuagood f3d8b7210f Extend CK fmha_batch_prefill kernel coverage to head_dim=256 (#3328) ...

Co-authored-by: Po Yen Chen <PoYen.Chen@amd.com>
Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com>

2026-01-30 11:18:20 +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

Extend CK fmha_batch_prefill kernel coverage to head_dim=256 (#3328)

2026-01-30 11:18:20 +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

Revert "Revert " Fp8 block scale quantization for fmha fwd (#3330)" (#3633)" (#3635)

2026-01-23 09:03:22 -08:00

fmha_pagedkv_prefill.py

Add attention sink support for FMHA FWD (#3368)

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