composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 22:39:03 +00:00

Files

Po Yen Chen 1c3151963b [CK_TILE][FMHA] Add FP8 support for batch_prefill kernel (#3425 )

* Add fp8bf16 support for batch_prefill

* Fix wrong scale_s re-compute logic in batch_prefill

* Fix wrong scale_s re-compute logic in fmha fwd

* Fix batch_prefill codegen error

* Remove no-longer used GetName() function

* Add fp8 logits=True instances

* Update CHANGELOG.md

2025-12-24 10:34:06 +08:00

fmha_batch_prefill_kernel.hpp

[CK_TILE][FMHA] Add FP8 support for batch_prefill kernel (#3425 )

2025-12-24 10:34:06 +08:00

fmha_bwd_kernel.hpp

chore(copyright): update copyright header for include directory (#3293 )

2025-11-26 11:00:05 -07:00

fmha_fwd_appendkv_kernel.hpp

chore(copyright): update copyright header for include directory (#3293 )

2025-11-26 11:00:05 -07:00

fmha_fwd_appendkv_tile_partitioner.hpp

chore(copyright): update copyright header for include directory (#3293 )