commit 8d07127f5c8f0e189ee0db4feb88c0c0b47608d5
author Giorgio Arena <giorgio.arena@arm.com> Tue Dec 07 13:49:10 2021 +0000
committer Sheri Zhang <sheri.zhang@arm.com> Fri Dec 10 14:48:54 2021 +0000
tree bca9b835e5f171d973e9bf2ca076f86cd71670ec
parent b75d62430e9871fcc6f19cf82879f65d2e7fb201

Use #if directive instead of regular condition in CLDirectConv2D

Resolve COMPMID-5004

Signed-off-by: Giorgio Arena <giorgio.arena@arm.com>
Change-Id: Ib3e1b5a891234316c411ea9825ec10c68c4ab5a3
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6788
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Sheri Zhang <sheri.zhang@arm.com>

src/gpu/cl/kernels/ClDirectConv2dKernel.cpp

diff --git a/src/gpu/cl/kernels/ClDirectConv2dKernel.cpp b/src/gpu/cl/kernels/ClDirectConv2dKernel.cpp
index 7107def..5af7aa9 100644
--- a/src/gpu/cl/kernels/ClDirectConv2dKernel.cpp
+++ b/src/gpu/cl/kernels/ClDirectConv2dKernel.cpp
@@ -453,6 +453,7 @@
         build_options.add_option("-DM0=" + support::cpp11::to_string(m0));
         build_options.add_option("-DK0=" + support::cpp11::to_string(k0));
         build_options.add_option("-DPARTIAL_N0=" + support::cpp11::to_string(partial_store_n0));
+        build_options.add_option_if((src->dimension(channel_idx) % k0) != 0, "-DLEFTOVER_LOOP");
         build_options.add_option("-DACTIVATION_TYPE=" + lower_string(string_from_activation_func(act_info.activation())));
 
         if(is_data_type_quantized(data_type))