diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2021-10-14 17:30:24 +0100 |
---|---|---|
committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2021-10-19 22:26:45 +0100 |
commit | e8d7ccb1a18b5c39ecf17fb063fcc50baacc74b3 (patch) | |
tree | d1a072111807145a786b4c4221707bd34c1373be /src/backends/cl/test/ClCustomAllocatorTests.cpp | |
parent | fc7c2f3c1dd990620a01ecf1a6641813e6f46ef8 (diff) | |
download | armnn-e8d7ccb1a18b5c39ecf17fb063fcc50baacc74b3.tar.gz |
IVGCVSW-6310 Update ICustomAllocator and add GetDefaultAllocator function to backends
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Change-Id: If2879e8a82692285e7fa3f4d09abd608ca28de12
Diffstat (limited to 'src/backends/cl/test/ClCustomAllocatorTests.cpp')
-rw-r--r-- | src/backends/cl/test/ClCustomAllocatorTests.cpp | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/src/backends/cl/test/ClCustomAllocatorTests.cpp b/src/backends/cl/test/ClCustomAllocatorTests.cpp index e614f4c624..60145139ff 100644 --- a/src/backends/cl/test/ClCustomAllocatorTests.cpp +++ b/src/backends/cl/test/ClCustomAllocatorTests.cpp @@ -10,13 +10,13 @@ #include <armnn/IRuntime.hpp> #include <armnn/Utils.hpp> #include <armnn/BackendRegistry.hpp> + #include <cl/ClBackend.hpp> #if defined(ARMCOMPUTENEON_ENABLED) #include <neon/NeonBackend.hpp> #endif - #include <doctest/doctest.h> - +#include <armnn/utility/IgnoreUnused.hpp> // Contains the OpenCl interfaces for mapping memory in the Gpu Page Tables // Requires the OpenCl backend to be included (GpuAcc) #include <arm_compute/core/CL/CLKernelLibrary.h> @@ -31,7 +31,7 @@ class SampleClBackendCustomAllocator : public armnn::ICustomAllocator public: SampleClBackendCustomAllocator() = default; - void* allocate(size_t size, size_t alignment) + void* allocate(size_t size, size_t alignment) override { // If alignment is 0 just use the CL_DEVICE_GLOBAL_MEM_CACHELINE_SIZE for alignment if (alignment == 0) @@ -49,12 +49,12 @@ public: } /** Interface to be implemented by the child class to free the allocated tensor */ - void free(void* ptr) + void free(void* ptr) override { std::free(ptr); } - armnn::MemorySource GetMemorySourceType() + armnn::MemorySource GetMemorySourceType() override { return armnn::MemorySource::Malloc; } |