From 9c2f47ff8f43998ae95d42cbfa40b34edae08f50 Mon Sep 17 00:00:00 2001
From: MirceaDan99 <mircea-aurelian.dan@intel.com>
Date: Thu, 3 Oct 2024 14:51:27 +0300
Subject: [PATCH] Add header for plugin extension in `intel-npu`

---
 .../intel_npu/src/plugin/include/extension.hpp    | 15 +++++++++++++++
 src/plugins/intel_npu/src/plugin/src/plugin.cpp   |  1 +
 2 files changed, 16 insertions(+)
 create mode 100644 src/plugins/intel_npu/src/plugin/include/extension.hpp

diff --git a/src/plugins/intel_npu/src/plugin/include/extension.hpp b/src/plugins/intel_npu/src/plugin/include/extension.hpp
new file mode 100644
index 00000000000000..7903ff9efe63ca
--- /dev/null
+++ b/src/plugins/intel_npu/src/plugin/include/extension.hpp
@@ -0,0 +1,15 @@
+// Copyright (C) 2018-2024 Intel Corporation
+// SPDX-License-Identifier: Apache-2.0
+//
+
+#pragma once
+
+#include "openvino/core/extension.hpp"
+#include "openvino/core/op_extension.hpp"
+#include "ov_ops/rms.hpp"
+
+#define OP_EXTENSION(NAME) std::make_shared<ov::OpExtension<NAME>>(),
+
+#define NPU_SUPPORTED_EXTENSIONS OP_EXTENSION(ov::op::internal::RMS)
+
+OPENVINO_CREATE_EXTENSIONS(std::vector<ov::Extension::Ptr>({NPU_SUPPORTED_EXTENSIONS}));
\ No newline at end of file
diff --git a/src/plugins/intel_npu/src/plugin/src/plugin.cpp b/src/plugins/intel_npu/src/plugin/src/plugin.cpp
index 260768b3690dd1..0a413502f51360 100644
--- a/src/plugins/intel_npu/src/plugin/src/plugin.cpp
+++ b/src/plugins/intel_npu/src/plugin/src/plugin.cpp
@@ -9,6 +9,7 @@
 #include "compiled_model.hpp"
 #include "compiler.hpp"
 #include "device_helpers.hpp"
+#include "extension.hpp"
 #include "intel_npu/al/config/common.hpp"
 #include "intel_npu/al/config/compiler.hpp"
 #include "intel_npu/al/config/npuw.hpp"