From e2af6f4322a1e2b8b3c391fb721a6a80c281477f Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Fri, 28 Jan 2022 17:59:18 +0000 Subject: IVGCVSW-6552 Add support of aligned host memory * Add AllocatedData functions to OutputHandler * Enable import aligned memory in ImportInputs * Enable import aligned memory in ImportOutputs * Allow to import input and output if the memory is aligned * Implement Reconfigure function on ClConvolution2dWorkload * End-to-end test on Ref and Cl to ensure that input and output memory are imported when aligned Signed-off-by: Narumol Prangnawarat Change-Id: I9e5e4c26d1ac2f1d806803ade5f64c6479c51718 --- src/backends/cl/workloads/ClConvolution2dWorkload.hpp | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'src/backends/cl/workloads/ClConvolution2dWorkload.hpp') diff --git a/src/backends/cl/workloads/ClConvolution2dWorkload.hpp b/src/backends/cl/workloads/ClConvolution2dWorkload.hpp index 8a4599df47..891d5096cd 100644 --- a/src/backends/cl/workloads/ClConvolution2dWorkload.hpp +++ b/src/backends/cl/workloads/ClConvolution2dWorkload.hpp @@ -13,6 +13,8 @@ #include #include +#include + #include namespace armnn @@ -38,6 +40,9 @@ public: arm_compute::ConvolutionMethod GetConvolutionMethod() const; +protected: + void Reconfigure() override; + private: mutable arm_compute::CLConvolutionLayer m_ConvolutionLayer; @@ -47,6 +52,9 @@ private: arm_compute::ConvolutionMethod m_ConvolutionMethod; void FreeUnusedTensors(); + + std::unique_ptr m_InputProxy; + std::unique_ptr m_OutputProxy; }; } //namespace armnn -- cgit v1.2.1