diff options
author | Giorgio Arena <giorgio.arena@arm.com> | 2021-11-15 17:03:22 +0000 |
---|---|---|
committer | Giorgio Arena <giorgio.arena@arm.com> | 2021-11-17 10:41:30 +0000 |
commit | e330fb41d85d7058f74902ce1d47b2dc00b10a52 (patch) | |
tree | e40a80f1c081721410615ced57f71192fd158459 /src/gpu/cl | |
parent | e3b197410842652f0a78d04fe7b2c333cbeabab6 (diff) | |
download | ComputeLibrary-e330fb41d85d7058f74902ce1d47b2dc00b10a52.tar.gz |
Improve start-up timer for ClIm2Col
Resolve COMPMID-4889
Signed-off-by: Giorgio Arena <giorgio.arena@arm.com>
Change-Id: I4a88082b13865fdaeaba1b7216503cd640aa54df
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6680
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/gpu/cl')
-rw-r--r-- | src/gpu/cl/kernels/ClIm2ColKernel.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
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 |