From 75a4dd527c181250b1cd2dab260874c153be4678 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=9D=B4=ED=95=9C=EC=A2=85/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Engineer/=EC=82=BC=EC=84=B1=EC=A0=84?= =?utf8?q?=EC=9E=90?= Date: Thu, 6 Sep 2018 10:33:22 +0900 Subject: [PATCH] [neurun] Add IBackendConfig getter to Backend (#2611) Add `IBackendConfig` getter `config()` method to `Backend`. We need to keep it otherwise we do not have any way to call `getOperandLayout()`. Signed-off-by: Hanjoung Lee --- runtimes/neurun/src/backend/BackendManager.cc | 4 +++- runtimes/neurun/src/backend/BackendManager.h | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/runtimes/neurun/src/backend/BackendManager.cc b/runtimes/neurun/src/backend/BackendManager.cc index 46adee5..d5ae3eb 100644 --- a/runtimes/neurun/src/backend/BackendManager.cc +++ b/runtimes/neurun/src/backend/BackendManager.cc @@ -17,11 +17,13 @@ namespace backend Backend::Backend(const std::shared_ptr &backend_config, const std::shared_ptr &initializer_gen, const std::shared_ptr &stage_gen) - : _initializer_gen(initializer_gen), _stage_gen(stage_gen) + : _config(backend_config), _initializer_gen(initializer_gen), _stage_gen(stage_gen) { backend_config->initialize(); } +const std::shared_ptr Backend::config() const { return _config; } + const std::shared_ptr Backend::initializer_gen() const { return _initializer_gen; diff --git a/runtimes/neurun/src/backend/BackendManager.h b/runtimes/neurun/src/backend/BackendManager.h index ca047d8..21c98a8 100644 --- a/runtimes/neurun/src/backend/BackendManager.h +++ b/runtimes/neurun/src/backend/BackendManager.h @@ -23,17 +23,19 @@ public: const std::shared_ptr &initializer_gen, const std::shared_ptr &stage_gen); - Backend(void) : _initializer_gen(nullptr), _stage_gen(nullptr) + Backend(void) : _config(nullptr), _initializer_gen(nullptr), _stage_gen(nullptr) { // DO NOTHING } public: + const std::shared_ptr config() const; const std::shared_ptr initializer_gen() const; const std::shared_ptr stage_gen() const; const std::shared_ptr tensor_builder() const; private: + std::shared_ptr _config; std::shared_ptr _initializer_gen; std::shared_ptr _stage_gen; }; -- 2.7.4