Update roi_pooling.cpp

allnes · Oct 25, 2023 · 2acb311 · 2acb311
1 parent 399a000
commit 2acb311
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/plugins/intel_cpu/src/nodes/roi_pooling.cpp b/src/plugins/intel_cpu/src/nodes/roi_pooling.cpp
@@ -441,6 +441,11 @@ void ROIPooling::initSupportedPrimitiveDescriptors() {
             refParams.src_prc = Precision::FP32;
     }
 
+#if defined(OV_CPU_ARM_ENABLE_FP16)
+    if (refParams.src_prc == Precision::FP16)
+        refParams.src_prc = Precision::FP32;
+#endif
+
     auto format = mayiuse(avx512_core) ? LayoutType::nCsp16c : LayoutType::nCsp8c;
     impl_desc_type impl_type;
     if (mayiuse(cpu::x64::avx512_core)) {
@@ -826,7 +831,6 @@ std::shared_ptr<ROIPooling::ROIPoolingExecutor> ROIPooling::ROIPoolingExecutor::
 
     OV_SWITCH(intel_cpu, ROIPoolingExecutorCreation, ctx, jpp.src_prc,
               OV_CASE(Precision::FP32, float),
-              OV_CASE(Precision::FP16, float16_t),
               OV_CASE(Precision::BF16, bfloat16_t))
 
     return ctx.executor;