Skip to content

GitLab

"llm/git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "a64570dcae17794adf100a85667180e03b6d7ef2"
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
profiler.cpp 5.4 KB