diff --git a/compiler/luci-interpreter/src/core/RuntimeGraph.h b/compiler/luci-interpreter/src/core/RuntimeGraph.h index 8184e249d85..076464c3d65 100644 --- a/compiler/luci-interpreter/src/core/RuntimeGraph.h +++ b/compiler/luci-interpreter/src/core/RuntimeGraph.h @@ -54,8 +54,8 @@ class RuntimeGraph void execute() const; private: - IMemoryManager *_memory_manager; RuntimeModule *_owning_module; + IMemoryManager *_memory_manager; std::vector> _tensors; std::vector _input_tensors; std::vector _output_tensors; diff --git a/compiler/luci-interpreter/src/loader/GraphLoader.h b/compiler/luci-interpreter/src/loader/GraphLoader.h index fe066ecf855..836568b88bd 100644 --- a/compiler/luci-interpreter/src/loader/GraphLoader.h +++ b/compiler/luci-interpreter/src/loader/GraphLoader.h @@ -44,10 +44,11 @@ class GraphLoader const loco::Graph *_graph; RuntimeGraph *_runtime_graph; RuntimeToIR &_runtime_to_ir; - IMemoryManager *_memory_manager; const std::unordered_map &_graph_to_runtime_graph; std::unordered_map &_node_to_tensor; + + IMemoryManager *_memory_manager; }; } // namespace luci_interpreter diff --git a/compiler/luci-interpreter/src/loader/ModuleLoader.h b/compiler/luci-interpreter/src/loader/ModuleLoader.h index 11326a2ee8c..0b1c00e4849 100644 --- a/compiler/luci-interpreter/src/loader/ModuleLoader.h +++ b/compiler/luci-interpreter/src/loader/ModuleLoader.h @@ -39,12 +39,12 @@ class ModuleLoader void load(); private: - IMemoryManager *_memory_manager; const luci::Module *_module; RuntimeModule *_runtime_module; RuntimeToIR &_runtime_to_ir; std::unordered_map &_node_to_tensor; std::unordered_map _graph_to_runtime_graph; + IMemoryManager *_memory_manager; }; } // namespace luci_interpreter