ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-12 01:10:17 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
f477fca4361ed1ff7ff7913ba1eb75b7ffe3179d
composable_kernel/example
History
Rostyslav Geyyer bd09b5c538 Add fp8 @ bf8 gemm support and example (#933) 
* Add f8 bf8 gemm example

* Add element-wise ops

* Add intrinsics

* Update reference calculation

* Add an additional type option for xdlops gemm

* Fix build process

* Add bf8 to buffer addressing

* Update blockwise op, split typeA and typeB

* Update for compatibility

* Uppdate naming to f8->fp8

* Update naming

* Format
2023-10-02 16:39:03 -05:00
..
01_gemm
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05: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
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
09_convnd_fwd
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
10_convnd_fwd_multiple_d_multiple_reduce
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
12_reduce
update copyright headers (#726)
2023-05-31 18:46:57 -05: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
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
16_gemm_multi_d_multi_reduces
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07: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
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
19_binary_elementwise
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
20_grouped_conv_bwd_weight
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
21_gemm_layernorm
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
22_cgemm
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
23_softmax
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
24_batched_gemm
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07: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 (#936)
2023-09-29 10:54:31 -05:00
27_layernorm
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07: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
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
31_batched_gemm_gemm
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
32_batched_gemm_scale_softmax_gemm
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07: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
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07: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
Add grouped conv bwd data wmma (#950)
2023-09-28 23:10:18 +02:00
39_permute
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
40_conv2d_fwd_quantization
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
41_grouped_conv_conv_fwd
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
42_groupnorm
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07: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
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
45_elementwise_normalization
update copyright headers (#726)
2023-05-31 18:46:57 -05: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
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07: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
Average pool backward deviceOP and example (#797)
2023-08-10 12:04:35 +08:00
52_im2col_col2im
Add column to image kernel (#930)
2023-09-27 17:19:06 +02:00
60_gemm_multi_ABD
get rid of gfx900/906, set rocm5.7 as default (#958)
2023-10-02 12:01:11 -07:00
60_gemm_multiABD
Add multiple A/B support (#906)
2023-09-26 21:16:23 -05:00
61_contraction_multi_ABD
Contraction multi abd (#957)
2023-10-02 09:18:36 -05:00
CMakeLists.txt
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00