Skip to content

GitLab

"test/gemm_splitk/gemm_splitk_fp32.cpp" did not exist on "0619ebf70bc6d0bd8b44cb41b5a662ddfc4def56"
Switch branch/tag
Find file BlameHistoryPermalink
  • Qianfeng's avatar
    BatchNorm backward instance/external API/profiler/tests (#519) · 63af525c
    Qianfeng authored 
    * Refine the device batchnorm-backward base API templates and data type assignments

* Remove duplicated kernel file

* Add batchnorm backward instances and external API

* Add batchnorm-backward profiler and tests

* Add client example which uses batchnorm backward external API

* Merge test/batchnorm_fwd and test/batchnorm_bwd into one directory

* Loose the threshold for batchnorm-backward check_err()
    63af525c
profile_batchnorm_bwd.cpp 6.89 KB