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>
diff --git a/filelist.json b/filelist.json
index 06a0913..f6e8547 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"]
}
}
},