ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 02:02:46 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
e7ef841a6813504d6a7bf8c1ea2f276bcb6a416e
composable_kernel/example
History
Sami Remes e7ef841a68 Add KBatch support for gemm_ab_scale (#2740) 
* Add KBatch support for gemm_ab_scale

* Revert kernel parameters change

* Remove printing

* fix formatting

* fix check

* Use {} in if

---------

Co-authored-by: Adam Osewski <19374865+aosewski@users.noreply.github.com>

[ROCm/composable_kernel commit: 9d4bfe3932]
2025-10-09 08:33:16 +02:00
..
01_gemm
[CK][Examples] Extending support for rdna3/4 part 3:
2025-10-08 18:14:38 +02:00
02_gemm_bilinear
Extend XDL kernel to Support RDNA3/4 - Part 4 (#2724)
2025-09-12 08:17:07 -07:00
03_gemm_bias_relu
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
04_gemm_add_add_fastgelu
[CK][Examples] Extending support for rdna3/4 part 3:
2025-10-08 18:14:38 +02:00
09_convnd_fwd
TF32 POC in Conv3d on MI30x platform #2763 (second attempt) (#2852)
2025-09-17 14:50:15 -07:00
10_convnd_fwd_multiple_d_multiple_reduce
[CK] Fix misc issues in CK examples (#2890)
2025-09-24 11:28:20 -07:00
12_reduce
[CK] Add command option instance_index and param_mask to run partial ck test (#2889)
2025-09-30 08:24:40 -07:00
13_pool2d_fwd
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
14_gemm_quantization
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
15_grouped_gemm
[CK][Examples] Extending support for rdna3/4 part 3:
2025-10-08 18:14:38 +02:00
16_gemm_multi_d_multi_reduces
Extend XDL kernel to Support RDNA3/4 - Part 4 (#2724)
2025-09-12 08:17:07 -07:00
17_convnd_bwd_data
Add Grouped Conv Fwd Large Tensor kernel (#1432)
2024-08-06 10:06:10 +02:00
18_batched_gemm_reduce
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
19_binary_elementwise
…
20_grouped_conv_bwd_weight
[CK][Examples] Extending support for rdna3/4 part 3:
2025-10-08 18:14:38 +02:00
21_gemm_layernorm
[CK] Fix misc issues in CK examples (#2890)
2025-09-24 11:28:20 -07:00
22_cgemm
[CK][Examples] Extending support for rdna3/4 part 3:
2025-10-08 18:14:38 +02:00
23_softmax
…
24_batched_gemm
[CK][Examples] Extending support for rdna3/4 part 3:
2025-10-08 18:14:38 +02:00
25_gemm_bias_e_permute
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
26_contraction
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
27_layernorm2d_fwd
upgrade from clang-format-12 to clang-format-18 (#2568)
2025-07-28 11:34:07 -07:00
28_grouped_gemm_bias_e_permute
[CK][Examples] Extending support for rdna3/4 in following examples: (#2884)
2025-09-29 09:05:04 -07:00
29_batched_gemm_bias_e_permute
[CK][Examples] Extending support for rdna3/4 in following examples: (#2884)
2025-09-29 09:05:04 -07:00
30_grouped_conv_fwd_multiple_d
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
31_batched_gemm_gemm
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
32_batched_gemm_scale_softmax_gemm
[CK][Examples] Extending support for rdna3/4 in following examples: (#2884)
2025-09-29 09:05:04 -07:00
33_multiple_reduce
…
34_batchnorm
Fix build errors on windows (#2456)
2025-07-16 07:58:23 -07:00
35_splitK_gemm
[CK][Examples] Extending support for rdna3/4 in following examples: (#2884)
2025-09-29 09:05:04 -07:00
36_sparse_embedding
[CK][Examples] Extending support for rdna3/4 part 2:
2025-10-08 18:14:38 +02:00
37_batched_gemm_add_add_relu_gemm_add
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
38_grouped_conv_bwd_data_multiple_d
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
39_permute
[CK][EXAMPLES] (#2826)
2025-09-11 12:33:00 -07:00
40_conv2d_fwd_quantization
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
41_grouped_conv_conv_fwd
Extend XDL kernel to Support RDNA3/4 - Part 4 (#2724)
2025-09-12 08:17:07 -07:00
42_groupnorm_fwd
[CK][EXAMPLES] (#2826)
2025-09-11 12:33:00 -07:00
43_splitk_gemm_bias_e_permute
[CK][Examples] Extending support for rdna3/4 part 3:
2025-10-08 18:14:38 +02:00
44_elementwise_permute
[CK][Examples] Extending support for rdna3/4 part 2:
2025-10-08 18:14:38 +02:00
45_elementwise_normalization
[CK] Add command option instance_index and param_mask to run partial ck test (#2889)
2025-09-30 08:24:40 -07:00
46_gemm_add_multiply
[CK][Examples] Extending support for rdna3/4 in following examples: (#2884)
2025-09-29 09:05:04 -07:00
47_gemm_bias_softmax_gemm_permute
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
48_pool3d_fwd
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
49_maxpool2d_bwd
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
50_put_element
…
51_avgpool3d_bwd
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
52_im2col_col2im
…
53_layernorm2d_bwd
…
54_groupnorm_bwd
[CK] Add command option instance_index and param_mask to run partial ck test (#2889)
2025-09-30 08:24:40 -07:00
59_grouped_gemm_multi_ABD
OCP FP8 support for gfx12. (#1710)
2024-12-03 08:42:55 -08:00
60_gemm_multi_ABD
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
61_contraction_multi_ABD
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
62_convnd_activ
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
63_layernorm4d_fwd
upgrade from clang-format-12 to clang-format-18 (#2568)
2025-07-28 11:34:07 -07:00
64_fpAintB_gemm
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
65_gemm_multiply_multiply
Add KBatch support for gemm_ab_scale (#2740)
2025-10-09 08:33:16 +02:00
66_complex_contraction_bilinear
[CK][Examples] Extending support for rdna3/4 in following examples: (#2884)
2025-09-29 09:05:04 -07:00
67_gemm_microscaling
Verify HostTensorDescriptor when it is created (#2829)
2025-09-25 18:22:13 -07:00
68_gemm_add
Extend XDL kernel to Support RDNA3/4 - Part 4 (#2724)
2025-09-12 08:17:07 -07:00
69_gemm_add_relu
Extend XDL kernel to Support RDNA3/4 - Part 4 (#2724)
2025-09-12 08:17:07 -07:00
ck_tile
Add Memory pipeline for AQuant Block Scale GEMM (#2987)
2025-10-08 17:22:30 -07:00
CMakeLists.txt
Extend XDL kernel to Support RDNA3/4 - Part 5 (#2725)
2025-09-15 10:59:25 -07:00
README.md
Add basic documentation structure (#1715)
2024-12-04 00:46:47 +01:00

README.md

Back to the main page

Composable Kernel examples

Reference in New Issue View Git Blame Copy Permalink