Anthony Chang f015c77687 use single threaded tensor generator (#161)

2022-03-30 22:28:30 -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

use single threaded tensor generator (#161)

2022-03-30 22:28:30 -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