ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-04 13:41:24 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
fe0a47a011c2adcb54dfc94a3029feb7b9980deb
composable_kernel/include/ck_tile/ops/fmha/pipeline
History
Yi DING fe0a47a011 [CK_TILE] FMHA BWD Add D96 Instances (#2916)
2025-09-24 17:04:23 +08: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] FMHA Fix synchronization issues in BWD pipelines (#2876)
2025-09-19 11:34:45 +05:00
block_fmha_bwd_dq_dk_dv_pipeline_kr_ktr_vr.hpp
[CK_TILE] FMHA avoid unnecessary vmcnt0 (#2715)
2025-08-25 20:55:12 +08:00
block_fmha_bwd_dq_dk_dv_pipeline_selector.hpp
[CK_TILE] FMHA BWD Decode Pipeline (#2643)
2025-08-12 17:02:52 +08:00
block_fmha_bwd_dq_dk_dv_pipeline_trload_kr_ktr_vr.hpp
[CK_TILE] FMHA Fix synchronization issues in BWD pipelines (#2876)
2025-09-19 11:34:45 +05:00
block_fmha_bwd_dq_dk_dv_pipeline_trload_qr_qtr_dor.hpp
[CK_TILE] FMHA BWD Fix Decode Accuracy (#2881)
2025-09-19 21:45:02 +08:00
block_fmha_bwd_pipeline_default_policy.hpp
[CK_TILE] FMHA BWD Add D96 Instances (#2916)
2025-09-24 17:04:23 +08:00
block_fmha_bwd_pipeline_problem.hpp
[CK_TILE] FMHA BWD Optimization For GFX950 (#2628)
2025-08-12 11:11:55 +08:00
block_fmha_bwd_pipeline_trload_default_policy.hpp
Reapply "[CK_TILE] FMHA BWD Enable Tile 16x192 (#2741)" (#2757) (#2761)
2025-09-08 09:21:14 -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_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
Revert "[CK-Tile] Add the API to load SGPR (#2878)" (#2904)
2025-09-23 14:33:51 -07: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
Revert "[CK-Tile] Add the API to load SGPR (#2878)" (#2904)
2025-09-23 14:33:51 -07:00
block_fmha_fwd_v3_pipeline_default_policy.hpp
[CK_TILE] Add FAv3 fwd pipeline (#2731)
2025-09-01 09:16:45 +08:00
block_fmha_fwd_v3_pipeline.hpp
[CK_TILE] FMHA FAv3 scheduling fine-tuning for performance (#2833)
2025-09-16 11:32:38 +08:00
block_fmha_pipeline_enum.hpp
Re-enable optimization for gfx950 fmha fwd (#2671)
2025-08-13 14:57:43 +08:00
block_fmha_pipeline_problem.hpp
[CK_TILE] Add FAv3 fwd pipeline (#2731)
2025-09-01 09:16:45 +08: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_trload_policy.hpp
[CK_TILE] CK_TILE GEMM WMMA Support for GFX11/GFX12 (#2466)
2025-08-15 16:22:27 -07:00
block_fmha_pipeline_qr_ks_vs_async_trload.hpp
[CK_TILE] FMHA FWD bug fix (#2888)
2025-09-23 15:00:46 +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
use iglp to improve dim256 fmha fwd in qr_ks_vs pipeline (#2711)
2025-08-28 11:39:39 +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
[FMHA FWD] gfx950 Accuracy enhancement & bug fix (#2900)
2025-09-24 00:59:41 +08:00
tile_fmha_shape.hpp
[CK_TILE] FMHA BWD Decode Pipeline (#2643)
2025-08-12 17:02:52 +08:00
tile_fmha_traits.hpp
[CK_TILE] Add FAv3 fwd pipeline (#2731)
2025-09-01 09:16:45 +08:00