[Bug] Fix the bug when setting npumgr param

This patch fixes the bug when setting npumgr param.

Signed-off-by: Dongju Chae <dongju.chae@samsung.com>

diff --git a/src/core/ne-scheduler.cc b/src/core/ne-scheduler.cc
index eb666e4..6a37520 100644 (file)
--- a/src/core/ne-scheduler.cc
+++ b/src/core/ne-scheduler.cc
@@ -35,6 +35,12 @@ Request::Request (npu_input_opmode opmode)
   request_id_ = Request::global_request_id_.fetch_add (1);
 }
 
+/** @brief destructor of request class */
+Request::~Request () {
+  if (npumgr_param_ != nullptr)
+    delete npumgr_param_;
+}
+
 /** @brief destructor of scheduler class */
 Scheduler::~Scheduler () {
   /** wait until all callbacks are called */

diff --git a/src/core/ne-scheduler.h b/src/core/ne-scheduler.h
index 896fb53..5911660 100644 (file)
--- a/src/core/ne-scheduler.h
+++ b/src/core/ne-scheduler.h
@@ -32,6 +32,7 @@
 class Request {
  public:
   Request (npu_input_opmode opmode);
+  ~Request ();
 
   void setModel (const Model *model) { model_ = model; }
   const Model *getModel () { return model_; }
@@ -61,6 +62,8 @@ class Request {
   npu_infer_mode getInferMode () { return infer_mode_; }
 
   void setNpumgrParam (const npumgr_param &param) {
+    if (npumgr_param_ == nullptr)
+      npumgr_param_ = new npumgr_param;
     memcpy (npumgr_param_, &param, sizeof (npumgr_param));
   }
   const npumgr_param *getNpumgrParam () { return npumgr_param_; }