ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-29 19:28:33 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
0fdca0e940ae8901afc3b6c23efb7cd721046de4
composable_kernel/example
History
Qianfeng Zhang 0fdca0e940 Replace template kUseSoftmax/kStoreLSE by boolean parameters in reference fwd codes to save compiling time
2026-06-05 15:51:39 +00:00
..
01_gemm
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
02_gemm_bilinear
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
03_gemm_bias_relu
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
04_gemm_add_add_fastgelu
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
09_convnd_fwd
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
10_convnd_fwd_multiple_d_multiple_reduce
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
11_convnd_fwd_bias
…
12_reduce
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
13_pool2d_fwd
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
14_gemm_quantization
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
15_grouped_gemm
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
16_gemm_multi_d_multi_reduces
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
17_convnd_bwd_data
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
18_batched_gemm_reduce
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
19_binary_elementwise
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
20_grouped_conv_bwd_weight
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
21_gemm_layernorm
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
22_cgemm
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
23_softmax
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
24_batched_gemm
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
25_gemm_bias_e_permute
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
26_contraction
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
27_layernorm2d_fwd
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
28_grouped_gemm_bias_e_permute
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
29_batched_gemm_bias_e_permute
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
30_grouped_conv_fwd_multiple_d
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
31_batched_gemm_gemm
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
32_batched_gemm_scale_softmax_gemm
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
33_multiple_reduce
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
34_batchnorm
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
35_splitK_gemm
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
36_sparse_embedding
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
37_batched_gemm_add_add_relu_gemm_add
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
38_grouped_conv_bwd_data_multiple_d
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
39_permute
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
40_conv2d_fwd_quantization
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
41_grouped_conv_conv_fwd
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
42_groupnorm_fwd
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
43_splitk_gemm_bias_e_permute
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
44_elementwise_permute
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
45_elementwise_normalization
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
46_gemm_add_multiply
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
47_gemm_bias_softmax_gemm_permute
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
48_pool3d_fwd
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
49_maxpool2d_bwd
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
50_put_element
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
51_avgpool3d_bwd
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
52_im2col_col2im
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
53_layernorm2d_bwd
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
54_groupnorm_bwd
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
59_grouped_gemm_multi_ABD
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
60_gemm_multi_ABD
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
61_contraction_multi_ABD
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
62_convnd_activ
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
63_layernorm4d_fwd
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
64_fpAintB_gemm
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
65_gemm_multiply_multiply
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
66_complex_contraction_bilinear
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
67_gemm_microscaling
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
68_gemm_add
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
69_gemm_add_relu
[CK_BUILDER] Put global CK functions in an the CK namespace (#3232)
2025-11-19 11:23:02 +01:00
ck_tile
Replace template kUseSoftmax/kStoreLSE by boolean parameters in reference fwd codes to save compiling time
2026-06-05 15:51:39 +00:00
CMakeLists.txt
…
README.md
…

README.md

Back to the main page

Composable Kernel examples

Reference in New Issue View Git Blame Copy Permalink