diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/backends/backendsCommon/WorkloadUtils.hpp | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/src/backends/backendsCommon/WorkloadUtils.hpp b/src/backends/backendsCommon/WorkloadUtils.hpp index b7e0e060f2..2b07b2b0d2 100644 --- a/src/backends/backendsCommon/WorkloadUtils.hpp +++ b/src/backends/backendsCommon/WorkloadUtils.hpp @@ -86,8 +86,13 @@ void CopyTensorContentsGeneric(const ITensorHandle* srcTensor, ITensorHandle* ds dstChannelStride, dstBatchStride); - auto srcData = static_cast<const uint8_t*>(srcTensor->Map()); - auto dstData = static_cast<uint8_t*>(dstTensor->Map()); + const unsigned char* srcData; + unsigned char* dstData; + { + ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "Synchronize buffers"); + srcData = static_cast<const uint8_t *>(srcTensor->Map()); + dstData = static_cast<uint8_t *>(dstTensor->Map()); + } size_t copyLength = std::min(srcWidth*srcWidthStride, dstWidth*dstWidthStride); size_t copyHeight = std::min(srcHeight, dstHeight); |