ClBackend::ClBackendCustomAllocatorWrapper Class Reference

#include <ClBackend.hpp>

Inheritance diagram for ClBackend::ClBackendCustomAllocatorWrapper:

Public Member Functions
	ClBackendCustomAllocatorWrapper (std::shared_ptr< ICustomAllocator > alloc)
void *	allocate (size_t size, size_t alignment) override
void	free (void *ptr) override
std::unique_ptr< arm_compute::IMemoryRegion >	make_region (size_t size, size_t alignment) override

Detailed Description

Definition at line 105 of file ClBackend.hpp.

Constructor & Destructor Documentation

ClBackendCustomAllocatorWrapper()

ClBackendCustomAllocatorWrapper ( std::shared_ptr< ICustomAllocator >  alloc )

inline

Definition at line 108 of file ClBackend.hpp.

                                                                               : m_CustomAllocator(alloc)
        {}

Member Function Documentation

allocate()

void* allocate ( size_t  size, size_t  alignment  )

inlineoverride

Definition at line 111 of file ClBackend.hpp.

        {
            auto alloc = m_CustomAllocator->allocate(size, alignment);
            return MapAllocatedMemory(alloc, size, m_CustomAllocator->GetMemorySourceType());
        }

free()

void free ( void *  ptr )

inlineoverride

Definition at line 116 of file ClBackend.hpp.

        {
            auto hostMemPtr = m_AllocatedBufferMappings[ptr];
            clReleaseMemObject(static_cast<cl_mem>(ptr));
            m_CustomAllocator->free(hostMemPtr);
        }

make_region()

std::unique_ptr<arm_compute::IMemoryRegion> make_region ( size_t  size, size_t  alignment  )

inlineoverride

Definition at line 122 of file ClBackend.hpp.

References armnn::DmaBuf, armnn::DmaBufProtected, armnn::error, and armnn::Malloc.

        {
            auto hostMemPtr = m_CustomAllocator->allocate(size, alignment);
            cl_mem buffer = MapAllocatedMemory(hostMemPtr, size, m_CustomAllocator->GetMemorySourceType());

            return std::make_unique<ClBackendCustomAllocatorMemoryRegion>(cl::Buffer(buffer),
                                                                          hostMemPtr,
                                                                          m_CustomAllocator->GetMemorySourceType());
        }

---

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

• src/backends/cl/ClBackend.hpp