diff options
author | Omar Al Khatib <omar.alkhatib@arm.com> | 2024-04-23 16:26:56 +0100 |
---|---|---|
committer | Omar Al Khatib <omar.alkhatib@arm.com> | 2024-05-08 12:16:10 +0000 |
commit | c1575b2c12a4cee3a60c711fe6521025a814b159 (patch) | |
tree | dd88487711f631a83989dec5c7e058d29cc076ec /src/cpu/kernels/softmax/generic/sme2/fp16.cpp | |
parent | 2fea13593a4753316ae488edf489cb4b00150153 (diff) | |
download | ComputeLibrary-c1575b2c12a4cee3a60c711fe6521025a814b159.tar.gz |
Add SME2 implementation of Softmax for QASYMM8 and QASYMM8_SIGNED.
Resolves: [COMPMID-6917]
Change-Id: Id8b96efd29f6c61dd43a371341c6e1fe087953e9
Signed-off-by: Omar Al Khatib <omar.alkhatib@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/11509
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/cpu/kernels/softmax/generic/sme2/fp16.cpp')
-rw-r--r-- | src/cpu/kernels/softmax/generic/sme2/fp16.cpp | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/src/cpu/kernels/softmax/generic/sme2/fp16.cpp b/src/cpu/kernels/softmax/generic/sme2/fp16.cpp index bcd34d1ca2..e70c9f4793 100644 --- a/src/cpu/kernels/softmax/generic/sme2/fp16.cpp +++ b/src/cpu/kernels/softmax/generic/sme2/fp16.cpp @@ -720,8 +720,15 @@ loop_3_end%=: ); } -void sme2_fp16_softmax(const ITensor *in, void *const, ITensor *out, const float beta, int axis, const Window &window) +void sme2_fp16_softmax(const ITensor *in, + void *const, + ITensor *out, + const float beta, + int axis, + const Window &window, + const float *lut_ptr) { + ARM_COMPUTE_UNUSED(lut_ptr); ARM_COMPUTE_UNUSED(axis); const auto *src_info = in->info(); |