From 9b89e9ad04da0a8218e8740f6fe935e92beec780 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EA=B9=80=EC=9A=A9=EC=84=AD/On-Device=20Lab=28SR=29/Enginee?= =?utf8?q?r/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Thu, 29 Aug 2019 14:35:18 +0900 Subject: [PATCH] [neurun] Apply InternalBufferManager to RNN on acl_cl (#6978) Apply InternalBufferManager to RNN on acl_cl Signed-off-by: Yongseop Kim --- runtimes/neurun/backend/acl_cl/KernelGenerator.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/runtimes/neurun/backend/acl_cl/KernelGenerator.cc b/runtimes/neurun/backend/acl_cl/KernelGenerator.cc index 923f503..27a19ae 100644 --- a/runtimes/neurun/backend/acl_cl/KernelGenerator.cc +++ b/runtimes/neurun/backend/acl_cl/KernelGenerator.cc @@ -1312,7 +1312,8 @@ void KernelGenerator::visit(const model::operation::RNNNode &node) _execution_builder->append(asAclFunction(std::move(copy_layer))); std::unique_ptr<::arm_compute::IFunction> fn; - auto rnn_layer = nnfw::cpp14::make_unique<::arm_compute::CLRNNLayerEx>(); + auto rnn_layer = nnfw::cpp14::make_unique<::arm_compute::CLRNNLayerEx>( + _tensor_builder->acl_memory_manager()->internal_buffer_manager()); rnn_layer->configure(input_alloc->handle(), weights_alloc->handle(), recurrent_weights_alloc->handle(), bias_alloc->handle(), hidden_state_out_alloc->handle(), output_alloc->handle(), act_info); -- 2.7.4