ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-02 13:17:36 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
563f970e102cdcef50241197fa3474c8e85f0d74
composable_kernel/include/ck/tensor_operation/gpu/grid
History
lirui927 563f970e10 Support OAI SwiGLU in MoE epilogue
2026-06-24 17:34:07 +08:00
..
batchnorm_multiblock
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gemm_layernorm
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
normalization
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
block_to_ctile_map.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
epilogue_cshuffle_v3_reduce_wmma.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
epilogue_cshuffle_v3_welford_wmma.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
epilogue_cshuffle_v3_wmma_base.hpp
Add support for direct store in epilogue and padding support for wave transfer without transpose (#3465)
2026-01-14 11:02:19 +01:00
epilogue_cshuffle_v3_wmma.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
epilogue_direct_store.hpp
Add support for direct store in epilogue and padding support for wave transfer without transpose (#3465)
2026-01-14 11:02:19 +01:00
gridwise_2d_multiple_reduction_multiblock.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_2d_multiple_reduction_threadwise.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_2d_reduction_multiblock.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_2d_reduction_threadwise_multi_d.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_2d_reduction_threadwise.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_ab_transfer_thread_tiles_preshuffle.hpp
Multi AB support for wave transfer (#3578)
2026-01-29 10:29:40 -08:00
gridwise_ab_transfer_thread_tiles.hpp
Multi AB support for wave transfer (#3578)
2026-01-29 10:29:40 -08:00
gridwise_ab_transfer_wave_tiles_interleave.hpp
Multi AB support for wave transfer (#3578)
2026-01-29 10:29:40 -08:00
gridwise_ab_transfer_wave_tiles.hpp
Multi AB support for wave transfer (#3578)
2026-01-29 10:29:40 -08:00
gridwise_batched_gemm_gemm_wmma_cshuffle_v3.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_batched_gemm_gemm_xdl_cshuffle_v1.hpp
[ck] add gridwise base class for in all xdl kernel (#186) (#3544)
2026-01-27 12:49:47 -08:00
gridwise_batched_gemm_multiple_d_gemm_multiple_d_xdl_cshuffle_v1.hpp
[ck] add gridwise base class for in all xdl kernel (#186) (#3544)
2026-01-27 12:49:47 -08:00
gridwise_batched_gemm_multiple_d_softmax_gemm_xdl_cshuffle_v1.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_batched_gemm_softmax_gemm_wmma_cshuffle.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_batched_gemm_softmax_gemm_xdl_cshuffle_v1.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_batchnorm_backward_blockwise_welford.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_batchnorm_forward_blockwise_welford.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gridwise_elementwise_1d_scale.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gridwise_elementwise_2d.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gridwise_elementwise_layernorm_welford_variance.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_fpAintB_gemm_wmma.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gridwise_gemm_bias_add_reduce_xdl_cshuffle_v1.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_gemm_dl_multiple_d.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gridwise_gemm_dl_v1r3.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gridwise_gemm_dpp.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_multiple_abd_xdl_cshuffle.hpp
[ck] add gridwise base class for in all xdl kernel (#186) (#3544)
2026-01-27 12:49:47 -08:00
gridwise_gemm_multiple_d_multiple_r_xdl_cshuffle.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_gemm_multiple_d_wmma_cshuffle.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gridwise_gemm_multiple_d_xdl_cshuffle_lds_direct_load.hpp
[ck] add gridwise base class for in all xdl kernel (#186) (#3544)
2026-01-27 12:49:47 -08:00
gridwise_gemm_multiple_d_xdl_cshuffle.hpp
[rocm-libraries] ROCm/rocm-libraries#6302 (commit 8d419e8)
2026-04-10 11:17:11 -04:00
gridwise_gemm_multiple_d_xdl_splitk_cshuffle.hpp
[rocm-libraries] ROCm/rocm-libraries#4299 (commit 668cd49)
2026-02-19 09:13:05 +01:00
gridwise_gemm_pipeline_selector.hpp
Simplify includes for CK builder reflection (#3357)
2025-12-05 07:44:10 -08:00
gridwise_gemm_pipeline_v1.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gridwise_gemm_pipeline_v2.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gridwise_gemm_pipeline_v3.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gridwise_gemm_pipeline_v4_direct_load.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gridwise_gemm_reduce_xdl_cshuffle_v1.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_gemm_split_k_multiple_d_xdl_cshuffle_v2.hpp
[ck] add gridwise base class for in all xdl kernel (#186) (#3544)
2026-01-27 12:49:47 -08:00
gridwise_gemm_split_k_multiple_d_xdl_cshuffle.hpp
[ck] add gridwise base class for in all xdl kernel (#186) (#3544)
2026-01-27 12:49:47 -08:00
gridwise_gemm_waveletmodel.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gridwise_gemm_wmma_cshuffle_v3_ab_scale.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_wmma_cshuffle_v3_common.hpp
[rocm-libraries] ROCm/rocm-libraries#4340 (commit 70a312f)
2026-02-26 00:28:09 +00:00
gridwise_gemm_wmma_cshuffle_v3.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_wmma.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gridwise_gemm_xdl_cshuffle_common.hpp
[rocm-libraries] ROCm/rocm-libraries#6118 (commit 2c7dcf7)
2026-04-21 07:24:48 +00:00
gridwise_gemm_xdl_cshuffle_conv_v3.hpp
[rocm-libraries] ROCm/rocm-libraries#6624 (commit 47d0162)
2026-04-23 11:16:55 +02:00
gridwise_gemm_xdl_cshuffle_streamk_v3.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_xdl_cshuffle_v1.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_xdl_cshuffle_v2.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_xdl_cshuffle_v3_b_preshuffle.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_xdl_cshuffle_v3_b_scale.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_xdl_cshuffle_v3_multi_abd.hpp
[rocm-libraries] ROCm/rocm-libraries#7111 (commit 651947f)
2026-05-08 07:14:14 -07:00
gridwise_gemm_xdl_cshuffle_v3_multi_d_ab_scale.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_xdl_cshuffle_v3_multi_d_b_preshuffle.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_xdl_cshuffle_v3_multi_d_blockscale_b_preshuffle.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_xdl_cshuffle_v3_multi_d.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_xdl_cshuffle_v3_mx_bpreshuffle.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_xdl_cshuffle_v3_mx.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_xdl_cshuffle_v3.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_xdl_layernorm_cshuffle_v1.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_gemm_xdl_waveletmodel_cshuffle.hpp
[ck] add gridwise base class for in all xdl kernel (#186) (#3544)
2026-01-27 12:49:47 -08:00
gridwise_gemm_xdlops_bwd_weight.hpp
[Compiler] Addressing new compiler warnings (#3640)
2026-02-02 09:39:48 -08:00
gridwise_gemm_xdlops_skip_b_lds_v1.hpp
[ck] add gridwise base class for in all xdl kernel (#186) (#3544)
2026-01-27 12:49:47 -08:00
gridwise_gemm_xdlops_splitk_lds_direct_load.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_xdlops_streamk.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_xdlops_v2r3.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_xdlops_v2r4.hpp
[ck] add gridwise base class for in all xdl kernel (#186) (#3544)
2026-01-27 12:49:47 -08:00
gridwise_gemm_xdlops_v2r4r2.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_gemm_xdlops_v3r1.hpp
[ck] add gridwise base class for in all xdl kernel (#186) (#3544)
2026-01-27 12:49:47 -08:00
gridwise_gemm_xdlops_v3r2.hpp
[ck] add gridwise base class for in all xdl kernel (#186) (#3544)
2026-01-27 12:49:47 -08:00
gridwise_gemm_xdlops_v3r3.hpp
[ck] add gridwise base class for in all xdl kernel (#186) (#3544)
2026-01-27 12:49:47 -08:00
gridwise_moe_gemm_blockscale.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_moe_gemm.hpp
Support OAI SwiGLU in MoE epilogue
2026-06-24 17:34:07 +08:00
gridwise_moe_mx_gemm_bns.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_moe_mx_gemm_bpreshuffle.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_moe_mx_gemm.hpp
[rocm-libraries] ROCm/rocm-libraries#6550 (commit c396de9)
2026-04-22 15:47:47 +00:00
gridwise_permute.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gridwise_put_element_1d.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gridwise_set_buffer_value.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gridwise_set_multiple_buffer_value.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gridwise_softmax.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_sparse_embeddings_forward_layernorm_builtins.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_sparse_embeddings_forward_layernorm.hpp
[rocm-libraries] ROCm/rocm-libraries#5031 (commit 1d86a92)
2026-03-18 08:45:22 -06:00
gridwise_tensor_rearrange.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00