#include <MemCopyWorkload.hpp>

Inheritance diagram for CopyMemGenericWorkload:

Collaboration diagram for CopyMemGenericWorkload:

Public Member Functions
	CopyMemGenericWorkload (const MemCopyQueueDescriptor &descriptor, const WorkloadInfo &info)

void	Execute () const override

void	ExecuteAsync (ExecutionData &executionData) override

Public Member Functions inherited from BaseWorkload< MemCopyQueueDescriptor >
	BaseWorkload (const MemCopyQueueDescriptor &descriptor, const WorkloadInfo &info)

virtual const std::string &	GetName () const override

void	ExecuteAsync (ExecutionData &executionData) override

void	PostAllocationConfigure () override

const MemCopyQueueDescriptor &	GetData () const

arm::pipe::ProfilingGuid	GetGuid () const final

virtual bool	SupportsTensorHandleReplacement () const override

void	ReplaceInputTensorHandle (ITensorHandle *tensorHandle, unsigned int slot) override

void	ReplaceOutputTensorHandle (ITensorHandle *tensorHandle, unsigned int slot) override

Public Member Functions inherited from IWorkload
virtual	~IWorkload ()

virtual arm::pipe::ProfilingGuid	GetGuid () const =0

virtual bool	SupportsTensorHandleReplacement () const =0

virtual const std::string &	GetName () const =0

virtual void	RegisterDebugCallback (const DebugCallbackFunction &)

virtual armnn::Optional< armnn::MemoryRequirements >	GetMemoryRequirements ()

Additional Inherited Members
Protected Attributes inherited from BaseWorkload< MemCopyQueueDescriptor >
MemCopyQueueDescriptor	m_Data

const arm::pipe::ProfilingGuid	m_Guid

const std::string	m_Name

Detailed Description

Definition at line 15 of file MemCopyWorkload.hpp.

Constructor & Destructor Documentation

◆ CopyMemGenericWorkload()

CopyMemGenericWorkload	(	const MemCopyQueueDescriptor &	descriptor,
		const WorkloadInfo &	info
	)

Definition at line 44 of file MemCopyWorkload.cpp.

     : BaseWorkload<MemCopyQueueDescriptor>(descriptor, info)
 {
     GatherTensorHandlePairs(descriptor, m_TensorHandlePairs);
 }

References armnn::GatherTensorHandlePairs(), and armnn::info.

Member Function Documentation

◆ Execute()

void Execute ( ) const

overridevirtual

Implements IWorkload.

Definition at line 51 of file MemCopyWorkload.cpp.

 {
     ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "CopyMemGeneric_Execute");
  
     auto copyFunc = [](void* dst, const void* src, size_t size)
         {
             memcpy(dst, src, size);
         };
  
     for (const auto& pair : m_TensorHandlePairs)
     {
         CopyTensorContentsGeneric(pair.first, pair.second, copyFunc);
     }
 }

References ARMNN_SCOPED_PROFILING_EVENT, armnn::CopyTensorContentsGeneric(), and armnn::Undefined.

◆ ExecuteAsync()

void ExecuteAsync ( ExecutionData & executionData )

overridevirtual

Implements IWorkload.

Definition at line 66 of file MemCopyWorkload.cpp.

 {
     ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "CopyMemGeneric_Execute_WorkingMemDescriptor");
  
     WorkingMemDescriptor* workingMemDescriptor = static_cast<WorkingMemDescriptor*>(executionData.m_Data);
     std::vector<TensorHandlePair> tensorHandlePairs;
  
     GatherTensorHandlePairs(*workingMemDescriptor, tensorHandlePairs);
  
     auto copyFunc = [](void* dst, const void* src, size_t size)
     {
         memcpy(dst, src, size);
     };
  
     for (const auto& pair : tensorHandlePairs)
     {
         CopyTensorContentsGeneric(pair.first, pair.second, copyFunc);
     }
 }

References ARMNN_SCOPED_PROFILING_EVENT, armnn::CopyTensorContentsGeneric(), armnn::GatherTensorHandlePairs(), ExecutionData::m_Data, and armnn::Undefined.

The documentation for this class was generated from the following files:

include/armnn/backends/MemCopyWorkload.hpp
src/backends/backendsCommon/MemCopyWorkload.cpp