Skip to content

GitLab

"example/01_gemm/gemm_xdl_fp16_splitk.cpp" did not exist on "41cdd3801a873def1c1220da9860f5202f36edbd"
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
profile_groupnorm.cpp 3.17 KB