[CPU] Add deprecation message to extension API (openvinotoolkit#20970)

siddhant-0707 · Nov 13, 2023 · fef0df6 · fef0df6
1 parent 306137f
commit fef0df6
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/plugins/intel_cpu/src/plugin.cpp b/src/plugins/intel_cpu/src/plugin.cpp
@@ -820,6 +820,12 @@ ov::Any Engine::get_ro_property(const std::string& name, const ov::AnyMap& optio
     return get_metric_legacy(name, options);
 }
 
+OPENVINO_SUPPRESS_DEPRECATED_START
+void Engine::add_extension(const InferenceEngine::IExtensionPtr& extension) {
+    extensionManager->AddExtension(extension);
+}
+OPENVINO_SUPPRESS_DEPRECATED_END
+
 ov::SupportedOpsMap Engine::query_model(const std::shared_ptr<const ov::Model>& model, const ov::AnyMap& config) const {
     WeightsSharing::Ptr fake_w_cache;
 
@@ -839,7 +845,7 @@ ov::SupportedOpsMap Engine::query_model(const std::shared_ptr<const ov::Model>&
     const Config::SnippetsMode snippetsMode = getSnippetsMode(config, conf);
 
     auto context =
-        std::make_shared<GraphContext>(conf, nullptr, fake_w_cache, false);
+        std::make_shared<GraphContext>(conf, extensionManager, fake_w_cache, false);
 
     auto supported = ov::get_supported_nodes(
         model,

diff --git a/src/plugins/intel_cpu/src/plugin.h b/src/plugins/intel_cpu/src/plugin.h
@@ -57,6 +57,10 @@ class Engine : public ov::IPlugin {
                                "get_default_context  is not supported by CPU plugin!");
     };
 
+    OPENVINO_SUPPRESS_DEPRECATED_START
+    void add_extension(const std::shared_ptr<InferenceEngine::IExtension>& extension) override;
+    OPENVINO_SUPPRESS_DEPRECATED_END
+
 private:
     bool is_legacy_api() const;