Enrico Degregori 12225ce645 Wmma support for multiple ABD GEMM (#2803) ...

* multi_abd wmma support:

 - Add multiple A and B support to multiple D implementation (gridwise level)
 - Add multi_abd GEMM (device level)
 - Add instances (xdl parity)
 - Add tests (both xdl and wmma)
 - Add examples
 - Add ckProfiler support (both xdl and wmma)

* Fix bug in device print function

* Fix unused template parameter

* Fix batched gemm for multiABD gridwise implementation

* Fix gemm_universal_reduce with multiABDs gridwise implementation

---------

Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com>

[ROCm/composable_kernel commit: 3d29bff2f0]

2025-09-22 18:49:06 -07:00

..

data_type_enum.hpp

Pool2d max/avg kernel in the BWD version (#1494)

2024-09-12 11:47:52 +02:00

profile_avg_pool2d_bwd_impl.hpp

Pool2d max/avg kernel in the BWD version (#1494)

2024-09-12 11:47:52 +02:00

profile_avg_pool3d_bwd_impl.hpp

MaxPool & AvgPool bwd instances, test, ckProfiler, client example (#861)

2023-08-31 21:01:50 +08:00

profile_batched_gemm_add_relu_gemm_add_impl.hpp

update copyright headers (#726)

2023-05-31 18:46:57 -05:00

profile_batched_gemm_b_scale_impl.hpp

Added Int4 mixed batch gemm support (#1839)

2025-02-10 11:17:02 +08:00

profile_batched_gemm_bias_softmax_gemm_permute_impl.hpp

OCP FP8 support for gfx12. (#1710)

2024-12-03 08:42:55 -08:00

profile_batched_gemm_gemm_impl.hpp

Implement batched gemm gemm for RDNA (3 and 4) (#2612)

2025-09-04 14:10:24 -07:00

profile_batched_gemm_impl.hpp

Fix argument order for calls to profile_batched_gemm_impl() (#2277)

2025-06-17 19:29:09 -07:00

profile_batched_gemm_reduce_impl.hpp

update copyright headers (#726)

2023-05-31 18:46:57 -05:00

profile_batched_gemm_softmax_gemm_impl.hpp

OCP FP8 support for gfx12. (#1710)

2024-12-03 08:42:55 -08:00

profile_batched_gemm_softmax_gemm_permute_impl.hpp

OCP FP8 support for gfx12. (#1710)

2024-12-03 08:42:55 -08:00

profile_batchnorm_backward_impl.hpp

update copyright headers (#726)

2023-05-31 18:46:57 -05:00

profile_batchnorm_forward_impl.hpp

update copyright headers (#726)

2023-05-31 18:46:57 -05:00

profile_batchnorm_infer_impl.hpp

update copyright headers (#726)

2023-05-31 18:46:57 -05:00

profile_contraction_impl.hpp

Extend support for contraction 6D (#1207)

2024-04-09 23:46:21 +02:00

profile_contraction_utils.hpp

Extend support for contraction 6D (#1207)

2024-04-09 23:46:21 +02:00

profile_conv_bwd_data_impl.hpp

upgrade from clang-format-12 to clang-format-18 (#2568)

2025-07-28 11:34:07 -07:00

profile_conv_fwd_bias_relu_add_impl.hpp

update copyright headers (#726)

2023-05-31 18:46:57 -05:00

profile_conv_fwd_bias_relu_impl.hpp

update copyright headers (#726)

2023-05-31 18:46:57 -05:00

profile_conv_fwd_impl.hpp

upgrade from clang-format-12 to clang-format-18 (#2568)

2025-07-28 11:34:07 -07:00

profile_conv_tensor_rearrange_impl.hpp

upgrade from clang-format-12 to clang-format-18 (#2568)

2025-07-28 11:34:07 -07:00

profile_elementwise_layernorm_impl.hpp

Clip fp8 to +/-240 on all targets. (#1172)

2024-02-27 14:31:05 -06:00

profile_gemm_ab_scale_impl.hpp

Mirchen/gemm blockscale wp segfault fix (#2638)

2025-08-19 01:19:17 -07:00

profile_gemm_add_add_fastgelu_impl.hpp

update copyright headers (#726)

2023-05-31 18:46:57 -05:00

profile_gemm_add_fastgelu_impl.hpp

Wmma support for multiple Ds based GEMMs (#2613)

2025-09-05 16:31:08 +02:00

profile_gemm_add_impl.hpp

Add support for mixed-precision f16bf16_int8 gemm (#1127)

2024-02-07 15:54:13 +01:00

profile_gemm_add_multiply_impl.hpp

Wmma support for multiple Ds based GEMMs (#2613)

2025-09-05 16:31:08 +02:00

profile_gemm_add_relu_add_layernorm_impl.hpp

Layernorm and groupnorm support to save mean and inverse std in forward (#929)

2023-10-19 07:36:29 +08:00

profile_gemm_add_relu_impl.hpp

Add support for mixed-precision f16bf16_int8 gemm (#1127)

2024-02-07 15:54:13 +01:00

profile_gemm_add_silu_impl.hpp

Add support for mixed-precision f16bf16_int8 gemm (#1127)

2024-02-07 15:54:13 +01:00

profile_gemm_b_scale_impl.hpp

Support b_scale: (#2350)

2025-07-24 18:49:58 -07:00

profile_gemm_bias_add_reduce_impl.hpp

Mirchen/gemm blockscale wp segfault fix (#2638)

2025-08-19 01:19:17 -07:00

profile_gemm_bilinear_impl.hpp

update copyright headers (#726)

2023-05-31 18:46:57 -05:00

profile_gemm_blockscale_wp_impl.hpp

Fix typo in profiler/include/profiler/profile_gemm_blockscale_wp_impl.hpp (#2767)

2025-09-03 00:12:24 +08:00

profile_gemm_fastgelu_impl.hpp

update copyright headers (#726)

2023-05-31 18:46:57 -05:00

profile_gemm_impl.hpp

Mirchen/gemm blockscale wp segfault fix (#2638)

2025-08-19 01:19:17 -07:00

profile_gemm_multi_abd_impl.hpp

Wmma support for multiple ABD GEMM (#2803)

2025-09-22 18:49:06 -07:00

profile_gemm_multiply_add_impl.hpp

fixed fp8 issues (#894)

2023-09-12 22:17:56 -05:00

profile_gemm_multiply_multiply_impl.hpp

[A8W8 GEMM] Optimized weight-preshuffled implementation & add quantization datatype for CK TILE rms_norm (#1862)

2025-02-20 14:00:27 -08:00

profile_gemm_multiply_multiply_wp_impl.hpp

[Block Scale GEMM] Optimized block scale gemm (#1950)

2025-03-11 10:11:21 -07:00

profile_gemm_mx_impl.hpp

MX GEMM - Add FP6 GEMM Test (#2488)

2025-07-11 15:32:12 -06:00

profile_gemm_quantization_impl.hpp

Added wmma support for gemm quantization: (#2841)

2025-09-16 16:23:29 -07:00

profile_gemm_reduce_impl.hpp

Mirchen/gemm blockscale wp segfault fix (#2638)

2025-08-19 01:19:17 -07:00

profile_gemm_splitk_impl.hpp

Mirchen/gemm blockscale wp segfault fix (#2638)

2025-08-19 01:19:17 -07:00

profile_gemm_streamk_impl.hpp

Mirchen/gemm blockscale wp segfault fix (#2638)

2025-08-19 01:19:17 -07:00

profile_gemm_universal_batched_impl.hpp

Add SplitK support into Batched GEMM V3 (#1729)

2024-12-13 21:08:35 +01:00

profile_gemm_universal_impl.hpp

Mirchen/gemm blockscale wp segfault fix (#2638)

2025-08-19 01:19:17 -07:00

profile_gemm_universal_preshuffle_impl.hpp

Mirchen/gemm blockscale wp segfault fix (#2638)

2025-08-19 01:19:17 -07:00

profile_gemm_universal_reduce_impl.hpp

WMMA support for GEMM reduce (#2823)

2025-09-12 21:36:43 +02:00

profile_gemm_universal_streamk_impl.hpp

Mirchen/gemm blockscale wp segfault fix (#2638)

2025-08-19 01:19:17 -07:00

profile_grouped_conv_bwd_data_impl.hpp

upgrade from clang-format-12 to clang-format-18 (#2568)

2025-07-28 11:34:07 -07:00

profile_grouped_conv_bwd_weight_impl.hpp

Automatic deduction of split-K value for grouped convolution (#2491)

2025-07-31 12:08:45 +02:00

profile_grouped_conv_fwd_bias_bnorm_clamp_impl.hpp

Fix clang format after conv changes (#2636)

2025-08-07 10:00:09 +02:00

profile_grouped_conv_fwd_bias_clamp_impl.hpp

upgrade from clang-format-12 to clang-format-18 (#2568)

2025-07-28 11:34:07 -07:00

profile_grouped_conv_fwd_impl.hpp

upgrade from clang-format-12 to clang-format-18 (#2568)

2025-07-28 11:34:07 -07:00

profile_grouped_conv_fwd_outelementop_impl.hpp

upgrade from clang-format-12 to clang-format-18 (#2568)

2025-07-28 11:34:07 -07:00

profile_grouped_gemm_fastgelu_impl.hpp

update copyright headers (#726)

2023-05-31 18:46:57 -05:00

profile_grouped_gemm_fixed_nk_impl.hpp

Split env.hpp header from the ck.hpp header. (#2049)

2025-04-03 15:30:21 -07:00

profile_grouped_gemm_impl.hpp

Split env.hpp header from the ck.hpp header. (#2049)

2025-04-03 15:30:21 -07:00

profile_grouped_gemm_multiply_tile_loop_impl.hpp

Split env.hpp header from the ck.hpp header. (#2049)

2025-04-03 15:30:21 -07:00

profile_grouped_gemm_tile_loop_impl.hpp

Split env.hpp header from the ck.hpp header. (#2049)

2025-04-03 15:30:21 -07:00

profile_groupnorm_bwd_data_impl.hpp

layernorm and groupnorm backward data (#1083)

2023-12-19 04:23:11 +08:00

profile_groupnorm_bwd_gamma_beta_impl.hpp

layernorm & groupnorm bwd gamma beta (#1133)

2024-01-25 19:53:15 +08:00

profile_groupnorm_fwd_impl.hpp

Layernorm4d (#1022)

2023-11-09 08:34:51 +08:00

profile_layernorm_bwd_data_impl.hpp

layernorm and groupnorm backward data (#1083)

2023-12-19 04:23:11 +08:00

profile_layernorm_bwd_gamma_beta_impl.hpp

layernorm & groupnorm bwd gamma beta (#1133)

2024-01-25 19:53:15 +08:00

profile_layernorm_fwd_impl.hpp

Layernorm4d (#1022)

2023-11-09 08:34:51 +08:00

profile_max_pool2d_bwd_impl.hpp

Pool2d max/avg kernel in the BWD version (#1494)

2024-09-12 11:47:52 +02:00

profile_max_pool3d_bwd_impl.hpp

MaxPool & AvgPool bwd instances, test, ckProfiler, client example (#861)

2023-08-31 21:01:50 +08:00

profile_permute_scale_impl.hpp

Introduce combined elementwise ops (#1217)

2024-04-02 17:23:49 -05:00

profile_pool2d_fwd_impl.hpp

Rewrite pool2d fwd (#1462)

2024-09-11 15:21:00 +02:00

profile_pool3d_fwd_impl.hpp

Generic threshold calculation after merge fixes (#1618)

2024-11-06 10:44:58 +01:00

profile_reduce_impl.hpp

update copyright headers (#726)

2023-05-31 18:46:57 -05:00

profile_softmax_impl.hpp

upgrade from clang-format-12 to clang-format-18 (#2568)

2025-07-28 11:34:07 -07:00

profile_transpose_impl.hpp

Refactor elementwise kernels (#1222)

2024-04-19 13:31:17 +02:00