diff options
author | Giorgio Arena <giorgio.arena@arm.com> | 2021-12-07 13:49:10 +0000 |
---|---|---|
committer | Sheri Zhang <sheri.zhang@arm.com> | 2021-12-10 14:48:54 +0000 |
commit | 8d07127f5c8f0e189ee0db4feb88c0c0b47608d5 (patch) | |
tree | bca9b835e5f171d973e9bf2ca076f86cd71670ec /src/gpu/cl/kernels | |
parent | b75d62430e9871fcc6f19cf82879f65d2e7fb201 (diff) | |
download | ComputeLibrary-8d07127f5c8f0e189ee0db4feb88c0c0b47608d5.tar.gz |
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>
Diffstat (limited to 'src/gpu/cl/kernels')
-rw-r--r-- | src/gpu/cl/kernels/ClDirectConv2dKernel.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/gpu/cl/kernels/ClDirectConv2dKernel.cpp b/src/gpu/cl/kernels/ClDirectConv2dKernel.cpp index 7107def8ff..5af7aa9662 100644 --- a/src/gpu/cl/kernels/ClDirectConv2dKernel.cpp +++ b/src/gpu/cl/kernels/ClDirectConv2dKernel.cpp @@ -453,6 +453,7 @@ void ClDirectConv2dKernel::configure(const CLCompileContext &compile_context, IT 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)) |