From 2e9a2a0e4653ce0f604983adf70d0a5b489e60ea Mon Sep 17 00:00:00 2001 From: Aleksandr Voron Date: Tue, 19 Nov 2024 16:19:50 +0100 Subject: [PATCH] return empty dequantizationScales check --- .../src/nodes/executors/acl/acl_lowp_fullyconnected.cpp | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/plugins/intel_cpu/src/nodes/executors/acl/acl_lowp_fullyconnected.cpp b/src/plugins/intel_cpu/src/nodes/executors/acl/acl_lowp_fullyconnected.cpp index 0b3cab51de0334..099fee648532cf 100644 --- a/src/plugins/intel_cpu/src/nodes/executors/acl/acl_lowp_fullyconnected.cpp +++ b/src/plugins/intel_cpu/src/nodes/executors/acl/acl_lowp_fullyconnected.cpp @@ -72,7 +72,7 @@ bool ACLLowpFullyConnectedExecutor::supports(const FCConfig &config) { VERIFY(checkPostOps(config.postOps), UNSUPPORTED_TYPE_OF_POSTOPS); VERIFY(one_of(srcRank(config), 2U, 3U, 4U), UNSUPPORTED_SRC_RANK); VERIFY(one_of(weiRank(config), 2U, 3U, 4U), UNSUPPORTED_WEI_RANK); - VERIFY(static_cast(config.attrs).dequantizationScales.size() == 1, UNSUPPORTED_PER_CHANNEL_QUANTIZATION); + VERIFY(static_cast(config.attrs).dequantizationScales.size() <= 1, UNSUPPORTED_PER_CHANNEL_QUANTIZATION); return true; } @@ -82,7 +82,11 @@ void ACLLowpFullyConnectedExecutor::updateTensorsShapes(ACLShapes& aclMemoryShap arm_compute::Status ACLLowpFullyConnectedExecutor::validateTensorsInfo(const ACLInfos & aclMemoryInfos) { auto &tensor_info = aclMemoryInfos[ACLArgs::ACL_SRC_0]; - tensor_info->set_quantization_info(arm_compute::QuantizationInfo(dequantizationScales[0])); + if (dequantizationScales.empty()) { + tensor_info->set_quantization_info(arm_compute::QuantizationInfo(1.f)); + } else { + tensor_info->set_quantization_info(arm_compute::QuantizationInfo(dequantizationScales[0])); + } auto& tensor_info_weights = aclMemoryInfos[ACLArgs::ACL_WEI]; tensor_info_weights->set_quantization_info(arm_compute::QuantizationInfo(1.f));