corrected comments

allnes · Oct 17, 2023 · d5974e6 · d5974e6
1 parent 090189d
commit d5974e6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/plugins/intel_cpu/src/nodes/normalize.cpp b/src/plugins/intel_cpu/src/nodes/normalize.cpp
@@ -796,6 +796,12 @@ void NormalizeL2::initSupportedPrimitiveDescriptors() {
             inputPrecision = outputPrecision = Precision::BF16;
     }
 
+    if (one_of(Precision::FP16, inputPrecision, outputPrecision) &&
+        (mayiuse(cpu::x64::avx512_core) || mayiuse(cpu::x64::avx2) || mayiuse(cpu::x64::sse41))) {
+        inputPrecision = outputPrecision = Precision::FP32;
+    }
+
+
     if (!one_of(inputPrecision, Precision::FP32, Precision::BF16, Precision::FP16, Precision::I8, Precision::U8)) {
         THROW_ERROR << "has unsupported input precision: " << inputPrecision;
     }