ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-18 20:09:25 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
fd8983d063ec0487c4d8bde213270557d8c8ba15
composable_kernel/example
History
Illia Silin 27de1d431a Make sure gemm_fp8_pk_i4 examples only build and run on gfx942/950. (#2010) 
* only build gemm_fp8_pk_i4 examples for gfx942/950

* fix cmake logic

* moved the architecture check to IsSupported function

* Revert "moved the architecture check to IsSupported function"

This reverts commit 056d2a08b3.

[ROCm/composable_kernel commit: 99b2bbc1d6]
2025-03-25 14:43:38 -07:00
..
01_gemm
Make sure gemm_fp8_pk_i4 examples only build and run on gfx942/950. (#2010)
2025-03-25 14:43:38 -07:00
02_gemm_bilinear
…
03_gemm_bias_relu
…
04_gemm_add_add_fastgelu
remove support for gfx940 and gfx941 targets (#1944)
2025-03-05 11:07:33 -08:00
09_convnd_fwd
Add a daily CI build on gfx908. (#1987)
2025-03-17 18:08:53 -07:00
10_convnd_fwd_multiple_d_multiple_reduce
…
12_reduce
Support large: 12d tensor size for reduction kenrel (#1465)
2024-08-13 16:15:47 +02:00
13_pool2d_fwd
…
14_gemm_quantization
…
15_grouped_gemm
Async grouped gemm v3 (#1940)
2025-03-17 16:42:43 +01:00
16_gemm_multi_d_multi_reduces
…
17_convnd_bwd_data
…
18_batched_gemm_reduce
remove support for gfx940 and gfx941 targets (#1944)
2025-03-05 11:07:33 -08:00
19_binary_elementwise
…
20_grouped_conv_bwd_weight
Apply universal gemm to bwd_weight_cshuffle operator (#1873)
2025-02-18 10:10:22 +01:00
21_gemm_layernorm
OCP FP8 support for gfx12. (#1710)
2024-12-03 08:42:55 -08:00
22_cgemm
…
23_softmax
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
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
Code clean-up (#1285)
2024-05-10 09:41:39 -07:00
27_layernorm2d_fwd
…
28_grouped_gemm_bias_e_permute
…
29_batched_gemm_bias_e_permute
…
30_grouped_conv_fwd_multiple_d
Remove CK_USE_AMD_MFMA_GFX950 (#1935)
2025-03-04 10:32:25 -08:00
31_batched_gemm_gemm
Merge from internal (#1857)
2025-02-07 15:05:05 -07:00
32_batched_gemm_scale_softmax_gemm
OCP FP8 support for gfx12. (#1710)
2024-12-03 08:42:55 -08:00
33_multiple_reduce
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
34_batchnorm
Refactor elementwise kernels (#1222)
2024-04-19 13:31:17 +02:00
35_splitK_gemm
OCP FP8 support for gfx12. (#1710)
2024-12-03 08:42:55 -08:00
36_sparse_embedding
…
37_batched_gemm_add_add_relu_gemm_add
OCP FP8 support for gfx12. (#1710)
2024-12-03 08:42:55 -08:00
38_grouped_conv_bwd_data_multiple_d
OCP FP8 support for gfx12. (#1710)
2024-12-03 08:42:55 -08:00
39_permute
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
40_conv2d_fwd_quantization
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
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
Prevent instantiation of undefined FP8 operators. (#1639)
2024-11-05 13:58:29 -08:00
45_elementwise_normalization
…
46_gemm_add_multiply
…
47_gemm_bias_softmax_gemm_permute
OCP FP8 support for gfx12. (#1710)
2024-12-03 08:42:55 -08:00
48_pool3d_fwd
…
49_maxpool2d_bwd
…
50_put_element
…
51_avgpool3d_bwd
…
52_im2col_col2im
…
53_layernorm2d_bwd
layernorm and groupnorm backward data (#1083)
2023-12-19 04:23:11 +08:00
54_groupnorm_bwd
…
59_grouped_gemm_multi_ABD
OCP FP8 support for gfx12. (#1710)
2024-12-03 08:42:55 -08:00
60_gemm_multi_ABD
…
61_contraction_multi_ABD
…
62_convnd_activ
remove support for gfx940 and gfx941 targets (#1944)
2025-03-05 11:07:33 -08:00
63_layernorm4d_fwd
…
64_fpAintB_gemm
…
65_gemm_multiply_multiply
change cmake (#2006)
2025-03-20 19:25:11 -07:00
66_complex_contraction_bilinear
…
67_gemm_microscaling
MX GEMM examples with FP8, FP16, and E8M0 scales (#2016)
2025-03-25 15:33:03 -06:00
ck_tile
[CK_TILE] Improve RMS/Layer Normalization 2 Pass Pipeline Performance (#1861)
2025-03-25 20:09:45 +08:00
CMakeLists.txt
Make sure gemm_fp8_pk_i4 examples only build and run on gfx942/950. (#2010)
2025-03-25 14:43:38 -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