Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "10f88ef6b0136cb5be49210b042fc765b9779122"
Switch branch/tag
Find file BlameHistoryPermalink
  • Jianfeng Yan's avatar
    Batched gemm and reduction (#156) · 34c661e7
    Jianfeng Yan authored 
    * 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
    34c661e7
profile_batched_gemm_reduce.cpp 5.89 KB