From 2cc50b39613cea5e55c8a4851ee08d284a3d4f66 Mon Sep 17 00:00:00 2001 From: Gian Marco Iodice Date: Mon, 30 May 2022 14:41:49 +0100 Subject: Extended direct conv 2d interface for tuning the OpenCl kernel Resolves COMPMID-5298 Change-Id: Ie9b907e5dcf86aa6add8d08799fa7ba7c264edea Signed-off-by: Gian Marco Iodice Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7888 Tested-by: Arm Jenkins Reviewed-by: SiCong Li Benchmark: Arm Jenkins --- arm_compute/core/KernelDescriptors.h | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'arm_compute/core') diff --git a/arm_compute/core/KernelDescriptors.h b/arm_compute/core/KernelDescriptors.h index b1086494e4..c45be9c06f 100644 --- a/arm_compute/core/KernelDescriptors.h +++ b/arm_compute/core/KernelDescriptors.h @@ -109,6 +109,15 @@ struct DWCComputeKernelInfo bool export_weights_to_cl_image{ false }; /**< Export the weights to cl_image */ }; +/** Compute descriptor used by the direct convolution kernel */ +struct DirectConvComputeKernelInfo +{ + int32_t m0{ 1 }; /**< Number of rows to be processed by the kernel */ + int32_t n0{ 1 }; /**< Number of columns to be processed by the kernel */ + int32_t k0{ 1 }; /**< Number of partial accumulations to be processed in a single iteration by the kernel */ + bool export_weights_to_cl_image{ false }; /**< Flag to export the weights to cl_image */ +}; + /** Descriptor used by the softmax kernels */ struct SoftmaxKernelInfo { -- cgit v1.2.1