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
18686cfe5b83a2d16424fa6cc3d3eecc5e1a24ef
composable_kernel/include/ck_tile/ops/fmha/pipeline
History
aska-0096 18686cfe5b tempsave, fmha_decode
2025-07-08 08:37:20 +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
Fix fmha fwd precision issue on MI3XX series (#2285)
2025-06-10 15:03:23 +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, fmha_decode
2025-07-08 08:37:20 +00:00
block_fmha_fwd_decode_pipeline_qr_ks_vs.hpp
tempsave, fmha_decode
2025-07-08 08:37:20 +00: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] Add logits soft-capping & customization support to the FMHA forward kernel/pipelines (#2163)
2025-05-13 12:19:25 +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
tempsave, fmha_decode
2025-07-08 08:37:20 +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
Ck tile/complete k prefetch (#1941)
2025-03-07 14:19:51 +08: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
Do not use warpSize as compile time constant as it is removed (#2320)
2025-06-17 11:54:30 -07: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
tempsave, fmha_decode
2025-07-08 08:37:20 +00:00