diff options
| author | Finn Williams <Finn.Williams@arm.com> | 2021-04-07 10:23:21 +0100 |
|---|---|---|
| committer | Finn Williams <Finn.Williams@arm.com> | 2021-04-14 15:18:38 +0100 |
| commit | b8181f72b8c7c9132373dbcf7f8709ec2c0f23c0 (patch) | |
| tree | 04cc91a6efb7e2601f80e4213a747938165b7184 /src/backends/reference/workloads/RefConstantWorkload.cpp | |
| parent | b898222a8856475f0217be5e78b4816aa1914f15 (diff) | |
| download | armnn-b8181f72b8c7c9132373dbcf7f8709ec2c0f23c0.tar.gz | |
IVGCVSW-5787 Add/Update Execute() implementations in RefActivationWorkload
* Added multithreaded StridedSliceEndToEndTest
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I4579db7b5959e0a22256f1bda00238c22e611dec
Diffstat (limited to 'src/backends/reference/workloads/RefConstantWorkload.cpp')
| -rw-r--r-- | src/backends/reference/workloads/RefConstantWorkload.cpp | 19 |
1 files changed, 9 insertions, 10 deletions
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<ConstantQueueDescriptor>(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<void>(0, data), data.m_LayerOutput->GetConstTensor<void>(), - outputInfo.GetNumBytes()); +void RefConstantWorkload::ExecuteAsync(WorkingMemDescriptor &workingMemDescriptor) +{ + Execute(workingMemDescriptor.m_Outputs); } -void RefConstantWorkload::Execute() const +void RefConstantWorkload::Execute(std::vector<ITensorHandle*> outputs) const { + memcpy(outputs[0]->Map(), m_Data.m_LayerOutput->GetConstTensor<void>(), GetTensorInfo(outputs[0]).GetNumBytes()); + ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefConstantWorkload_Execute"); } |