aboutsummaryrefslogtreecommitdiff
path: root/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_s8s32_dot_6x16.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_s8s32_dot_6x16.hpp')
-rw-r--r--src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_s8s32_dot_6x16.hpp17
1 files changed, 12 insertions, 5 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_s8s32_dot_6x16.hpp b/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_s8s32_dot_6x16.hpp
index 6b3f84064c..759a78a413 100644
--- a/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_s8s32_dot_6x16.hpp
+++ b/src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_s8s32_dot_6x16.hpp
@@ -37,9 +37,9 @@
namespace arm_gemm
{
-
// Actual kernel implementations
void a64_hybrid_s8s32_dot_6x16( ARGLIST );
+void a64_hybrid_s8s32_dot_6x16_a55( ARGLIST );
class cls_a64_hybrid_s8s32_dot_6x16
{
@@ -72,10 +72,11 @@ public:
StdTransformsFixed<operand_type, result_type, 6, 16, 4> transforms = {};
- static PerformanceParameters get_performance_parameters(const CPUInfo *ci) {
+ static PerformanceParameters get_performance_parameters(const CPUInfo *ci)
+ {
switch (ci->get_cpu_model()) {
case CPUModel::A55r1:
- return { 9.5238, 2.0799, 0.2279 };
+ return { 12.667, 2.0799, 0.2279 };
default:
return { 29.6736, 11.4025, 0.5591 };
}
@@ -83,9 +84,15 @@ public:
// Default to the generic kernel
kern_type kernel=a64_hybrid_s8s32_dot_6x16;
-
- cls_a64_hybrid_s8s32_dot_6x16(const CPUInfo *)
+ cls_a64_hybrid_s8s32_dot_6x16(const CPUInfo *ci)
{
+ switch(ci->get_cpu_model()) {
+ default:
+ break;
+ case CPUModel::A55r1:
+ kernel=a64_hybrid_s8s32_dot_6x16_a55;
+ break;
+ }
}
};