diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2020-06-22 13:45:17 +0100 |
|---|---|---|
| committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2020-06-22 15:22:24 +0000 |
| commit | 0028d7c7230d3fda419db5c6d5d8141830bd13f9 (patch) | |
| tree | 4dd017d6ec95cba9e48b2902e7ed70ecbe5c53ee /src | |
| parent | 1732da86669aaa330671077dfa0e60cdc0bcc332 (diff) | |
| download | ComputeLibrary-0028d7c7230d3fda419db5c6d5d8141830bd13f9.tar.gz | |
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 <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3431
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src')
| -rw-r--r-- | src/core/NEON/kernels/NEGEMMLowpOffsetContributionKernel.cpp | 16 |
1 files changed, 4 insertions, 12 deletions
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<Status, Window> validate_and_configure_window(ITensorInfo *mm_result, return std::make_pair(err, win); } -template <bool is_gemm3d> 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<true>(window, _mm_result, _vector_sum_col, _vector_sum_row, _a_offset, _b_offset, _k_offset, _slide_vector_sum_col); - } - else - { - run_offset_contribution<false>(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); +} |