Mingtao Gu 6fb0e00df5 Added Int4 mixed batch gemm support (#1839) ...

* remove redundant kernels.

* added batched_gemm_xdl_fp16int4_b_scale_v3

* Enabled the split K.

* added the batched_gemm_b_scale ckProfiler, meet function issue

* fix some typo

* fix ckProfiler build issue

* fix some bugs

* updated some debug info

* comment some code

* Fix

* fixed some bugs and refactor the code

* fixed a function bug.

* formatted files.

* formatted

* uncommented the ckProfiler CMakeLists

* fixed.

* fix ckProfiler for batched_gemm_b_scale

---------

Co-authored-by: mtgu0705 <mtgu@amd.com>
Co-authored-by: aska-0096 <haocwang@amd.com>
Co-authored-by: Bartlomiej Kocot <barkocot@amd.com>

[ROCm/composable_kernel commit: d9f1ead347]

2025-02-10 11:17:02 +08: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

OCP FP8 support for gfx12. (#1710)

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

profile_batched_gemm_impl.hpp

Add DeviceBatchedGemmMultipleD_Dl (#732)

2023-06-12 08:37:15 -05: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

Add Grouped Conv Fwd Large Tensor kernel (#1432)

2024-08-06 10:06:10 +02: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

Add Grouped Conv Fwd Large Tensor kernel (#1432)

2024-08-06 10:06:10 +02:00

profile_conv_tensor_rearrange_impl.hpp

Add support for groups in Img2Col/Col2Img (#1007)

2023-10-31 10:46:32 +01: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

[GEMM] F8 GEMM, performance optimized. (#1384)

2024-07-19 22:06:52 +08: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

update copyright headers (#726)

2023-05-31 18:46:57 -05: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

update copyright headers (#726)

2023-05-31 18:46:57 -05: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

fix a bug for int4 scale weight only kernel (#1820)

2025-01-19 11:18:18 +08:00

profile_gemm_bias_add_reduce_impl.hpp

update copyright headers (#726)

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

profile_gemm_bilinear_impl.hpp

update copyright headers (#726)

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

profile_gemm_fastgelu_impl.hpp

update copyright headers (#726)

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

profile_gemm_impl.hpp

OCP FP8 support for gfx12. (#1710)

2024-12-03 08:42:55 -08: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

add int8 gemm multiply multiply a8w8 (#1591)

2024-10-26 16:39:34 +08:00

profile_gemm_reduce_impl.hpp

update copyright headers (#726)

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

profile_gemm_splitk_impl.hpp

[GEMM] Optimization for MI200/300. (#1135)

2024-01-19 07:02:22 -06:00

profile_gemm_streamk_impl.hpp

add generic instances (#858)

2023-08-23 09:18:10 -05: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

Fix universal gemm profiler for pk_i4_t (#1790)

2025-01-04 14:01:33 +01:00

profile_gemm_universal_reduce_impl.hpp

Universal gemm splitk using reduce (with multi-d) (#1341)

2024-07-19 22:01:22 +08:00

profile_gemm_universal_streamk_impl.hpp

Universal streamk with atomics (#1360)

2024-07-05 21:40:30 -07:00

profile_grouped_conv_bwd_data_impl.hpp

Support bf16/f32/f16 and NHWGC conv2d_bwd_data (#757)

2023-06-21 08:20:31 -05:00

profile_grouped_conv_bwd_weight_impl.hpp

Grouped convolution backward weight special vector size loads (#1772)

2025-01-10 22:02:30 +08:00

profile_grouped_conv_fwd_impl.hpp

Add support for NGCHW in grouped conv fwd (#1499)

2024-09-20 10:45:46 +02:00

profile_grouped_conv_fwd_outelementop_impl.hpp

Add ckProfiler support for forward 3D convolutions with OUT element-wise operations. (#1354)

2024-07-08 10:55:54 -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

replace the ENV macro with CK_ENV (#1296)

2024-05-17 10:42:51 -07:00

profile_grouped_gemm_impl.hpp

fix profiler_grouped_gemm (#1766)

2024-12-19 17:24:05 -08:00

profile_grouped_gemm_multiply_tile_loop_impl.hpp

Polished Grouped GEMM APIs and new BF16 instances (#1600)

2024-11-27 13:02:44 +01:00

profile_grouped_gemm_tile_loop_impl.hpp

Polished Grouped GEMM APIs and new BF16 instances (#1600)

2024-11-27 13:02:44 +01: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

Padded Generic Kernel Instance (#730)

2023-06-16 23:43:11 -05:00

profile_transpose_impl.hpp

Refactor elementwise kernels (#1222)

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