ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-11 17:00:18 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
a78be3f69e749e7a5e4d42f08415b0d13dbdb955
composable_kernel/example
History
zjing14 602c4cc0d9 Optimizing fp8_fp16 mixedprec gemm (#1150) 
* add delayed cvt

* extend fp16 gemm_splitk instances for fp8_fp16 gemm

* add f8 example

* add 128 kperblk instances for fp8

* add kpb128 instance

* added more instances into kpb128

* clean code

* clean code

* fix

* fix

* fixed

* Update example/35_splitK_gemm/splitK_gemm_xdl_fp16_fp8.cpp

Co-authored-by: Bartłomiej Kocot <barkocot@amd.com>

* Update include/ck/tensor_operation/gpu/thread/threadwise_tensor_slice_transfer.hpp

Co-authored-by: Bartłomiej Kocot <barkocot@amd.com>

* Update library/src/tensor_operation_instance/gpu/gemm_splitk/device_gemm_xdl_splitk_f16_fp8_f16_mk_nk_mn_kpb128_instance.cpp

Co-authored-by: Bartłomiej Kocot <barkocot@amd.com>

---------

Co-authored-by: Jing Zhang <jizha@amd.com>
Co-authored-by: Bartłomiej Kocot <barkocot@amd.com>
2024-02-12 09:45:42 -08:00
..
01_gemm
Fixing most of the cppcheck errors. (#1142)
2024-01-24 13:47:48 -08:00
02_gemm_bilinear
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
03_gemm_bias_relu
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
04_gemm_add_add_fastgelu
Fixing most of the cppcheck errors. (#1142)
2024-01-24 13:47:48 -08:00
09_convnd_fwd
Introduce multiABD api and deprecate multiD (#1035)
2023-11-14 17:00:40 +01:00
10_convnd_fwd_multiple_d_multiple_reduce
Fixing most of the cppcheck errors. (#1142)
2024-01-24 13:47:48 -08:00
12_reduce
Change 1d,2d,... to 1D,2D,... (#997)
2023-10-19 16:53:18 +02:00
13_pool2d_fwd
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
14_gemm_quantization
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
15_grouped_gemm
Fixed GroupedGemmFixedNK with hipGraph (#1065)
2023-11-30 15:09:27 -06:00
16_gemm_multi_d_multi_reduces
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
17_convnd_bwd_data
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
18_batched_gemm_reduce
Fixing most of the cppcheck errors. (#1142)
2024-01-24 13:47:48 -08:00
19_binary_elementwise
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
20_grouped_conv_bwd_weight
Fix bf8 conversion issues (#1003)
2023-10-20 08:00:45 -05:00
21_gemm_layernorm
Layernorm and groupnorm support to save mean and inverse std in forward (#929)
2023-10-19 07:36:29 +08:00
22_cgemm
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
23_softmax
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
24_batched_gemm
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
25_gemm_bias_e_permute
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
26_contraction
Add support for mixed precision in contraction scale and bilinear (#973)
2023-11-02 14:26:33 -07:00
27_layernorm2d_fwd
Split the static library into several files. (#1044)
2023-11-28 11:17:37 -08:00
28_grouped_gemm_bias_e_permute
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
29_batched_gemm_bias_e_permute
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
30_grouped_conv_fwd_multiple_d
Fixing most of the cppcheck errors. (#1142)
2024-01-24 13:47:48 -08:00
31_batched_gemm_gemm
Fixing most of the cppcheck errors. (#1142)
2024-01-24 13:47:48 -08:00
32_batched_gemm_scale_softmax_gemm
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
33_multiple_reduce
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
34_batchnorm
Batchnorm splitk single kernel (#771)
2023-07-06 10:58:55 -05:00
35_splitK_gemm
Optimizing fp8_fp16 mixedprec gemm (#1150)
2024-02-12 09:45:42 -08:00
36_sparse_embedding
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
37_batched_gemm_add_add_relu_gemm_add
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
38_grouped_conv_bwd_data_multiple_d
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
39_permute
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00
40_conv2d_fwd_quantization
Introduce multiABD api and deprecate multiD (#1035)
2023-11-14 17:00:40 +01:00
41_grouped_conv_conv_fwd
Fixing most of the cppcheck errors. (#1142)
2024-01-24 13:47:48 -08:00
42_groupnorm_fwd
Split the static library into several files. (#1044)
2023-11-28 11:17:37 -08:00
43_splitk_gemm_bias_e_permute
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
44_elementwise_permute
Add missing copyrights in elementwise_permute examples (#1118)
2024-01-04 17:38:24 +01:00
45_elementwise_normalization
Layernorm and groupnorm support to save mean and inverse std in forward (#929)
2023-10-19 07:36:29 +08:00
46_gemm_add_multiply
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
47_gemm_bias_softmax_gemm_permute
Enable gfx941 and gfx942 architectures. (#752)
2023-06-15 08:20:59 -07:00
48_pool3d_fwd
Fixing most of the cppcheck errors. (#1142)
2024-01-24 13:47:48 -08:00
49_maxpool2d_bwd
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
50_put_element
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
51_avgpool3d_bwd
Fixing most of the cppcheck errors. (#1142)
2024-01-24 13:47:48 -08:00
52_im2col_col2im
Add support for groups in Img2Col/Col2Img (#1007)
2023-10-31 10:46:32 +01:00
53_layernorm2d_bwd
layernorm and groupnorm backward data (#1083)
2023-12-19 04:23:11 +08:00
54_groupnorm_bwd
layernorm and groupnorm backward data (#1083)
2023-12-19 04:23:11 +08:00
60_gemm_multi_ABD
Add contraction_multi_abd (#972)
2023-10-17 20:17:58 -05:00
61_contraction_multi_ABD
Add missing ComputeDatatype in contraction_multi_ABD_xdl_fp16 (#1024)
2023-11-03 08:22:11 -07:00
62_conv_fwd_activ
Support broadcast for bias in grouped conv fwd (#1081)
2023-12-08 11:07:42 +01:00
63_layernorm4d_fwd
Split the static library into several files. (#1044)
2023-11-28 11:17:37 -08:00
CMakeLists.txt
Fix cmake dtype check (#989)
2023-10-21 22:19:43 +02:00