rocking eb7f617713 fp8 fmha async pipeline (#3339) ...

* replace qr with async pipeline

* Add fp8fp32 to DTYPE_BITS

* Add kAlignmentRandVal to avoid compile fail

* format

---------

Co-authored-by: Thomas Ning <Thomas.Ning@amd.com>

2025-12-04 12:18:25 +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

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

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

fmha_fwd.py

fp8 fmha async pipeline (#3339)

2025-12-04 12:18:25 +08:00

fmha_pagedkv_prefill.py

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

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