ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-17 03:19:48 +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 ed44e71894
commit 3b685d0452
12 changed files with 544 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
profiler/src/profile_normalization.cpp
View File

@@ -13,8 +13,7 @@ using ck::profiler::NormType;
struct ArgParser
{
std::unordered_map<std::string, NormType> norm_dict = {{"layernorm", NormType::LAYERNORM},
{"batchnorm", NormType::BATCHNORM},
std::unordered_map<std::string, NormType> norm_dict = {{"batchnorm", NormType::BATCHNORM},
{"softmax", NormType::SOFTMAX}};
std::unordered_map<std::string, std::vector<int>> long_opts = {