diff options
| author | Viet-Hoa Do <viet-hoa.do@arm.com> | 2023-12-06 11:01:15 +0000 |
|---|---|---|
| committer | Omar Al Khatib <omar.alkhatib@arm.com> | 2024-04-02 09:25:23 +0000 |
| commit | 77bbe2e08b0376edfd3f504950be7f4b5720eeb0 (patch) | |
| tree | 153e4298e569565f5293017b19a22e3fb5718bbe /src/cpu/kernels/softmax/list.h | |
| parent | 905786ea0c1abb2b8df36c56eae93a97823cace1 (diff) | |
| download | ComputeLibrary-77bbe2e08b0376edfd3f504950be7f4b5720eeb0.tar.gz | |
Add SME2 implementation of softmax for FP32
Signed-off-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Change-Id: I8a63610cfb9ccff89dec6045d023439fc19b027a
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/11357
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/cpu/kernels/softmax/list.h')
| -rw-r--r-- | src/cpu/kernels/softmax/list.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/cpu/kernels/softmax/list.h b/src/cpu/kernels/softmax/list.h index f9295ebbcc..16fbd31a19 100644 --- a/src/cpu/kernels/softmax/list.h +++ b/src/cpu/kernels/softmax/list.h @@ -37,6 +37,13 @@ DECLARE_SOFTMAX_KERNEL(neon_fp16_softmax); DECLARE_SOFTMAX_KERNEL(neon_qasymm8_softmax); DECLARE_SOFTMAX_KERNEL(neon_qasymm8_signed_softmax); +#ifdef ARM_COMPUTE_ENABLE_SME2 + +void sme2_fp32_softmax( + const ITensor *in, void *const tmp, ITensor *out, const float beta, int axis, const Window &window); + +#endif // ARM_COMPUTE_ENABLE_SME2 + #undef DECLARE_SOFTMAX_KERNEL } // namespace cpu } // namespace arm_compute |