From f52e9fd4b42b75916a5e5a777d4ee9c58f721d14 Mon Sep 17 00:00:00 2001
From: mathad01 <matthew.haddon@arm.com>
Date: Wed, 5 May 2021 12:51:56 +0100
Subject: IVGCVSW-5418 ExecuteNetwork test for MobileBERT

 * Fix bug in stack layer which causes mobileBERT model to seg fault
when running in concurrent mode.

Signed-off-by: mathad01 <matthew.haddon@arm.com>
Change-Id: Ib2b73ffaa5b70147b837b7c764064e530baba637
---
 src/backends/reference/workloads/RefStackWorkload.cpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'src/backends/reference/workloads/RefStackWorkload.cpp')

diff --git a/src/backends/reference/workloads/RefStackWorkload.cpp b/src/backends/reference/workloads/RefStackWorkload.cpp
index 31949e967e..3f7fd7bda2 100644
--- a/src/backends/reference/workloads/RefStackWorkload.cpp
+++ b/src/backends/reference/workloads/RefStackWorkload.cpp
@@ -32,6 +32,9 @@ void RefStackWorkload::Execute(std::vector<ITensorHandle*> inputs, std::vector<I
 {
     ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefStackWorkload_Execute");
 
+    const TensorInfo& inputInfo  = GetTensorInfo(inputs[0]);
+    const TensorInfo& outputInfo = GetTensorInfo(outputs[0]);
+
     std::vector<std::unique_ptr<Decoder<float>>> inputDecoders;
     for (unsigned int i=0; i<inputs.size(); ++i)
     {
@@ -41,7 +44,7 @@ void RefStackWorkload::Execute(std::vector<ITensorHandle*> inputs, std::vector<I
     std::unique_ptr<Encoder<float>> outputEncoder = MakeEncoder<float>(GetTensorInfo(outputs[0]),
                                                                        outputs[0]->Map());
 
-    Stack(m_Data, inputDecoders, *outputEncoder);
+    Stack(m_Data, inputDecoders, *outputEncoder, inputInfo, outputInfo);
 }
 
 } // namespace armnn
-- 
cgit v1.2.1