1 files changed, 10 insertions, 10 deletions
diff --git a/runtimes/neurun/src/frontend/wrapper/execution.h b/runtimes/neurun/src/frontend/wrapper/execution.h
index 374201eb2..b68a7b967 100644
--- a/runtimes/neurun/src/frontend/wrapper/execution.h
+++ b/runtimes/neurun/src/frontend/wrapper/execution.h
@@ -17,28 +17,28 @@
 #ifndef __EXECUTION_H__
 #define __EXECUTION_H__
 
-#include "codegen/Plan.h"
+#include "compiler/Plan.h"
 #include "exec/Source.h"
 #include "exec/Sink.h"
 
 struct ANeuralNetworksExecution
 {
 public:
-  ANeuralNetworksExecution(const std::shared_ptr<const neurun::codegen::Plan> &plan) : _plan{plan}
+  ANeuralNetworksExecution(const std::shared_ptr<const neurun::compiler::Plan> &plan) : _plan{plan}
   {
     _sources.resize(_plan->model().getInputs().size());
     _sinks.resize(_plan->model().getOutputs().size());
   }
 
 public:
-  const neurun::codegen::Plan &plan(void) const { return *_plan; }
+  const neurun::compiler::Plan &plan(void) const { return *_plan; }
 
 private:
-  std::shared_ptr<const neurun::codegen::Plan> _plan;
+  std::shared_ptr<const neurun::compiler::Plan> _plan;
 
 public:
   // TODO Use InputIndex instead of int
-  void source(int n, std::unique_ptr<neurun::exec::Source> &&source)
+  void source(int n, std::unique_ptr<neurun::exec::ISource> &&source)
   {
     _sources.at(n) = std::move(source);
   }
@@ -48,22 +48,22 @@ public:
   }
 
 public:
-  const neurun::exec::Source &source(int n) const { return *(_sources.at(n)); }
+  const neurun::exec::ISource &source(int n) const { return *(_sources.at(n)); }
 
 public:
   // TODO Use OutputIndex instead of int
-  void sink(int n, std::unique_ptr<neurun::exec::Sink> &&sink) { _sinks.at(n) = std::move(sink); }
+  void sink(int n, std::unique_ptr<neurun::exec::ISink> &&sink) { _sinks.at(n) = std::move(sink); }
   template <typename T, typename... Args> void sink(int n, Args &&... args)
   {
     sink(n, std::unique_ptr<T>{new T{std::forward<Args>(args)...}});
   }
 
 public:
-  const neurun::exec::Sink &sink(int n) const { return *(_sinks.at(n)); }
+  const neurun::exec::ISink &sink(int n) const { return *(_sinks.at(n)); }
 
 private:
-  std::vector<std::unique_ptr<neurun::exec::Source>> _sources;
-  std::vector<std::unique_ptr<neurun::exec::Sink>> _sinks;
+  std::vector<std::unique_ptr<neurun::exec::ISource>> _sources;
+  std::vector<std::unique_ptr<neurun::exec::ISink>> _sinks;
 };
 
 #endif