[neurun] Apply InternalBufferManager to RNN on acl_cl (#6978)
author김용섭/On-Device Lab(SR)/Engineer/삼성전자 <yons.kim@samsung.com>
Thu, 29 Aug 2019 05:35:18 +0000 (14:35 +0900)
committer오형석/On-Device Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Thu, 29 Aug 2019 05:35:18 +0000 (14:35 +0900)
Apply InternalBufferManager to RNN on acl_cl

Signed-off-by: Yongseop Kim <yons.kim@samsung.com>
runtimes/neurun/backend/acl_cl/KernelGenerator.cc

index 923f503..27a19ae 100644 (file)
@@ -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);