From 208604155267dc372a9d995499c4aa388fb24726 Mon Sep 17 00:00:00 2001 From: Aleksandr Voron Date: Fri, 8 Dec 2023 06:21:16 +0100 Subject: [PATCH] [CPU] Fix XmlModelWithExtensionFromDSO test (#21507) --- src/plugins/intel_cpu/src/nodes/generic.cpp | 1 + src/plugins/intel_cpu/tests/functional/extension/extension.cpp | 3 ++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/src/plugins/intel_cpu/src/nodes/generic.cpp b/src/plugins/intel_cpu/src/nodes/generic.cpp index 36b8ba8fc60..689d6a1f60a 100644 --- a/src/plugins/intel_cpu/src/nodes/generic.cpp +++ b/src/plugins/intel_cpu/src/nodes/generic.cpp @@ -195,6 +195,7 @@ void Generic::initDescriptor(const NodeConfig &config) { } for (auto &outConf : rightConfig.outConfs) { if (outConf.inPlace() < static_cast(getParentEdges().size()) && + outConf.inPlace() >= 0 && getParentEdgeAt(static_cast(outConf.inPlace()))->getParent()->getChildEdges().size() > 1) { outConf.inPlace(-1); } diff --git a/src/plugins/intel_cpu/tests/functional/extension/extension.cpp b/src/plugins/intel_cpu/tests/functional/extension/extension.cpp index b2f9c1c59ac..1b9c0158e51 100644 --- a/src/plugins/intel_cpu/tests/functional/extension/extension.cpp +++ b/src/plugins/intel_cpu/tests/functional/extension/extension.cpp @@ -205,7 +205,7 @@ static std::string get_extension_path() { } -TEST(Extension, XmlModelWithExtensionFromDSO) { +TEST(Extension, smoke_XmlModelWithExtensionFromDSO) { std::string model = R"V0G0N( @@ -260,6 +260,7 @@ TEST(Extension, XmlModelWithExtensionFromDSO) { std::vector input_values{1, 2, 3, 4, 5, 6, 7, 8}; std::vector expected{12, 13, 14, 15, 16, 17, 18, 19}; InferenceEngine::Core ie; + ie.SetConfig({ { ov::hint::inference_precision.name(), ov::element::f32.get_type_name() } }, "CPU"); ie.AddExtension(std::make_shared(get_extension_path())); InferenceEngine::Blob::CPtr weights; auto network = ie.ReadNetwork(model, weights);