composable_kernel

block_fmha_batch_prefill_pipeline_qr_ks_vs_async_default_policy.hpp

Optimize batch prefill kernel performance for VECTORIZED_LAYOUT KV cache (#3657 )

2026-01-29 07:18:41 +08:00

block_fmha_batch_prefill_pipeline_qr_ks_vs_async.hpp

[CK] Fix 32-bit overflow in batch prefill kernel for >4GB KV cache (#4999 )

2026-03-05 09:08:01 +08:00

block_fmha_bwd_convert_dq.hpp

Fix

2026-04-07 23:12:39 -05:00

block_fmha_bwd_dot_do_o.hpp

[CK Tile] Add sink token gradient support in FMHA backward pass (#5504 )

2026-04-02 11:17:01 +08:00

block_fmha_bwd_dq_dk_dv_pipeline_kr_ktr_vr_iglp.hpp

[CK_TILE] Fix FMHA BWD IGLP incorrect results due to AGPR misallocation (#5991 )

2026-04-01 13:44:04 +08:00

block_fmha_bwd_dq_dk_dv_pipeline_kr_ktr_vr.hpp

[CK_TILE] Fix FMHA BWD register pressure by wrapping num_total_loop with amd_wave_read_first_lane (#5915 )

2026-03-30 09:44:35 +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

[CK_TILE] Fix FMHA BWD register pressure by wrapping num_total_loop with amd_wave_read_first_lane (#5915 )

2026-03-30 09:44:35 +08:00

block_fmha_bwd_dq_dk_dv_pipeline_trload_qr_qtr_dor.hpp

Cleanup and refactoring related to tile loading (#4294 )

2026-03-02 12:20:55 +00:00

block_fmha_bwd_pipeline_default_policy.hpp

[CK_TILE] Flatten nested static_for loops into static_ford (#5939 )

2026-04-07 08:36:45 -06:00

block_fmha_bwd_pipeline_problem.hpp

Merge origin/develop into users/yiding12/fmha-bwd-workspace

2026-04-07 05:28:49 -05: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] Fix gptoss sink (#4313 )

2026-03-02 09:53:52 +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] Fix gptoss sink (#4313 )

2026-03-02 09:53:52 +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] Fix gptoss sink (#4313 )

2026-03-02 09:53:52 +08:00

block_fmha_fwd_v3_pipeline_default_policy.hpp

[CK] Add FP8 per-tensor quantization support for FMHA V3 pipeline (#6051 )

2026-04-07 22:19:28 +08:00

block_fmha_fwd_v3_pipeline.hpp

[CK] Add FP8 per-tensor quantization support for FMHA V3 pipeline (#6051 )

2026-04-07 22:19:28 +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

[CK_TILE][FMHA] Support microscaling (mxfp8 and mxfp4) on gfx950 (#4368 )

2026-03-11 09:59:50 +00: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

Cleanup and refactoring related to tile loading (#4294 )

2026-03-02 12:20:55 +00:00

block_fmha_pipeline_qr_ks_vs_async.hpp

[CK_TILE][FMHA] Support microscaling (mxfp8 and mxfp4) on gfx950 (#4368 )

2026-03-11 09:59:50 +00: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] Support microscaling (mxfp8 and mxfp4) on gfx950 (#4368 )

2026-03-11 09:59:50 +00: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

[CK_TILE][FMHA] Support microscaling (mxfp8 and mxfp4) on gfx950 (#4368 )

2026-03-11 09:59:50 +00: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