Export init and main separately

Author: nikita-kud

src/plugins/intel_npu/src/plugin/include/compiled_model.hpp

@@ -79,6 +79,7 @@ class CompiledModel final : public ICompiledModel {
     std::map<std::string, std::tuple<bool, ov::PropertyMutability, std::function<ov::Any(const Config&)>>> _properties;
     std::vector<ov::PropertyName> _supportedProperties;
 
+    // mutable bool exportInitFlag = false;
     std::shared_ptr<IGraph> _graph;
     std::shared_ptr<IGraph> _initGraph;
     std::shared_ptr<ov::Model> _initModel;

src/plugins/intel_npu/src/plugin/src/compiled_model.cpp

@@ -131,6 +131,14 @@ void CompiledModel::export_model(std::ostream& stream) const {
     size_t blobSizeBeforeVersioning = _graph->export_blob(stream);
     auto meta = Metadata<CURRENT_METADATA_VERSION>(blobSizeBeforeVersioning, ov::get_openvino_version().buildNumber);
     meta.write(stream);
+
+    // DEBUG EXPORT
+    // if (!exportInitFlag) {
+    //     _graph->export_blob(stream);
+    //     exportInitFlag = true;
+    // } else {
+    //     _initGraph->export_blob(stream);
+    // }
 }
 
 std::shared_ptr<const ov::Model> CompiledModel::get_runtime_model() const {

src/plugins/intel_npu/tools/compile_tool/main.cpp

@@ -467,6 +467,20 @@ int main(int argc, char* argv[]) {
             std::cout << "Writing into file - " << outputName << std::endl;
             compiledModel.export_model(outputFile);
         }
+
+        // DEBUG EXPORT
+        // {
+        //     std::string outputInitName = outputName.substr(0, outputName.size()-5) + "_init.blob";
+        //     std::ofstream outputInitFile{outputInitName, std::ios::out | std::ios::binary};
+        //     if (!outputInitFile.is_open()) {
+        //         std::cout << "Outputting file " << outputInitName << " can't be opened for writing" << std::endl;
+        //         return EXIT_FAILURE;
+        //     } else {
+        //         std::cout << "Writing into file - " << outputInitName << std::endl;
+        //         compiledModel.export_model(outputInitFile);
+        //     }
+        // }
+
         std::cout << "Done. LoadNetwork time elapsed: " << loadNetworkTimeElapsed.count() << " ms" << std::endl;
     } catch (const std::exception& error) {
         std::cerr << error.what() << std::endl;