[FIX] Reset callback in advance

This patch resets a blocking request's callback before calling its
callback.

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

diff --git a/src/core/ne-scheduler.cc b/src/core/ne-scheduler.cc
index 2acef27..1eb25ce 100644 (file)
--- a/src/core/ne-scheduler.cc
+++ b/src/core/ne-scheduler.cc
@@ -157,6 +157,10 @@ Scheduler::handleCallback (Request *req) {
   outputCallback callback = req->getCallback ();
 
   req->setSubmitted (false);
+  if (req->getInferMode () == NPU_INFER_BLOCKING) {
+    /* remove the internal callback for next submissions */
+    req->setCallback (nullptr);
+  }
 
   if (!req->isStopped () && callback != nullptr)
     callback ();
@@ -164,9 +168,6 @@ Scheduler::handleCallback (Request *req) {
   if (!req->isPreserved ()) {
     /* the request instance is also deleted here */
     removeRequest (req);
-  } else if (req->getInferMode () == NPU_INFER_BLOCKING) {
-    /* remove the internal callback for next submissions */
-    req->setCallback (nullptr);
   }
 }