From 77bbe2e08b0376edfd3f504950be7f4b5720eeb0 Mon Sep 17 00:00:00 2001 From: Viet-Hoa Do Date: Wed, 6 Dec 2023 11:01:15 +0000 Subject: Add SME2 implementation of softmax for FP32 Signed-off-by: Viet-Hoa Do Change-Id: I8a63610cfb9ccff89dec6045d023439fc19b027a Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/11357 Tested-by: Arm Jenkins Reviewed-by: Gunes Bayir Comments-Addressed: Arm Jenkins Benchmark: Arm Jenkins --- filelist.json | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'filelist.json') diff --git a/filelist.json b/filelist.json index 06a09139db..f6e85473c2 100644 --- a/filelist.json +++ b/filelist.json @@ -2237,7 +2237,8 @@ "common": [ "src/cpu/kernels/softmax/generic/sve/impl.cpp" ] }, "sve2":{ - "common" :["src/cpu/kernels/softmax/generic/sve2/impl.cpp"] + "common" :["src/cpu/kernels/softmax/generic/sve2/impl.cpp"], + "fp32" :["src/cpu/kernels/softmax/generic/sme2/fp32.cpp"] } } }, -- cgit v1.2.1