ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-20 04:49:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

ckProfiler for layernorm (#330)

Browse Source 
* Refine parameter

* Add base class for layernorm

* Add layernorm instance

* Add layernorm to ckProfiler

* Remove redundant

* Add verification

* Fix compile error due to merge

[ROCm/composable_kernel commit: fdfd7eb597]
This commit is contained in:
rocking5566
2022-08-12 06:03:54 +08:00
committed by GitHub
parent e4cc81b8ba
commit 01a7af65bb
12 changed files with 544 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
profiler/include/profile_normalization_impl.hpp
View File

@@ -36,7 +36,6 @@ namespace profiler {
enum struct NormType
{
LAYERNORM,
BATCHNORM,
SOFTMAX,
};