ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 10:09:41 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
6a3f751bf759853fa74d89c712b8687b2d971cbd
composable_kernel/example
History
Jianfeng Yan cb97ce68d8 Batched gemm and reduction (#156) 
* adding batched_gemm_and_reduction

* batched_gemm_reduce works with bactch_count=1

* fix a bug in grid_size; batched_gemm_reduce works for batch_count > 1

* adding profiler for batched_gemm_fp16

* fixed a bug in declaration of d1 and d0; both example and profiler work

* clang-format

* cleanup

* batched_gemm_reduce: add test

* minor change

* fixed some typo in function names

[ROCm/composable_kernel commit: 34c661e71c]
2022-03-30 11:21:18 -05:00
..
01_gemm
Batched gemm and reduction (#156)
2022-03-30 11:21:18 -05:00
02_gemm_alpha_beta
Gemm_c_shuffle (4 layouts) X (fp32 bf16 int8) (#131)
2022-03-21 15:59:51 -05:00
03_gemm_bias_relu
Gemm_c_shuffle (4 layouts) X (fp32 bf16 int8) (#131)
2022-03-21 15:59:51 -05:00
04_gemm_bias_relu_add
Gemm_c_shuffle (4 layouts) X (fp32 bf16 int8) (#131)
2022-03-21 15:59:51 -05:00
05_conv2d_fwd
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
06_conv2d_fwd_bias_relu
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
07_conv2d_fwd_bias_relu_add
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
08_conv3d_fwd
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
09_convnd_fwd
Unified conv3D API + support for all data types. (#133)
2022-03-23 10:23:13 -05:00
10_conv2d_bwd_data
Unified implementation of 1d/2d/3d conv bwd-data. fp32/fp16/bfp16/int8 (#134)
2022-03-29 10:52:25 -05:00
11_conv2d_bwd_wgt
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
12_reduce
Reduction for int8 and bfloat16 (#125)
2022-03-22 14:35:14 -05:00
13_pool2d_fwd
Reduction for int8 and bfloat16 (#125)
2022-03-22 14:35:14 -05:00
14_gemm_xdl_requant_relu_requant
Refine kernel parameter of int8 (ScalarPerVector) (#155)
2022-03-29 17:36:21 -05:00
15_grouped_gemm
fixed alloc mem size (#145)
2022-03-23 22:19:38 -05:00
16_gemm_reduce
Batched gemm and reduction (#156)
2022-03-30 11:21:18 -05:00
17_convnd_bwd_data_xdl
Unified implementation of 1d/2d/3d conv bwd-data. fp32/fp16/bfp16/int8 (#134)
2022-03-29 10:52:25 -05:00
18_batched_gemm_reduce
Batched gemm and reduction (#156)
2022-03-30 11:21:18 -05:00
CMakeLists.txt
Batched gemm and reduction (#156)
2022-03-30 11:21:18 -05:00