profiler/include/profile_batchnorm_backward_impl.hpp · 63af525c06363f398b851967da2740a2ace382b5 · gaoqiong / composable_kernel

BatchNorm backward instance/external API/profiler/tests (#519) · 63af525c

Qianfeng authored Dec 01, 2022

* 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_backward_impl.hpp 15.2 KB

Replace profile_batchnorm_backward_impl.hpp