Jianfeng Yan d91f9f119c Batched gemm bf16 (#142) ...

* add bf16 for batched gemm

* batched_gemm_bf16 works

* recover accidently changed files

2022-03-22 18:18:43 -05:00

..

profile_batched_gemm_impl.hpp

Batched gemm bf16 (#142)

2022-03-22 18:18:43 -05:00

profile_conv_bwd_data_impl.hpp

Fix conv2d bwd data bug when filter is 1x1 and stride = 2 (#132)

2022-03-21 10:53:23 -05:00

profile_conv_fwd_bias_relu_add_impl.hpp

GEMM+Bias+ReLU+Add (#76)

2022-02-06 22:32:47 -06:00

profile_conv_fwd_bias_relu_atomic_add_impl.hpp

Fusion Conv+Bias+ReLU(+Add) (#62)

2021-12-26 07:43:42 -07:00

profile_conv_fwd_bias_relu_impl.hpp

GEMM+Bias+ReLU+Add (#76)

2022-02-06 22:32:47 -06:00

profile_conv_fwd_impl.hpp

[Bf16 & int8] [example & ckprofiler] (#100)

2022-03-04 15:56:44 -06:00

profile_gemm_bias_2d_impl.hpp

Gemm alpha beta profiler (fp32 & fp16) (#91)

2022-02-21 11:35:21 -06:00

profile_gemm_bias_relu_add_impl.hpp

GEMM+Bias+ReLU+Add (#76)

2022-02-06 22:32:47 -06:00

profile_gemm_bias_relu_impl.hpp

GEMM+Bias+ReLU+Add (#76)

2022-02-06 22:32:47 -06:00

profile_gemm_impl.hpp

Gemm_c_shuffle (4 layouts) X (fp32 bf16 int8) (#131)

2022-03-21 15:59:51 -05:00

profile_grouped_gemm_impl.hpp

Grouped GEMM for fp16 (#126)

2022-03-22 18:18:18 -05:00

profile_reduce_impl.hpp

Reduction for int8 and bfloat16 (#125)

2022-03-22 14:35:14 -05:00