Nodes which not support bf16/f16 on avx2_vnni_2 platforms will fall back to f32

Author: liubo-intel

src/plugins/intel_cpu/src/nodes/topk.cpp

@@ -18,6 +18,7 @@
 #include "onednn/dnnl.h"
 #include "openvino/core/parallel.hpp"
 #include "openvino/op/topk.hpp"
+#include "utils/cpu_utils.hpp"
 #include "utils/ngraph_utils.hpp"
 
 using namespace dnnl;
@@ -1985,11 +1986,14 @@ void TopK::initSupportedPrimitiveDescriptors() {
                                                            ov::element::u8};
 
     ov::element::Type dataPrecision = getOriginalOutputPrecisionAtPort(TOPK_DATA);
-    if (dataPrecision == ov::element::bf16 && !mayiuse(avx512_core)) {
-        THROW_CPU_NODE_ERR("gets incorrect isa for BF16! AVX512 must be supported!");
+    if (dataPrecision == ov::element::bf16 && !hasHardwareSupport(ov::element::bf16)) {
+        THROW_CPU_NODE_ERR("gets incorrect isa for BF16!");
     }
     bool precisionSupported = std::find(std::begin(supportedPrecision), std::end(supportedPrecision), dataPrecision) !=
                               std::end(supportedPrecision);
+    // BF16 is not supported for AVX2_VNNI_2 platforms
+    precisionSupported =
+        (dataPrecision == ov::element::bf16 && mayiuse(cpu::x64::avx2_vnni_2)) ? false : precisionSupported;
     if (!precisionSupported) {
         if (dataPrecision.is_real()) {
             dataPrecision = ov::element::f32;

src/plugins/intel_cpu/tests/functional/custom/single_layer_tests/normalize.cpp

@@ -64,7 +64,13 @@ class NormalizeL2LayerCPUTest : public testing::WithParamInterface<NormalizeL2La
         if (selectedType.empty()) {
             selectedType = getPrimitiveType();
         }
-        selectedType = makeSelectedTypeStr("unknown", inType);
+        // BF16 is not supported for NormalizeL2 on AVX2_VNNI_2 platforms
+        if (ov::with_cpu_x86_avx2_vnni_2() && inType == ElementType::bf16) {
+            selectedType = makeSelectedTypeStr("unknown", ElementType::f32);
+        } else {
+            selectedType = makeSelectedTypeStr("unknown", inType);
+        }
+
         targetDevice = ov::test::utils::DEVICE_CPU;
         init_input_shapes({shapes});
 

src/plugins/intel_cpu/tests/functional/custom/single_layer_tests/roi_pooling.cpp

@@ -194,7 +194,12 @@ class ROIPoolingCPULayerTest : public testing::WithParamInterface<ROIPoolingCPUT
             selectedType = getPrimitiveType();
         }
         selectedType.push_back('_');
-        selectedType += netPrecision.to_string();
+        // BF16 is not supported for ROIPooling on AVX2_VNNI_2 platforms
+        if (ov::with_cpu_x86_avx2_vnni_2() && netPrecision == ElementType::bf16) {
+            selectedType += ov::element::f32.to_string();
+        } else {
+            selectedType += netPrecision.to_string();
+        }
 
         if (netPrecision == ov::element::bf16) {
             rel_threshold = 1e-2;

src/plugins/intel_cpu/tests/functional/custom/single_layer_tests/topk.cpp

@@ -102,7 +102,12 @@ class TopKLayerCPUTest : public testing::WithParamInterface<TopKLayerCPUTestPara
             inPrc = outPrc = netPrecision;
         configuration.insert(additionalConfig.begin(), additionalConfig.end());
 
-        selectedType = getPrimitiveType() + "_" + ov::element::Type(netPrecision).get_type_name();
+        // BF16 is not supported for TopK on AVX2_VNNI_2 platforms
+        if (ov::with_cpu_x86_avx2_vnni_2() && netPrecision == ElementType::bf16) {
+            selectedType = makeSelectedTypeStr(getPrimitiveType(), ElementType::f32);
+        } else {
+            selectedType = makeSelectedTypeStr(getPrimitiveType(), netPrecision);
+        }
 
         staticShape = inputShape.first.rank() == 0;
         if (staticShape) {