From df6a3b05842a98702437347ca269138ccd55f852 Mon Sep 17 00:00:00 2001 From: Ramy Elgammal Date: Wed, 30 Nov 2022 16:23:10 +0000 Subject: Use heuristics for setting dynamic fusion direct conv2d tile sizes Resolves: COMPMID-5735 Change-Id: I9958413b69c5052cfa205dd0e9457cc4953aaf35 Signed-off-by: Ramy Elgammal Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/c/VisualCompute/ComputeLibrary/+/474818 Tested-by: bsgcomp Reviewed-by: Gian Marco Iodice Comments-Addressed: bsgcomp Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/8724 Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins Benchmark: Arm Jenkins --- .../sketch/gpu/components/cl/ClComponentDirectConv2d.h | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) (limited to 'src/dynamic_fusion/sketch/gpu/components/cl/ClComponentDirectConv2d.h') diff --git a/src/dynamic_fusion/sketch/gpu/components/cl/ClComponentDirectConv2d.h b/src/dynamic_fusion/sketch/gpu/components/cl/ClComponentDirectConv2d.h index fec22b84a5..c3a70ef3ae 100644 --- a/src/dynamic_fusion/sketch/gpu/components/cl/ClComponentDirectConv2d.h +++ b/src/dynamic_fusion/sketch/gpu/components/cl/ClComponentDirectConv2d.h @@ -25,6 +25,7 @@ #define SRC_DYNAMIC_FUSION_SKETCH_GPU_COMPONENTS_CL_CLCOMPONENTDIRECTCONV2D #include "arm_compute/core/Error.h" +#include "arm_compute/core/KernelDescriptors.h" #include "src/dynamic_fusion/sketch/gpu/components/IGpuKernelComponent.h" #include @@ -56,9 +57,15 @@ public: /** Get fast_relaxed_math flag */ bool fast_relaxed_math() const; + /** Set direct convolution descriptor */ + ClComponentDirectConv2dSettings &direct_conv_descriptor(const DirectConvComputeKernelInfo &desc); + /** Get direct convolution descriptor */ + DirectConvComputeKernelInfo direct_conv_descriptor() const; + private: - bool _export_to_cl_image{ false }; - bool _fast_relaxed_math{ true }; + bool _export_to_cl_image{ false }; + bool _fast_relaxed_math{ true }; + DirectConvComputeKernelInfo _desc{}; // Direct convolution descriptor }; /** Forward declaration */ -- cgit v1.2.1