aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/NEON/kernels/assembly/kernels/a64_sgemm_12x8.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/core/NEON/kernels/assembly/kernels/a64_sgemm_12x8.hpp')
-rw-r--r--arm_compute/core/NEON/kernels/assembly/kernels/a64_sgemm_12x8.hpp9
1 files changed, 9 insertions, 0 deletions
diff --git a/arm_compute/core/NEON/kernels/assembly/kernels/a64_sgemm_12x8.hpp b/arm_compute/core/NEON/kernels/assembly/kernels/a64_sgemm_12x8.hpp
index e229e215ef..603ad8dc0a 100644
--- a/arm_compute/core/NEON/kernels/assembly/kernels/a64_sgemm_12x8.hpp
+++ b/arm_compute/core/NEON/kernels/assembly/kernels/a64_sgemm_12x8.hpp
@@ -28,6 +28,9 @@
// Actual kernel implementations
#include "a64_sgemm_12x8/generic.hpp"
#include "a64_sgemm_12x8/a53.hpp"
+#include "a64_sgemm_12x8/a55.hpp"
+#include "a64_sgemm_12x8/a55r1.hpp"
+
// 12x8 SGEMM "strategy" class.
//
@@ -66,6 +69,12 @@ public:
if (ci->CPU == CPUTarget::A53) {
kernel = a64_sgemm_asimd_12x8_a53;
}
+ else if (ci->CPU == CPUTarget::A55) {
+ kernel = a64_sgemm_asimd_12x8_a55;
+ }
+ else if (ci->CPU == CPUTarget::A55_DOT) {
+ kernel = a64_sgemm_asimd_12x8_a55r1;
+ }
}
};