Skip to content

GitLab

Commit bff80918 authored by Qianfeng Zhang's avatar Qianfeng Zhang
Browse files

Tiny fix in the batchnorm_fwd profiler

parent 3b18f1e3
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
profiler/src/profile_batchnorm_fwd.cpp
View file @ bff80918
...@@ -148,7 +148,7 @@ int profile_batchnorm_forward(int argc, char* argv[]) ...@@ -148,7 +148,7 @@ int profile_batchnorm_forward(int argc, char* argv[])
{ {
if(arg_parser.inLengths.size() == 4 && arg_parser.reduceDims.size() == 3) if(arg_parser.inLengths.size() == 4 && arg_parser.reduceDims.size() == 3)
{ {
profile_batchnorm_forward_impl<F16, F16, F32, F16, F16, F16, 4, 3>( profile_batchnorm_forward_impl<F16, F16, F32, F16, F16, F32, 4, 3>(
arg_parser.do_verification, arg_parser.do_verification,
arg_parser.init_method, arg_parser.init_method,
arg_parser.do_dumpout, arg_parser.do_dumpout,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment