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/neon | |
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/neon')
-rw-r--r-- | src/backends/neon/NeonBackend.cpp | 25 | ||||
-rw-r--r-- | src/backends/neon/NeonBackend.hpp | 2 |
2 files changed, 18 insertions, 9 deletions
diff --git a/src/backends/neon/NeonBackend.cpp b/src/backends/neon/NeonBackend.cpp index 2c3abfd70d..0500ee3411 100644 --- a/src/backends/neon/NeonBackend.cpp +++ b/src/backends/neon/NeonBackend.cpp @@ -22,15 +22,16 @@ #include <armnn/utility/PolymorphicDowncast.hpp> -#include "workloads/NeonAdditionWorkload.hpp" -#include "workloads/NeonBatchNormalizationWorkload.hpp" -#include "workloads/NeonConvolution2dWorkload.hpp" -#include "workloads/NeonDepthwiseConvolutionWorkload.hpp" -#include "workloads/NeonDivisionWorkload.hpp" -#include "workloads/NeonFullyConnectedWorkload.hpp" -#include "workloads/NeonMultiplicationWorkload.hpp" -#include "workloads/NeonReduceWorkload.hpp" -#include "workloads/NeonSubtractionWorkload.hpp" +#include <neon/workloads/NeonAdditionWorkload.hpp> +#include <neon/workloads/NeonBatchNormalizationWorkload.hpp> +#include <neon/workloads/NeonConvolution2dWorkload.hpp> +#include <neon/workloads/NeonDepthwiseConvolutionWorkload.hpp> +#include <neon/workloads/NeonDivisionWorkload.hpp> +#include <neon/workloads/NeonFullyConnectedWorkload.hpp> +#include <neon/workloads/NeonMultiplicationWorkload.hpp> +#include <neon/workloads/NeonReduceWorkload.hpp> +#include <neon/workloads/NeonSubtractionWorkload.hpp> +#include <backendsCommon/DefaultAllocator.hpp> #include <Optimizer.hpp> @@ -427,4 +428,10 @@ void NeonBackend::RegisterTensorHandleFactories(class TensorHandleFactoryRegistr registry.RegisterFactory(std::make_unique<NeonTensorHandleFactory>(memoryManager)); } +std::unique_ptr<ICustomAllocator> NeonBackend::GetDefaultAllocator() const +{ + return std::make_unique<DefaultAllocator>(); +} + + } // namespace armnn diff --git a/src/backends/neon/NeonBackend.hpp b/src/backends/neon/NeonBackend.hpp index 37e1722984..68d60a4c04 100644 --- a/src/backends/neon/NeonBackend.hpp +++ b/src/backends/neon/NeonBackend.hpp @@ -65,6 +65,8 @@ public: { return cpuAccCapabilities; }; + + std::unique_ptr<ICustomAllocator> GetDefaultAllocator() const override; }; } // namespace armnn |