IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CPU] Fixed NormalizeL2 on SSE4.2 for blocked layout (#9071)

Browse Source
This commit is contained in:
Egor Shulman 2021-12-08 11:20:33 +03:00 committed by GitHub
parent e04ca1516d
commit 69b81be14e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference-engine/src/mkldnn_plugin/nodes/mkldnn_normalize_node.cpp
View File

@ -934,7 +934,7 @@ public:
normalize_kernel.reset(
new jit_uni_normalize_kernel_f32<cpu::x64::avx2>(jcp, *kernel_attrs.get()));
} else if (mayiuse(cpu::x64::sse41)) {
blk_size = 4;
blk_size = jcp.is_blk ? 8 : 4;
normalize_modulo_kernel.reset(new jit_uni_normalize_modulo_kernel_f32<cpu::x64::sse41>(jcp));
normalize_kernel.reset(
new jit_uni_normalize_kernel_f32<cpu::x64::sse41>(jcp, *kernel_attrs.get()));