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
9af519ee86aacf3929d032cb5a50599569d4d1fa
composable_kernel/example
History
Rostyslav Geyyer 9c54eaab04 Enable f16/f8 mixed precision mode (#820) 
* Enable f16/f8 mixed precision

* Add an argument to enable mixed precision

* Update for compatibility

* Add mixed precision example

* Introduce ComputeType argument
2023-08-09 08:44:23 -05:00
..
01_gemm
Enable f16/f8 mixed precision mode (#820)
2023-08-09 08:44:23 -05:00
02_gemm_bilinear
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
03_gemm_bias_relu
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
04_gemm_add_add_fastgelu
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
09_convnd_fwd
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
10_convnd_fwd_multiple_d_multiple_reduce
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
12_reduce
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
13_pool2d_fwd
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
14_gemm_quantization
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
15_grouped_gemm
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
16_gemm_multi_d_multi_reduces
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
17_convnd_bwd_data
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
18_batched_gemm_reduce
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
19_binary_elementwise
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
20_grouped_conv_bwd_weight
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
21_gemm_layernorm
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
22_cgemm
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
23_softmax
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
24_batched_gemm
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
25_gemm_bias_e_permute
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
26_contraction
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
27_layernorm
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
28_grouped_gemm_bias_e_permute
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
29_batched_gemm_bias_e_permute
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
30_grouped_conv_fwd_multiple_d
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
31_batched_gemm_gemm
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
32_batched_gemm_scale_softmax_gemm
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -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
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -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
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
38_grouped_conv_bwd_data_multiple_d
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
39_permute
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
40_conv2d_fwd_quantization
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
41_grouped_conv_conv_fwd
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
42_groupnorm
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
43_splitk_gemm_bias_e_permute
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
44_elementwise_permute
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
45_elementwise_normalization
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
46_gemm_add_multiply
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -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
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
49_maxpool2d_bwd
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
50_put_element
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
CMakeLists.txt
Add packages for examples and profiler (#502)
2022-11-10 13:19:33 -06:00