ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 02:02:46 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
177648de02d732f126cdac7e10ad4433e2b6806e
composable_kernel/example
History
Chao Liu 177648de02 conv+conv (1x1 only) example using gemm+gemm (#393) 
* refactor conv

* add conv+conv example, 1x1 only

[ROCm/composable_kernel commit: 4df6d93f60]
2022-08-31 11:27:11 -05:00
..
01_gemm
Add examples of Gemm (data type: int4) (#367)
2022-08-23 18:25:05 -05:00
02_gemm_bilinear
Clean up conv example, Instances, profiler and test (#324)
2022-07-29 18:19:25 -05:00
03_gemm_bias_relu
Clean up conv example, Instances, profiler and test (#324)
2022-07-29 18:19:25 -05:00
04_gemm_add_add_fastgelu
Add example of Gemm + AddAddFastGelu (data type: int4) (#369)
2022-08-23 10:38:41 -05:00
09_convnd_fwd
conv+conv (1x1 only) example using gemm+gemm (#393)
2022-08-31 11:27:11 -05:00
12_reduce
Add int4 reduction examples (#372)
2022-08-25 16:58:48 -05:00
13_pool2d_fwd
Clean up conv example, Instances, profiler and test (#324)
2022-07-29 18:19:25 -05:00
14_gemm_xdl_requant_relu_requant
Clean up conv example, Instances, profiler and test (#324)
2022-07-29 18:19:25 -05:00
15_grouped_gemm
GEMM batched/splitK/cgemm/grouped int4 examples (#383)
2022-08-25 17:19:15 -05:00
16_gemm_multi_d_multi_reduces
Gemm reduce examples int4/int8/fp32/bf16 (#368)
2022-08-30 11:38:26 -05:00
17_convnd_bwd_data
Clean up conv example, Instances, profiler and test (#324)
2022-07-29 18:19:25 -05:00
18_batched_gemm_reduce
Fused attention (#345)
2022-08-13 00:16:14 -05:00
19_binary_elementwise
Batchnorm-forward and Batchnorm-infer Implemented using generic kernels (#320)
2022-08-15 10:11:02 -05:00
20_convnd_bwd_weight
Clean up conv example, Instances, profiler and test (#324)
2022-07-29 18:19:25 -05:00
21_gemm_layernorm
Batchnorm-forward and Batchnorm-infer Implemented using generic kernels (#320)
2022-08-15 10:11:02 -05:00
22_cgemm
GEMM batched/splitK/cgemm/grouped int4 examples (#383)
2022-08-25 17:19:15 -05:00
23_softmax
Clean up conv example, Instances, profiler and test (#324)
2022-07-29 18:19:25 -05:00
24_batched_gemm
GEMM batched/splitK/cgemm/grouped int4 examples (#383)
2022-08-25 17:19:15 -05:00
24_batched_gemm_e_permute
Add examples of batched/grouped/SplitK Gemm for int8/bfp16/fp16/fp32 (#361)
2022-08-23 14:41:56 -05:00
25_gemm_bias_e_permute
add g; fixed strides (#355)
2022-08-12 15:22:39 -05:00
26_contraction
Clean up conv example, Instances, profiler and test (#324)
2022-07-29 18:19:25 -05:00
27_layernorm
Layernorm welford (#346)
2022-08-13 09:43:18 -05:00
28_grouped_gemm_bias_e_permute
Add batched/grouped_gemm contraction deviceOps (#349)
2022-08-10 12:20:29 -05:00
29_batched_gemm_bias_e_permute
Add batched/grouped_gemm contraction deviceOps (#349)
2022-08-10 12:20:29 -05:00
30_grouped_convnd_fwd_bias_relu_add
conv+conv (1x1 only) example using gemm+gemm (#393)
2022-08-31 11:27:11 -05:00
31_batched_gemm_gemm
Attention with output permutation (#370)
2022-08-23 14:52:56 -05:00
32_batched_gemm_scale_softmax_gemm
Padding for attention: bmm+scale+softmax+bmm kernel (#385)
2022-08-30 11:01:37 -05:00
33_multiple_reduce
Batchnorm-forward and Batchnorm-infer Implemented using generic kernels (#320)
2022-08-15 10:11:02 -05:00
34_batchnorm
Batchnorm-forward and Batchnorm-infer Implemented using generic kernels (#320)
2022-08-15 10:11:02 -05:00
35_splitK_gemm
GEMM batched/splitK/cgemm/grouped int4 examples (#383)
2022-08-25 17:19:15 -05:00
41_grouped_conv_conv_fwd
conv+conv (1x1 only) example using gemm+gemm (#393)
2022-08-31 11:27:11 -05:00
CMakeLists.txt
conv+conv (1x1 only) example using gemm+gemm (#393)
2022-08-31 11:27:11 -05:00