From 0028d7c7230d3fda419db5c6d5d8141830bd13f9 Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Mon, 22 Jun 2020 13:45:17 +0100 Subject: COMPMID-3537: Remove template from NEGEMMLowpOffsetContributionKernel This change reduces the core's library size by 14.5Kb. Change-Id: I0f4a1ed4de85423be5cc4a9fb6abc503bbde0767 Signed-off-by: Michalis Spyrou Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3431 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Comments-Addressed: Arm Jenkins --- .../NEON/kernels/NEGEMMLowpOffsetContributionKernel.cpp | 16 ++++------------ 1 file changed, 4 insertions(+), 12 deletions(-) (limited to 'src/core/NEON') diff --git a/src/core/NEON/kernels/NEGEMMLowpOffsetContributionKernel.cpp b/src/core/NEON/kernels/NEGEMMLowpOffsetContributionKernel.cpp index 22939266e5..158b2fdf06 100644 --- a/src/core/NEON/kernels/NEGEMMLowpOffsetContributionKernel.cpp +++ b/src/core/NEON/kernels/NEGEMMLowpOffsetContributionKernel.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2019 ARM Limited. + * Copyright (c) 2017-2020 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -123,10 +123,9 @@ std::pair validate_and_configure_window(ITensorInfo *mm_result, return std::make_pair(err, win); } -template void run_offset_contribution(const Window &window, ITensor *mm_result, const ITensor *vector_sum_col, const ITensor *vector_sum_row, - int32_t a_offset, int32_t b_offset, int32_t k_offset, bool slide_vector_sum_col) + int32_t a_offset, int32_t b_offset, int32_t k_offset, bool slide_vector_sum_col, bool is_gemm3d) { Window collapsed_window = window.collapse_if_possible(window, Window::DimZ); @@ -398,12 +397,5 @@ void NEGEMMLowpOffsetContributionKernel::run(const Window &window, const ThreadI && _mm_result->info()->num_dimensions() > 1 && _mm_result->info()->tensor_shape().y() != _vector_sum_row->info()->tensor_shape().x(); - if(reinterpret_as_3d) - { - run_offset_contribution(window, _mm_result, _vector_sum_col, _vector_sum_row, _a_offset, _b_offset, _k_offset, _slide_vector_sum_col); - } - else - { - run_offset_contribution(window, _mm_result, _vector_sum_col, _vector_sum_row, _a_offset, _b_offset, _k_offset, _slide_vector_sum_col); - } -} \ No newline at end of file + run_offset_contribution(window, _mm_result, _vector_sum_col, _vector_sum_row, _a_offset, _b_offset, _k_offset, _slide_vector_sum_col, reinterpret_as_3d); +} -- cgit v1.2.1