From e330fb41d85d7058f74902ce1d47b2dc00b10a52 Mon Sep 17 00:00:00 2001 From: Giorgio Arena Date: Mon, 15 Nov 2021 17:03:22 +0000 Subject: Improve start-up timer for ClIm2Col Resolve COMPMID-4889 Signed-off-by: Giorgio Arena Change-Id: I4a88082b13865fdaeaba1b7216503cd640aa54df Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6680 Tested-by: Arm Jenkins Reviewed-by: Gian Marco Iodice Comments-Addressed: Arm Jenkins --- src/gpu/cl/kernels/ClIm2ColKernel.cpp | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'src/gpu/cl/kernels/ClIm2ColKernel.cpp') diff --git a/src/gpu/cl/kernels/ClIm2ColKernel.cpp b/src/gpu/cl/kernels/ClIm2ColKernel.cpp index c42762b99c..6d1271d248 100644 --- a/src/gpu/cl/kernels/ClIm2ColKernel.cpp +++ b/src/gpu/cl/kernels/ClIm2ColKernel.cpp @@ -195,10 +195,16 @@ Im2ColConfiguration configure_opencl_kernel(const ITensorInfo *src, const Size2D if(kernel_dims == Size2D(3U, 3U)) { kernel_name = "im2col3x3_"; + build_opts.add_option("-DIM2COL_3X3"); } else if(kernel_dims == Size2D(9U, 9U)) { kernel_name = "im2col9x9_"; + build_opts.add_option("-DIM2COL_9X9"); + } + else + { + build_opts.add_option("-DIM2COL_GENERIC"); } // Get boundary vector (the first/last vector with potentially a partial vector size) size -- cgit v1.2.1