composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-29 11:16:59 +00:00

Files

msaffari-amd 403d99124d [AITERKER-112] Add PER_TOKEN_HEAD FP8 quant scheme to batch_prefill

- New BlockAttentionQuantScaleEnum::PER_TOKEN_HEAD enum value
- Pipeline overload in block_fmha_batch_prefill_pipeline_qr_ks_vs_async
  applying per-token Q/K descale via GEMM0-post outer product and
  per-head V descale at epilogue
- fmha_batch_prefill_kernel kargs + MakeKargs + pipeline dispatch
- fmha_fwd.hpp host-side traits/args wiring
- quant.hpp trait specialization
- Codegen emits PER_TOKEN_HEAD kernel variants

2026-05-19 15:41:32 +00:00

fmha_batch_prefill_kernel.hpp

[AITERKER-112] Add PER_TOKEN_HEAD FP8 quant scheme to batch_prefill

2026-05-19 15:41:32 +00:00

fmha_bwd_kernel.hpp

[rocm-libraries] ROCm/rocm-libraries#7331 (commit 5692db0)

2026-05-14 13:34:32 +00: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 )