[newrt] Use ACL backend for all operations (#1874)
author이한종/동작제어Lab(SR)/Engineer/삼성전자 <hanjoung.lee@samsung.com>
Thu, 5 Jul 2018 05:49:21 +0000 (14:49 +0900)
committer이춘석/동작제어Lab(SR)/Staff Engineer/삼성전자 <chunseok.lee@samsung.com>
Thu, 5 Jul 2018 05:49:21 +0000 (14:49 +0900)
Currently we cannot run with heterogeneous backends, set all backends as
ACL. Later, we are going to control it with environment variables.

Signed-off-by: Hanjoung Lee <hanjoung.lee@samsung.com>
runtimes/new_runtime/src/compilation.cc

index feaa028..8f721de 100644 (file)
@@ -77,10 +77,10 @@ public:
   BackendResolver(::internal::BackendManager &backend_manager)
   {
     auto acl_gen = backend_manager.get("arm_compute");
-    auto cpu_gen = backend_manager.get("cpu");
+    // auto cpu_gen = backend_manager.get("cpu"); // Unused for now
 
     // TODO Set generator map according to environment variable
-    _gen_map[typeid(::internal::tflite::op::Conv2D::implicit::Node)] = cpu_gen;
+    _gen_map[typeid(::internal::tflite::op::Conv2D::implicit::Node)] = acl_gen;
     _gen_map[typeid(::internal::tflite::op::MaxPool2D::implicit::Node)] = acl_gen;
     _gen_map[typeid(::internal::tflite::op::AvgPool2D::implicit::Node)] = acl_gen;
     _gen_map[typeid(::internal::tflite::op::Concat::Node)] = acl_gen;