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
8e232849229b1fa3132aecb571e76651e8e42d06
composable_kernel/example
History
Enrico Degregori 8e23284922 Extend support for ak1 / bk1 WMMA (#3073) 
* Extend AK1 / BK1 support:

 - Add support for AK1 != BK1
 - Add support for AK1, BK1 > 8
 - Introduce KInner template parameter for pipelines when loading multiple tiles with one instruction

* fix clang format

[ROCm/composable_kernel commit: 1c544abf57]
2025-11-11 07:38:15 -08:00
..
01_gemm
Extend support for ak1 / bk1 WMMA (#3073)
2025-11-11 07:38:15 -08:00
02_gemm_bilinear
[CK][Examples] Fixing stride issues in ck examples by workaround - Bypassing hostTensor validation.
2025-10-23 08:46:02 +02:00
03_gemm_bias_relu
[CK][Examples] Fixing stride issues in ck examples by workaround - Bypassing hostTensor validation.
2025-10-23 08:46:02 +02:00
04_gemm_add_add_fastgelu
[CK][Examples] Fixing stride issues in ck examples by workaround - Bypassing hostTensor validation.
2025-10-23 08:46:02 +02:00
09_convnd_fwd
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
10_convnd_fwd_multiple_d_multiple_reduce
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
11_convnd_fwd_bias
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
12_reduce
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
13_pool2d_fwd
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
14_gemm_quantization
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
15_grouped_gemm
[CK][Examples] Fix for example_grouped_gemm_multiple_d_dl_fp16 - corrected stride for B matrix. (#3104)
2025-10-28 09:47:25 -07:00
16_gemm_multi_d_multi_reduces
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
17_convnd_bwd_data
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
18_batched_gemm_reduce
[CK][Examples] Fixing stride issues in ck examples by workaround - Bypassing hostTensor validation.
2025-10-23 08:46:02 +02:00
19_binary_elementwise
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
20_grouped_conv_bwd_weight
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
21_gemm_layernorm
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
22_cgemm
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
23_softmax
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
24_batched_gemm
[CK][Examples] Fixing stride issues in ck examples by workaround - Bypassing hostTensor validation.
2025-10-23 08:46:02 +02:00
25_gemm_bias_e_permute
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
26_contraction
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
27_layernorm2d_fwd
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
28_grouped_gemm_bias_e_permute
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
29_batched_gemm_bias_e_permute
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
30_grouped_conv_fwd_multiple_d
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
31_batched_gemm_gemm
Add copyright notices to missing files (#3133)
2025-10-31 07:35:11 -07:00
32_batched_gemm_scale_softmax_gemm
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
33_multiple_reduce
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
34_batchnorm
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
35_splitK_gemm
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
36_sparse_embedding
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
37_batched_gemm_add_add_relu_gemm_add
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
38_grouped_conv_bwd_data_multiple_d
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
39_permute
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
40_conv2d_fwd_quantization
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
41_grouped_conv_conv_fwd
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
42_groupnorm_fwd
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
43_splitk_gemm_bias_e_permute
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
44_elementwise_permute
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
45_elementwise_normalization
[CK][Examples] Fix for example_grouped_gemm_multiple_d_dl_fp16 - corrected stride for B matrix. (#3104)
2025-10-28 09:47:25 -07:00
46_gemm_add_multiply
[CK][Examples] Fixing stride issues in ck examples by workaround - Bypassing hostTensor validation.
2025-10-23 08:46:02 +02:00
47_gemm_bias_softmax_gemm_permute
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
48_pool3d_fwd
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
49_maxpool2d_bwd
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
50_put_element
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
51_avgpool3d_bwd
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
52_im2col_col2im
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
53_layernorm2d_bwd
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
54_groupnorm_bwd
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
59_grouped_gemm_multi_ABD
[ck] correct memory size in grouped_gemm_multi_abd_xdl_fixed_nk_bias_bf16_i8 (#3168)
2025-11-10 10:58:08 -08:00
60_gemm_multi_ABD
[CK][Examples] Fixing stride issues in ck examples by workaround - Bypassing hostTensor validation.
2025-10-23 08:46:02 +02:00
61_contraction_multi_ABD
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
62_convnd_activ
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
63_layernorm4d_fwd
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
64_fpAintB_gemm
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
65_gemm_multiply_multiply
[CK][Examples] Fixing stride issues in ck examples by workaround - Bypassing hostTensor validation.
2025-10-23 08:46:02 +02:00
66_complex_contraction_bilinear
[DOCS] Documentation Addition (Readme updates) (#2495)
2025-10-16 03:10:57 -07:00
67_gemm_microscaling
Ck moe mxfp4 blockm32 (#3098)
2025-11-07 08:45:41 +08:00
68_gemm_add
Add copyright notices to missing files (#3133)
2025-10-31 07:35:11 -07:00
69_gemm_add_relu
Add copyright notices to missing files (#3133)
2025-10-31 07:35:11 -07:00
ck_tile
fix input range (#3188)
2025-11-10 11:08:41 -08: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