diff options
Diffstat (limited to 'src/core/NEON/kernels/arm_gemm/gemv_batched.hpp')
-rw-r--r-- | src/core/NEON/kernels/arm_gemm/gemv_batched.hpp | 18 |
1 files changed, 15 insertions, 3 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/gemv_batched.hpp b/src/core/NEON/kernels/arm_gemm/gemv_batched.hpp index 12216009d2..ad504f2664 100644 --- a/src/core/NEON/kernels/arm_gemm/gemv_batched.hpp +++ b/src/core/NEON/kernels/arm_gemm/gemv_batched.hpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2020 Arm Limited. + * Copyright (c) 2017-2021, 2024 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -88,13 +88,25 @@ public: return _subgemm->get_B_pretransposed_array_size(); } - void pretranspose_B_array(void *buffer, const To *B, const int ldb, const int B_multi_stride) override { - _subgemm->pretranspose_B_array(buffer, B, ldb, B_multi_stride); + void pretranspose_B_array(void *buffer, const To *B, const int ldb, const int B_multi_stride, bool transposed) override { + _subgemm->pretranspose_B_array(buffer, B, ldb, B_multi_stride, transposed); } void set_pretransposed_B_data(void *buffer) override { _subgemm->set_pretransposed_B_data(buffer); } + + GemmConfig get_config() override { + GemmConfig c = _subgemm->get_config(); + + std::string new_filter = "gemv_batched["; + new_filter.append(c.filter); + new_filter.append("]"); + + c.filter = new_filter; + + return c; + } }; } // namespace arm_gemm |