From 680333b2db16ef65230e40f15fba3898caaf7114 Mon Sep 17 00:00:00 2001 From: Egor Duplenskii Date: Mon, 14 Aug 2023 17:22:43 +0200 Subject: [PATCH] [CPU][ARM] Enable FP16 precision for CumSum node (#19176) --- src/plugins/intel_cpu/src/nodes/cum_sum.cpp | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/plugins/intel_cpu/src/nodes/cum_sum.cpp b/src/plugins/intel_cpu/src/nodes/cum_sum.cpp index 65d3a55ada0..7e37a9ef26f 100644 --- a/src/plugins/intel_cpu/src/nodes/cum_sum.cpp +++ b/src/plugins/intel_cpu/src/nodes/cum_sum.cpp @@ -12,6 +12,7 @@ #include #include "cum_sum.h" #include "utils/bfloat16.hpp" +#include "openvino/core/type/float16.hpp" using namespace InferenceEngine; @@ -72,7 +73,10 @@ void CumSum::initSupportedPrimitiveDescriptors() { return; dataPrecision = getOriginalInputPrecisionAtPort(CUM_SUM_DATA); - if (!one_of(dataPrecision, Precision::I8, Precision::U8, Precision::I16, Precision::BF16, Precision::I32, Precision::FP32, Precision::I64, Precision::U64)) + if (!one_of(dataPrecision, + Precision::I8, Precision::U8, + Precision::I16, Precision::I32, Precision::I64, Precision::U64, + Precision::BF16, Precision::FP16, Precision::FP32)) IE_THROW() << errorPrefix << " has unsupported 'data' input precision: " << dataPrecision.name(); if (inputShapes.size() == numOfInputs) { @@ -101,6 +105,7 @@ void CumSum::execute(dnnl::stream strm) { OV_CASE(Precision::U8, uint8_t), OV_CASE(Precision::I16, int16_t), OV_CASE(Precision::BF16, bfloat16_t), + OV_CASE(Precision::FP16, ov::float16), OV_CASE(Precision::I32, int32_t), OV_CASE(Precision::FP32, float), OV_CASE(Precision::I64, int64_t),