Skip to content

GitLab

"composable_kernel/include/utility/utility.hpp" did not exist on "8a4b59785b4f5ba48468d53618ca270c5da599a7"
Switch branch/tag
Find file BlameHistoryPermalink
  • rocking's avatar
    Layernorm and groupnorm support to save mean and inverse std in forward (#929) · 3696fe1c
    rocking authored 
    * save mean and inverse std in normalization

* Save mean and inverse std in splitK

* Vector save mean and inv std

* Modify instance for save mean and std

* simplify the layernorm example

* Save mean and std in groupnorm example

* Save mean and inv std in ckProfiler and test

* Remove compute data type from base class

* Save mean and inv std in client example

* Add changelog

* clang format

* Fix compile error

* Refine naming

* Avoid error in bf16

* revert changelog
    3696fe1c
test_layernorm2d_fp16.cpp 1.89 KB