From b8181f72b8c7c9132373dbcf7f8709ec2c0f23c0 Mon Sep 17 00:00:00 2001 From: Finn Williams Date: Wed, 7 Apr 2021 10:23:21 +0100 Subject: IVGCVSW-5787 Add/Update Execute() implementations in RefActivationWorkload * Added multithreaded StridedSliceEndToEndTest Signed-off-by: Finn Williams Change-Id: I4579db7b5959e0a22256f1bda00238c22e611dec --- .../reference/workloads/RefConstantWorkload.cpp | 19 +++++++++---------- 1 file changed, 9 insertions(+), 10 deletions(-) (limited to 'src/backends/reference/workloads/RefConstantWorkload.cpp') diff --git a/src/backends/reference/workloads/RefConstantWorkload.cpp b/src/backends/reference/workloads/RefConstantWorkload.cpp index d3e65e6615..6290237d69 100644 --- a/src/backends/reference/workloads/RefConstantWorkload.cpp +++ b/src/backends/reference/workloads/RefConstantWorkload.cpp @@ -20,21 +20,20 @@ RefConstantWorkload::RefConstantWorkload( const ConstantQueueDescriptor& descriptor, const WorkloadInfo& info) : BaseWorkload(descriptor, info) {} -void RefConstantWorkload::PostAllocationConfigure() +void RefConstantWorkload::Execute() const { - const ConstantQueueDescriptor& data = this->m_Data; - - ARMNN_ASSERT(data.m_LayerOutput != nullptr); - - const TensorInfo& outputInfo = GetTensorInfo(data.m_Outputs[0]); - ARMNN_ASSERT(data.m_LayerOutput->GetTensorInfo().GetNumBytes() == outputInfo.GetNumBytes()); + Execute(m_Data.m_Outputs); +} - memcpy(GetOutputTensorData(0, data), data.m_LayerOutput->GetConstTensor(), - outputInfo.GetNumBytes()); +void RefConstantWorkload::ExecuteAsync(WorkingMemDescriptor &workingMemDescriptor) +{ + Execute(workingMemDescriptor.m_Outputs); } -void RefConstantWorkload::Execute() const +void RefConstantWorkload::Execute(std::vector outputs) const { + memcpy(outputs[0]->Map(), m_Data.m_LayerOutput->GetConstTensor(), GetTensorInfo(outputs[0]).GetNumBytes()); + ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefConstantWorkload_Execute"); } -- cgit v1.2.1