[Profile] Allow multiple calls to get profile data

[platform/adaptation/npu/trix-engine.git] / src / core / ne-profiler.cc

diff --git a/src/core/ne-profiler.cc b/src/core/ne-profiler.cc
index 603284e..20fcb80 100644 (file)
--- a/src/core/ne-profiler.cc
+++ b/src/core/ne-profiler.cc
@@ -26,6 +26,11 @@ ModelProfiler::appendRequest (int req_id, const Model *model) {
 }
 
 int
+ModelProfiler::removeRequest (int req_id) {
+  return profile_map_.remove (req_id);
+}
+
+int
 ModelProfiler::getProfile (int req_id, const npu_profile_opt &opt, npu_profile *profile) {
   ProfileData *data = profile_map_.find (req_id);
   if (data == nullptr)
@@ -47,7 +52,6 @@ ModelProfiler::getProfile (int req_id, const npu_profile_opt &opt, npu_profile *
       status = -EINVAL;
   }
 
-  profile_map_.remove (req_id);
   return status;
 }