Move redundant modelBuffer check to metadata.cpp

MirceaDan99 · Dec 17, 2024 · cd884fb · cd884fb
1 parent ee210c9
commit cd884fb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 9 deletions.
diff --git a/src/plugins/intel_npu/src/plugin/src/metadata.cpp b/src/plugins/intel_npu/src/plugin/src/metadata.cpp
@@ -113,7 +113,6 @@ bool Metadata<METADATA_VERSION_1_0>::is_compatible() {
 }
 
 std::unique_ptr<MetadataBase> read_metadata_from(std::istream& stream) {
-    Logger logger("NPUPlugin", Logger::global().level());
     size_t magicBytesSize = MAGIC_BYTES.size();
     std::string blobMagicBytes;
     blobMagicBytes.resize(magicBytesSize);
@@ -151,6 +150,11 @@ std::unique_ptr<MetadataBase> read_metadata_from(std::istream& stream) {
 
 std::unique_ptr<MetadataBase> read_metadata_from(std::istream& stream, const std::shared_ptr<ov::AlignedBuffer>& modelBuffer) {
     Logger logger("NPUPlugin", Logger::global().level());
+
+    if (modelBuffer == nullptr) {
+        return read_metadata_from(stream);
+    }
+
     size_t magicBytesSize = MAGIC_BYTES.size();
     std::string blobMagicBytes;
     blobMagicBytes.resize(magicBytesSize);
@@ -175,7 +179,6 @@ std::unique_ptr<MetadataBase> read_metadata_from(std::istream& stream, const std
     if (storedMeta != nullptr) {
         storedMeta->read(stream);
     } else {
->>>>>>> a76d0bdf91 (POC for supporting blob metadata)
         logger.warning("Imported blob metadata version: %d.%d, but the current version is: %d.%d",
                        get_major(metaVersion),
                        get_minor(metaVersion),

diff --git a/src/plugins/intel_npu/src/plugin/src/plugin.cpp b/src/plugins/intel_npu/src/plugin/src/plugin.cpp
@@ -760,13 +760,7 @@ std::shared_ptr<ov::ICompiledModel> Plugin::import_model(std::istream& stream, c
         auto compiler = compilerAdapterFactory.getCompiler(_backends->getIEngineBackend(), localConfig);
 
         std::unique_ptr<BlobContainer> blobPtr;
-        std::unique_ptr<MetadataBase> storedMeta;
-
-        if (dynamic_cast<ov::OwningSharedStreamBuffer*>(stream.rdbuf())) {
-            storedMeta = read_metadata_from(stream, dynamic_cast<ov::OwningSharedStreamBuffer*>(stream.rdbuf())->get_buffer());
-        } else {
-            storedMeta = read_metadata_from(stream);
-        }
+        auto storedMeta = read_model_from(stream, modelBuffer);
 
         if (storedMeta == nullptr) {
             OPENVINO_THROW("Could not read metadata!");