zjing14 716f1c7fb1 Grouped GEMM for fp16 (#126) ...

* init of grouped_gemm

* 2 gemm test

* perf test

* clean

* wrap desc into a struct

* test cast static_arr to pointer

* add ptr to GemmDesc

* add grouped gemm profiler

* fixed mem issue with unique_ptr

* clean

* clean

* finished ckprofiler

* Update README.md

* readme

* fixed readme

* add example

* improve code

* fixed comments: reserve, seperate ptr and gemm_shapes

* merge group and non-group

* fixed comments: replace push_back with emplace_back to avoid copy constructor

* fixed comments: unified blk2ctile; add test

* ci fix

* fixed ci

* fixed ci

* fixed ci

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

..

profile_batched_gemm_impl.hpp

refactored deviceBatchedGemm; removed GridwiseBatchedGemm; added fp32 and int8 to profiler (#120)

2022-03-21 16:45:14 -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