Skip to content

GitLab

"projects/vscode:/vscode.git/clone" did not exist on "bf2ff5a2412a03e707effa52c2e7b75954967cb8"
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_impl.hpp 19.8 KB