From c1575b2c12a4cee3a60c711fe6521025a814b159 Mon Sep 17 00:00:00 2001 From: Omar Al Khatib Date: Tue, 23 Apr 2024 16:26:56 +0100 Subject: Add SME2 implementation of Softmax for QASYMM8 and QASYMM8_SIGNED. Resolves: [COMPMID-6917] Change-Id: Id8b96efd29f6c61dd43a371341c6e1fe087953e9 Signed-off-by: Omar Al Khatib Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/11509 Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins Reviewed-by: Gunes Bayir Benchmark: Arm Jenkins --- src/BUILD.bazel | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src/BUILD.bazel') diff --git a/src/BUILD.bazel b/src/BUILD.bazel index e3cac07de1..8b8842dfce 100644 --- a/src/BUILD.bazel +++ b/src/BUILD.bazel @@ -119,6 +119,8 @@ filegroup( "cpu/kernels/lut/generic/sve2/u8.cpp", "cpu/kernels/softmax/generic/sme2/fp16.cpp", "cpu/kernels/softmax/generic/sme2/fp32.cpp", + "cpu/kernels/softmax/generic/sme2/qasymm8.cpp", + "cpu/kernels/softmax/generic/sme2/qasymm8_signed.cpp", "cpu/kernels/softmax/generic/sve2/impl.cpp"] + glob(["**/*.h", "**/*.hpp", -- cgit v1.2.1