Jianfeng Yan 34c661e71c 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

2022-03-30 11:21:18 -05:00

..

CMakeLists.txt

Gemm+Reduce Fusion (#128)

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

gemm_reduce_fp16.cpp

Batched gemm and reduction (#156)

2022-03-30 11:21:18 -05:00