Jeff Huang 993d3e2f0e [FMHA] Enable page size 16 for batch prefill kernel (#3568) ...

* [FMHA] Enable page size 16 for batch prefill kernel

* Refactor batch prefill KV offset logic to simplify template arguments
- Remove redundant `kLog2PageSize` and `kIsVTileFitsInPage` from template args.
- Add static assert to forbid `page_size=1` with vectorized layout.

2026-01-15 22:11:44 +08:00

..

block_fmha_batch_prefill_pipeline_qr_ks_vs_async_default_policy.hpp

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

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

block_fmha_batch_prefill_pipeline_qr_ks_vs_async.hpp

[FMHA] Enable page size 16 for batch prefill kernel (#3568)

2026-01-15 22:11:44 +08:00

block_fmha_bwd_convert_dq.hpp

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

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

block_fmha_bwd_dot_do_o.hpp

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

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

block_fmha_bwd_dq_dk_dv_pipeline_kr_ktr_vr_iglp.hpp

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

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

block_fmha_bwd_dq_dk_dv_pipeline_kr_ktr_vr.hpp

Add a workaround for a compiler issue for bwd on gfx90a and ROCm 7.1.1 (#3369)

2025-12-08 07:44:17 -08:00

block_fmha_bwd_dq_dk_dv_pipeline_selector.hpp

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

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

block_fmha_bwd_dq_dk_dv_pipeline_trload_kr_ktr_vr.hpp

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

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

block_fmha_bwd_dq_dk_dv_pipeline_trload_qr_qtr_dor.hpp

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

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

block_fmha_bwd_pipeline_default_policy.hpp

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

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

block_fmha_bwd_pipeline_problem.hpp

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

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

block_fmha_bwd_pipeline_trload_default_policy.hpp

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

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

block_fmha_fwd_appendkv_pipeline_default_policy.hpp

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

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

block_fmha_fwd_appendkv_pipeline.hpp

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

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

block_fmha_fwd_pagedkv_pipeline_qr_ks_vs_default_policy.hpp

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

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

block_fmha_fwd_pagedkv_pipeline_qr_ks_vs.hpp

[CK_TILE][FMHA] Enable gpt-oss sink (#3490)

2026-01-14 21:32:06 +08:00

block_fmha_fwd_splitkv_combine_pipeline_default_policy.hpp

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

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

block_fmha_fwd_splitkv_combine_pipeline.hpp

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

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

block_fmha_fwd_splitkv_pipeline_nwarp_sshuffle_qr_ks_vs_default_policy.hpp

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

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

block_fmha_fwd_splitkv_pipeline_nwarp_sshuffle_qr_ks_vs.hpp

[CK_TILE][FMHA] Enable gpt-oss sink (#3490)

2026-01-14 21:32:06 +08:00

block_fmha_fwd_splitkv_pipeline_qr_ks_vs_default_policy.hpp

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

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

block_fmha_fwd_splitkv_pipeline_qr_ks_vs.hpp

[CK_TILE][FMHA] Enable gpt-oss sink (#3490)

2026-01-14 21:32:06 +08:00

block_fmha_fwd_v3_pipeline_default_policy.hpp

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

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

block_fmha_fwd_v3_pipeline.hpp

[CK_TILE][FMHA] Add logits soft-capping support for FAv3 (WIP) (#3355)

2025-12-18 16:08:45 +08:00

block_fmha_pipeline_enum.hpp

[CK_TILE][FMHA] Integrate FAv2 & FAv3 (WIP) in the single fmha_fwd() API (#3153)

2025-12-05 10:31:12 +08:00

block_fmha_pipeline_problem.hpp

[FMHA] Enable page size 16 for batch prefill kernel (#3568)

2026-01-15 22:11:44 +08:00

block_fmha_pipeline_qr_ks_vs_async_default_policy.hpp

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

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

block_fmha_pipeline_qr_ks_vs_async_trload_policy.hpp

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

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

block_fmha_pipeline_qr_ks_vs_async_trload.hpp

[CK_TILE][FMHA] Enable gpt-oss sink (#3490)

2026-01-14 21:32:06 +08:00

block_fmha_pipeline_qr_ks_vs_async.hpp

[CK_TILE][FMHA] Enable gpt-oss sink (#3490)

2026-01-14 21:32:06 +08:00

block_fmha_pipeline_qr_ks_vs_default_policy.hpp

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

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

block_fmha_pipeline_qr_ks_vs_fp8.hpp

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

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

block_fmha_pipeline_qr_ks_vs_whole_k_prefetch_default_policy.hpp

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

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

block_fmha_pipeline_qr_ks_vs_whole_k_prefetch.hpp

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

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

block_fmha_pipeline_qr_ks_vs.hpp

[CK_TILE][FMHA] Enable gpt-oss sink (#3490)

2026-01-14 21:32:06 +08:00

block_fmha_pipeline_qs_ks_vs_default_policy.hpp

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

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

block_fmha_pipeline_qs_ks_vs.hpp

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

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

block_fmha_pipeline_qx_ks_vs_custom_policy.hpp

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

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

tile_fmha_shape.hpp

Shuffle fix for gfx950 (#3491)

2026-01-13 09:21:29 -08:00

tile_fmha_traits.hpp

[FMHA] Batch Prefill Support Improvements: Change KV Cache Layout & Large Page Size Support (#3442)

2026-01-05 18:41:47 +08:00