ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-11 08:50:17 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
1b468bac0bee62381fa7591ee2c114f8ea83061f
composable_kernel/include/ck_tile/ops/fmha/pipeline
History
aska-0096 1b468bac0b tempsave, trload+asyncload done
2025-07-21 05:55:55 +00:00
..
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
tempsave, trload+asyncload done
2025-07-21 05:55:55 +00:00
block_fmha_fwd_decode_pipeline_qr_ks_vs.hpp
tempsave, trload+asyncload done
2025-07-21 05:55:55 +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