2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
6 #include "MemCopyWorkload.hpp"
8 #include "CpuTensorHandle.hpp"
10 #include <TypeUtils.hpp>
12 #include <boost/cast.hpp>
22 template <typename SrcTensorHandleType, typename DstTensorHandleType>
23 void GatherTensorHandlePairs(const MemCopyQueueDescriptor& descriptor,
24 std::vector<std::pair<SrcTensorHandleType*, DstTensorHandleType*>>& tensorHandlePairs)
26 const unsigned int numInputs = static_cast<unsigned int>(descriptor.m_Inputs.size());
27 tensorHandlePairs.reserve(numInputs);
29 for (unsigned int i = 0; i < numInputs; ++i)
31 SrcTensorHandleType* const srcTensorHandle = boost::polymorphic_downcast<SrcTensorHandleType*>(
32 descriptor.m_Inputs[i]);
33 DstTensorHandleType* const dstTensorHandle = boost::polymorphic_downcast<DstTensorHandleType*>(
34 descriptor.m_Outputs[i]);
36 tensorHandlePairs.emplace_back(srcTensorHandle, dstTensorHandle);
43 CopyMemGenericWorkload::CopyMemGenericWorkload(const MemCopyQueueDescriptor& descriptor,
44 const WorkloadInfo& info)
45 : BaseWorkload<MemCopyQueueDescriptor>(descriptor, info)
47 GatherTensorHandlePairs(descriptor, m_TensorHandlePairs);
50 void CopyMemGenericWorkload::Execute() const
52 ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "CopyMemGeneric_Execute");
54 auto copyFunc = [](void* dst, const void* src, size_t size)
56 memcpy(dst, src, size);
59 for (const auto& pair : m_TensorHandlePairs)
61 CopyTensorContentsGeneric(pair.first, pair.second, copyFunc);