ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-30 11:47:48 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
daa013bf62aebdc1cd88a2540b0552879ef3deb2
composable_kernel/include/ck/tensor_operation/gpu/grid
History
mtgu0705 daa013bf62 moe gemm2 bpreshuffle function passed
2025-06-18 00:23:44 -05:00
..
batchnorm_multiblock
[HotFix] add config and version files to pass on build info (#856)
2023-08-23 11:36:17 -07:00
gemm_layernorm
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
normalization
layernorm and groupnorm backward data (#1083)
2023-12-19 04:23:11 +08:00
block_to_ctile_map.hpp
MI308 fix for streamk 1-Tile floating point exception (#2101)
2025-04-21 11:44:07 -07:00
gridwise_2d_multiple_reduction_multiblock.hpp
Fix for beta!=0 in reduce (#1440)
2024-08-06 09:10:39 -07:00
gridwise_2d_multiple_reduction_threadwise.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_2d_reduction_multiblock.hpp
Fix for beta!=0 in reduce (#1440)
2024-08-06 09:10:39 -07:00
gridwise_2d_reduction_threadwise_multi_d.hpp
Universal gemm splitk using reduce (with multi-d) (#1341)
2024-07-19 22:01:22 +08:00
gridwise_2d_reduction_threadwise.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_batched_gemm_gemm_xdl_cshuffle_v1.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_batched_gemm_multiple_d_gemm_multiple_d_xdl_cshuffle_v1.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_batched_gemm_multiple_d_softmax_gemm_xdl_cshuffle_v1.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_batched_gemm_softmax_gemm_wmma_cshuffle.hpp
Merging the gfx12 code into public repo. (#1362)
2024-06-27 00:33:34 -07:00
gridwise_batched_gemm_softmax_gemm_xdl_cshuffle_v1.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_batchnorm_backward_blockwise_welford.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_batchnorm_forward_blockwise_welford.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_elementwise_1d_scale.hpp
Refactor elementwise kernels (#1222)
2024-04-19 13:31:17 +02:00
gridwise_elementwise_2d.hpp
Grouped conv backward data GKCYX support (#2029)
2025-04-01 13:24:38 -07:00
gridwise_elementwise_layernorm_welford_variance.hpp
enable compilation of INSTANCES_ONLY for Windows (#1082)
2023-12-20 14:34:53 -08:00
gridwise_fpAintB_gemm_wmma.hpp
Merging the gfx12 code into public repo. (#1362)
2024-06-27 00:33:34 -07:00
gridwise_gemm_bias_add_reduce_xdl_cshuffle_v1.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_dl_multiple_d.hpp
Add Grouped Conv Fwd Large Tensor kernel (#1432)
2024-08-06 10:06:10 +02:00
gridwise_gemm_dl_v1r3.hpp
Add Grouped Conv Fwd Large Tensor kernel (#1432)
2024-08-06 10:06:10 +02:00
gridwise_gemm_dpp.hpp
Add support for more Navi2x and Navi3x models. (#1152)
2024-02-02 11:35:26 -08:00
gridwise_gemm_multiple_abd_xdl_cshuffle.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_multiple_d_multiple_r_xdl_cshuffle.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_multiple_d_wmma_cshuffle.hpp
Statically Cast Pointer Offset (#1631)
2024-11-05 09:59:08 -08:00
gridwise_gemm_multiple_d_xdl_cshuffle_lds_direct_load.hpp
Refactor thread_copy_lds_direct_load; fix gfx942 direct lds load example; fix f16_pki4 example
2025-06-03 14:54:30 +00:00
gridwise_gemm_multiple_d_xdl_cshuffle.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_multiple_d_xdl_splitk_cshuffle.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_pipeline_selector.hpp
Rebase the PR #1520 to ROCm repo. (#1574)
2025-02-20 18:58:14 -08:00
gridwise_gemm_pipeline_v1.hpp
Navi3 rel (#1176)
2024-03-08 17:11:51 -08:00
gridwise_gemm_pipeline_v2.hpp
Fixed Gemmv2r3 kpad (#938)
2023-09-26 18:40:00 -05:00
gridwise_gemm_pipeline_v3.hpp
initial stream-k implementation with example (#699)
2023-07-26 14:18:15 -05:00
gridwise_gemm_pipeline_v4_direct_load.hpp
Add support for double buffering in direct load GEMM kernel (#1052)
2023-12-03 23:08:47 +01:00
gridwise_gemm_reduce_xdl_cshuffle_v1.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_split_k_multiple_d_xdl_cshuffle_v2.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_split_k_multiple_d_xdl_cshuffle.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_waveletmodel.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_gemm_wmma_cshuffle_v3.hpp
DeviceGemm_Wmma_CShuffleV3 with BlockGemmPipelineVersion::v3 (#2096)
2025-04-28 10:14:21 +05:00
gridwise_gemm_wmma.hpp
Merging the gfx12 code into public repo. (#1362)
2024-06-27 00:33:34 -07:00
gridwise_gemm_xdl_cshuffle_conv_v3.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_xdl_cshuffle_streamk_v3.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_xdl_cshuffle_v1.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_xdl_cshuffle_v2.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_xdl_cshuffle_v3_b_preshuffle.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_xdl_cshuffle_v3_b_scale.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_xdl_cshuffle_v3_multi_abd.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_xdl_cshuffle_v3_multi_d_ab_scale.hpp
remove single rate mfma restriction for f8 blockscale gemm
2025-05-29 10:13:50 +00:00
gridwise_gemm_xdl_cshuffle_v3_multi_d_b_preshuffle.hpp
doc the kGroup definition
2025-06-03 08:53:04 +00:00
gridwise_gemm_xdl_cshuffle_v3_multi_d_blockscale_b_preshuffle.hpp
doc the kGroup definition
2025-06-03 08:53:04 +00:00
gridwise_gemm_xdl_cshuffle_v3_multi_d.hpp
clang format, remove single rate mfma restriction for f8
2025-05-29 10:10:12 +00:00
gridwise_gemm_xdl_cshuffle_v3_mx_bpreshuffle.hpp
Use packed_size_v for A/BPackedSize
2025-06-03 06:10:07 +00:00
gridwise_gemm_xdl_cshuffle_v3_mx.hpp
Use packed_size_v for A/BPackedSize
2025-06-03 06:10:07 +00:00
gridwise_gemm_xdl_cshuffle_v3.hpp
clang format, remove single rate mfma restriction for f8
2025-05-29 10:10:12 +00:00
gridwise_gemm_xdl_layernorm_cshuffle_v1.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_xdl_waveletmodel_cshuffle.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_xdlops_bwd_weight.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_xdlops_skip_b_lds_v1.hpp
Add support for more Navi2x and Navi3x models. (#1152)
2024-02-02 11:35:26 -08:00
gridwise_gemm_xdlops_splitk_lds_direct_load.hpp
Refactor thread_copy_lds_direct_load; fix gfx942 direct lds load example; fix f16_pki4 example
2025-06-03 14:54:30 +00:00
gridwise_gemm_xdlops_streamk.hpp
Add support for more Navi2x and Navi3x models. (#1152)
2024-02-02 11:35:26 -08:00
gridwise_gemm_xdlops_v2r3.hpp
Change flag to CK_GFX90A_DENORM_WORKAROUND (#1817)
2025-01-28 09:58:39 -05:00
gridwise_gemm_xdlops_v2r4.hpp
Add support for more Navi2x and Navi3x models. (#1152)
2024-02-02 11:35:26 -08:00
gridwise_gemm_xdlops_v2r4r2.hpp
Split env.hpp header from the ck.hpp header. (#2049)
2025-04-03 15:30:21 -07:00
gridwise_gemm_xdlops_v3r1.hpp
Use new mfma instructions for FP8 on gfx950 (#2202)
2025-05-19 17:29:51 -07:00
gridwise_gemm_xdlops_v3r2.hpp
Add support for more Navi2x and Navi3x models. (#1152)
2024-02-02 11:35:26 -08:00
gridwise_gemm_xdlops_v3r3.hpp
Add support for more Navi2x and Navi3x models. (#1152)
2024-02-02 11:35:26 -08:00
gridwise_moe_gemm_blockscale.hpp
doc the kGroup definition
2025-06-03 08:53:04 +00:00
gridwise_moe_gemm.hpp
doc the kGroup definition
2025-06-03 08:53:04 +00:00
gridwise_moe_mx_gemm_bns.hpp
fixed gemm1 LDS allocation issue
2025-06-12 04:08:18 -05:00
gridwise_moe_mx_gemm_bpreshuffle.hpp
moe gemm2 bpreshuffle function passed
2025-06-18 00:23:44 -05:00
gridwise_permute.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_put_element_1d.hpp
Average pool backward deviceOP and example (#797)
2023-08-10 12:04:35 +08:00
gridwise_set_buffer_value.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_set_multiple_buffer_value.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_softmax.hpp
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gridwise_sparse_embeddings_forward_layernorm_builtins.hpp
Replace buffer load/store intrinsics with builtins (#1876)
2025-03-05 14:33:28 -08:00
gridwise_sparse_embeddings_forward_layernorm.hpp
[HotFix] add config and version files to pass on build info (#856)
2023-08-23 11:36:17 -07:00
gridwise_tensor_rearrange.hpp
Statically Cast Pointer Offset (#1631)
2024-11-05 09:59:08 -08:00