[neurun] Recover prepare of IFunction (#6975)

Recover prepare of IFunction for #6156

Signed-off-by: Yongseop Kim <yons.kim@samsung.com>

diff --git a/runtimes/neurun/backend/acl_common/AclFunction.h b/runtimes/neurun/backend/acl_common/AclFunction.h
index 24235a3..a63f380 100644 (file)
--- a/runtimes/neurun/backend/acl_common/AclFunction.h
+++ b/runtimes/neurun/backend/acl_common/AclFunction.h
@@ -47,6 +47,7 @@ public:
     run();
     arm_compute::CLScheduler::get().sync();
   }
+  void prepare() override { _func->prepare(); }
 
 private:
   std::unique_ptr<::arm_compute::IFunction> _func;

diff --git a/runtimes/neurun/core/include/exec/IFunction.h b/runtimes/neurun/core/include/exec/IFunction.h
index 350fc56..5cc29ea 100644 (file)
--- a/runtimes/neurun/core/include/exec/IFunction.h
+++ b/runtimes/neurun/core/include/exec/IFunction.h
@@ -28,6 +28,7 @@ public:
   virtual ~IFunction() = default;
   virtual void run() = 0;
   virtual void runSync() = 0;
+  virtual void prepare() {}
 };
 
 } // namespace exec

diff --git a/runtimes/neurun/core/src/exec/FunctionSequence.cc b/runtimes/neurun/core/src/exec/FunctionSequence.cc
index 1cc8483..d3b6f1f 100644 (file)
--- a/runtimes/neurun/core/src/exec/FunctionSequence.cc
+++ b/runtimes/neurun/core/src/exec/FunctionSequence.cc
@@ -37,6 +37,14 @@ void FunctionSequence::runSync()
   }
 }
 
+void FunctionSequence::prepare()
+{
+  for (const auto &function : _functions)
+  {
+    function->prepare();
+  }
+}
+
 void FunctionSequence::append(std::unique_ptr<IFunction> &&function)
 {
   _functions.push_back(std::move(function));

diff --git a/runtimes/neurun/core/src/exec/FunctionSequence.h b/runtimes/neurun/core/src/exec/FunctionSequence.h
index ab0bf4a..b61d3c0 100644 (file)
--- a/runtimes/neurun/core/src/exec/FunctionSequence.h
+++ b/runtimes/neurun/core/src/exec/FunctionSequence.h
@@ -34,6 +34,7 @@ public:
 
   void run() override;
   void runSync() override;
+  void prepare() override;
 
   /**
    * @brief Appends an IFunction object to the function sequence