composable_kernel

block_fmha_batch_prefill_pipeline_qr_ks_vs_async_default_policy.hpp

[CK_TILE] Add logits soft-capping & customization support to the FMHA forward kernel/pipelines (#2163 )

2025-05-13 12:19:25 +08:00

block_fmha_batch_prefill_pipeline_qr_ks_vs_async.hpp

Revert "Eliminate warning caused by failed to meet occupancy requirement (#2389 )" (#2514 )

2025-07-17 10:09:01 +08:00

block_fmha_bwd_convert_dq.hpp

[CK_TILE] FA bwd kernels optimization (#1397 )

2024-08-16 13:40:10 -07:00

block_fmha_bwd_dot_do_o.hpp

[CK_TILE] FA bwd kernels optimization (#1397 )

2024-08-16 13:40:10 -07:00

block_fmha_bwd_dq_dk_dv_pipeline_kr_ktr_vr_iglp.hpp

[CK_TILE] Improve headdim96 performance for fmha-bwd (#1573 )

2024-10-16 18:14:32 +08:00

block_fmha_bwd_dq_dk_dv_pipeline_kr_ktr_vr.hpp

[CK_TILE] Improve headdim96 performance for fmha-bwd (#1573 )

2024-10-16 18:14:32 +08:00

block_fmha_bwd_pipeline_default_policy.hpp

[CK_TILE] Improve headdim96 performance for fmha-bwd (#1573 )

2024-10-16 18:14:32 +08:00

block_fmha_bwd_pipeline_enum.hpp

[CK_TILE] FA bwd kernels optimization (#1397 )

2024-08-16 13:40:10 -07:00

block_fmha_bwd_pipeline_problem.hpp

[CK_TILE] FA bwd kernels optimization (#1397 )

2024-08-16 13:40:10 -07:00

block_fmha_fwd_appendkv_pipeline_default_policy.hpp

[CK_TILE] Add PagedAttention kernels (#1387 )

2024-08-28 20:50:43 +08:00

block_fmha_fwd_appendkv_pipeline.hpp

[CK_TILE] Add PagedAttention kernels (#1387 )

2024-08-28 20:50:43 +08:00

block_fmha_fwd_decode_pipeline_qr_ks_vs_policy.hpp

Load Q through lds, implement xor;

2025-08-04 06:49:01 +00:00

block_fmha_fwd_decode_pipeline_qr_ks_vs.hpp

add vmcnt guard before load ktile

2025-08-04 10:02:17 +00:00

block_fmha_fwd_pagedkv_pipeline_qr_ks_vs_default_policy.hpp

ck tile pagedkv prefill (#2405 )

2025-07-07 16:16:54 +08:00

block_fmha_fwd_pagedkv_pipeline_qr_ks_vs.hpp

Add separate mask checking for scope [aligned_physical_seqlen_k_start, physical_seqlen_k_end) (#2487 )

2025-07-11 18:14:47 +08:00

block_fmha_fwd_splitkv_combine_pipeline_default_policy.hpp

[CK_TILE] Add fmha fwd N-Warp S-Shuffle pipeline (fmha fwd splitkv pipeline variant) (#1705 )

2024-12-20 14:41:01 +08:00

block_fmha_fwd_splitkv_combine_pipeline.hpp

roll back splitkv

2025-01-08 10:09:54 -08:00

block_fmha_fwd_splitkv_pipeline_nwarp_sshuffle_qr_ks_vs_default_policy.hpp

Ck tile/complete k prefetch (#1941 )

2025-03-07 14:19:51 +08:00

block_fmha_fwd_splitkv_pipeline_nwarp_sshuffle_qr_ks_vs.hpp

[CK_TILE] Add logits soft-capping & customization support to the FMHA forward kernel/pipelines (#2163 )

2025-05-13 12:19:25 +08:00

block_fmha_fwd_splitkv_pipeline_qr_ks_vs_default_policy.hpp

Ck tile/complete k prefetch (#1941 )

2025-03-07 14:19:51 +08:00

block_fmha_fwd_splitkv_pipeline_qr_ks_vs.hpp

ck tile pagedkv prefill (#2405 )

2025-07-07 16:16:54 +08:00

block_fmha_pipeline_enum.hpp

temp save, waiting for debug

2025-06-21 15:02:57 +00:00

block_fmha_pipeline_problem.hpp

Merge branch 'test_copy_fix' of https://github.com/ROCm/composable_kernel into fa_decode_pipeline

2025-07-17 07:24:32 +00:00

block_fmha_pipeline_qr_ks_vs_async_default_policy.hpp

Ck tile/complete k prefetch (#1941 )

2025-03-07 14:19:51 +08:00

block_fmha_pipeline_qr_ks_vs_async.hpp

Fix fmha fwd precision issue on MI3XX series (#2285 )

2025-06-10 15:03:23 +08:00

block_fmha_pipeline_qr_ks_vs_default_policy.hpp

Ck tile/complete k prefetch (#1941 )

2025-03-07 14:19:51 +08:00

block_fmha_pipeline_qr_ks_vs_fp8.hpp

[CK_TILE] Add fmha fwd headdim96 support (#1608 )

2024-10-30 14:03:16 +08:00

block_fmha_pipeline_qr_ks_vs_whole_k_prefetch_default_policy.hpp

Ck tile/complete k prefetch (#1941 )

2025-03-07 14:19:51 +08:00

block_fmha_pipeline_qr_ks_vs_whole_k_prefetch.hpp

Update for xformers (#2372 )

2025-06-22 00:28:30 -07:00

block_fmha_pipeline_qr_ks_vs.hpp

[CK_TILE] Add logits soft-capping & customization support to the FMHA forward kernel/pipelines (#2163 )

2025-05-13 12:19:25 +08:00

block_fmha_pipeline_qs_ks_vs_default_policy.hpp

Ck tile/complete k prefetch (#1941 )

2025-03-07 14:19:51 +08:00

block_fmha_pipeline_qs_ks_vs.hpp

[CK_TILE] Add logits soft-capping & customization support to the FMHA forward kernel/pipelines (#2163 )

2025-05-13 12:19:25 +08:00

block_fmha_pipeline_qx_ks_vs_custom_policy.hpp

[CK_TILE] FMHA Support hdim_v to as a Multiple of 32 (#2114 )

2025-06-24 01:33:31 +08:00

tile_fmha_shape.hpp

Reapply "[CK_TILE] support hdim=192/128 pair for deepseekv3 (#1961 )" … (#1971 )

2025-03-13 11:41:39 +08:00

tile_fmha_traits.hpp

Merge branch 'test_copy_fix' of https://github.com/ROCm/composable_kernel into fa_decode_pipeline

2025-07-17 07:24:32 +00:00