1 files changed, 127 insertions, 0 deletions
diff --git a/runtime/neurun/core/src/exec/ExecutorBase.h b/runtime/neurun/core/src/exec/ExecutorBase.h
new file mode 100644
index 000000000..a93e036a5
--- /dev/null
+++ b/runtime/neurun/core/src/exec/ExecutorBase.h
@@ -0,0 +1,127 @@
+/*
+ * Copyright (c) 2019 Samsung Electronics Co., Ltd. All Rights Reserved
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#ifndef __NEURUN_EXEC_EXECUTOR_BASE_H__
+#define __NEURUN_EXEC_EXECUTOR_BASE_H__
+
+#include <mutex>
+
+#include "Source.h"
+#include "exec/ExecutionObservers.h"
+#include "Sink.h"
+#include "exec/IExecutor.h"
+#include "ir/Graph.h"
+#include "ir/LowerInfoMap.h"
+#include "backend/IConfig.h"
+#include "backend/Backend.h"
+#include "compiler/OperandContext.h"
+#include "backend/ExecTime.h"
+#include "exec/IFunction.h"
+#include "backend/ITensorManager.h"
+#include "exec/ExecutionObservee.h"
+#include <list>
+
+namespace neurun
+{
+namespace exec
+{
+
+class ExecutorBase : public IExecutor
+{
+public:
+  ExecutorBase(const ir::Graph &graph,
+               const std::shared_ptr<compiler::OperandContext> &operand_context,
+               std::unique_ptr<backend::TensorManagerSet> tensor_mgrs);
+
+  virtual ~ExecutorBase() = default;
+
+  const ir::Graph &graph() final { return _graph; }
+
+  void execute(const IODescription &desc) final;
+
+  // Used only in Dataflow and Parallel Executors
+  void setIndexedRanks(std::shared_ptr<ir::OperationIndexMap<int64_t>> ranks) final
+  {
+    _indexed_ranks = std::move(ranks);
+  };
+
+  virtual void executeImpl(void) = 0;
+
+  void addObserver(std::unique_ptr<IExecutionObserver> ref) { _subject.add(std::move(ref)); };
+
+private:
+  std::unique_ptr<ISource> source(const ir::IOIndex &index, const ir::TypeInfo &type,
+                                  const void *buffer, size_t length, ir::Layout io_layout);
+  std::unique_ptr<ISink> sink(const ir::IOIndex &index, const ir::TypeInfo &type, void *buffer,
+                              size_t length, ir::Layout io_layout);
+
+  template <typename T>
+  std::unique_ptr<ISource> source(const ir::IOIndex &index, const void *buffer, size_t length,
+                                  ir::Layout io_layout)
+  {
+    const auto operand_index = _graph.getInputs().at(index);
+    const auto &operand = _graph.operands().at(operand_index);
+
+    const auto tensor = _operand_context->at(operand_index);
+    const auto tensor_layout = tensor->layout();
+
+    if (((io_layout == ir::Layout::NHWC) && (tensor_layout == ir::Layout::NCHW)) ||
+        ((io_layout == ir::Layout::NCHW) && (tensor_layout == ir::Layout::NHWC)))
+    {
+      return nnfw::cpp14::make_unique<PermutateSource<T>>(buffer, length, operand.shape(),
+                                                          io_layout);
+    }
+    // TODO Change this to return error
+    assert(io_layout != ir::Layout::UNKNOWN ||
+           (tensor_layout != ir::Layout::NCHW && tensor_layout != ir::Layout::NCHW));
+
+    return nnfw::cpp14::make_unique<CopySource<T>>(buffer, length, operand.shape());
+  }
+
+  template <typename T>
+  std::unique_ptr<ISink> sink(const ir::IOIndex &index, void *buffer, size_t length,
+                              ir::Layout io_layout)
+  {
+    const auto operand_index = _graph.getOutputs().at(index);
+    const auto &operand = _graph.operands().at(operand_index);
+    const auto tensor = _operand_context->at(operand_index);
+    const auto tensor_layout = tensor->layout();
+
+    if (((tensor_layout == ir::Layout::NCHW) && (io_layout == ir::Layout::NHWC)) ||
+        ((tensor_layout == ir::Layout::NHWC) && (io_layout == ir::Layout::NCHW)))
+    {
+      return nnfw::cpp14::make_unique<PermutateSink<T>>(buffer, length, operand.shape(), io_layout);
+    }
+    // TODO Change this to return error
+    assert(io_layout != ir::Layout::UNKNOWN ||
+           (tensor_layout != ir::Layout::NCHW && tensor_layout != ir::Layout::NCHW));
+
+    return nnfw::cpp14::make_unique<CopySink<T>>(buffer, length, operand.shape());
+  }
+
+protected:
+  ExecutionObservee _subject;
+  std::shared_ptr<ir::OperationIndexMap<int64_t>> _indexed_ranks;
+  const ir::Graph &_graph;
+  std::shared_ptr<compiler::OperandContext> _operand_context;
+  std::unique_ptr<backend::TensorManagerSet> _tensor_mgrs;
+  std::mutex _mutex;
+};
+
+} // namespace exec
+} // namespace neurun
+
+#endif // __NEURUN_EXEC_EXECUTOR_BASE_H__