ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-06 15:54:31 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
af493e6fcabce8f3572d41149e94e048b3ce16ff
composable_kernel/example
History
Bartłomiej Kocot 6b6fcd370b [CK TILE] Check for num loop < Prefetch Stages in gemm (#1886)
2025-02-14 09:15:27 +01:00
..
01_gemm
Merge from internal (#1857)
2025-02-07 15:05:05 -07:00
02_gemm_bilinear
…
03_gemm_bias_relu
…
04_gemm_add_add_fastgelu
Merge from internal (#1857)
2025-02-07 15:05:05 -07:00
09_convnd_fwd
…
10_convnd_fwd_multiple_d_multiple_reduce
…
12_reduce
…
13_pool2d_fwd
…
14_gemm_quantization
…
15_grouped_gemm
…
16_gemm_multi_d_multi_reduces
…
17_convnd_bwd_data
…
18_batched_gemm_reduce
Merge from internal (#1857)
2025-02-07 15:05:05 -07:00
19_binary_elementwise
…
20_grouped_conv_bwd_weight
…
21_gemm_layernorm
…
22_cgemm
…
23_softmax
…
24_batched_gemm
Added Int4 mixed batch gemm support (#1839)
2025-02-10 11:17:02 +08:00
25_gemm_bias_e_permute
…
26_contraction
…
27_layernorm2d_fwd
…
28_grouped_gemm_bias_e_permute
…
29_batched_gemm_bias_e_permute
…
30_grouped_conv_fwd_multiple_d
Merge from internal (#1857)
2025-02-07 15:05:05 -07:00
31_batched_gemm_gemm
Merge from internal (#1857)
2025-02-07 15:05:05 -07:00
32_batched_gemm_scale_softmax_gemm
…
33_multiple_reduce
…
34_batchnorm
…
35_splitK_gemm
…
36_sparse_embedding
…
37_batched_gemm_add_add_relu_gemm_add
…
38_grouped_conv_bwd_data_multiple_d
…
39_permute
…
40_conv2d_fwd_quantization
…
41_grouped_conv_conv_fwd
Merge from internal (#1857)
2025-02-07 15:05:05 -07:00
42_groupnorm_fwd
…
43_splitk_gemm_bias_e_permute
…
44_elementwise_permute
…
45_elementwise_normalization
…
46_gemm_add_multiply
…
47_gemm_bias_softmax_gemm_permute
…
48_pool3d_fwd
…
49_maxpool2d_bwd
…
50_put_element
…
51_avgpool3d_bwd
…
52_im2col_col2im
…
53_layernorm2d_bwd
…
54_groupnorm_bwd
…
59_grouped_gemm_multi_ABD
…
60_gemm_multi_ABD
…
61_contraction_multi_ABD
…
62_convnd_activ
Merge from internal (#1857)
2025-02-07 15:05:05 -07:00
63_layernorm4d_fwd
…
64_fpAintB_gemm
…
65_gemm_multiply_multiply
…
66_complex_contraction_bilinear
…
67_gemm_microscaling
Add a host mx gemm reference kernel (#1864)
2025-02-07 15:07:07 -07:00
ck_tile
[CK TILE] Check for num loop < Prefetch Stages in gemm (#1886)
2025-02-14 09:15:27 +01:00
CMakeLists.txt
Merge from internal (#1857)
2025-02-07 15:05:05 -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