diff options
author | ramelg01 <ramy.elgammal@arm.com> | 2021-12-08 02:00:16 +0000 |
---|---|---|
committer | Sheri Zhang <sheri.zhang@arm.com> | 2021-12-14 10:53:54 +0000 |
commit | b6e49ebb5b83edbf45513978c0c77a8871c6b36c (patch) | |
tree | 411724910a6db7ae8d3451b457796ea6ec4a86a2 /src/core/NEON/kernels/arm_gemm/kernels/a64_interleaved_s8s32_mmla_8x12.hpp | |
parent | 6f3a9f5f4ef6ec7aa8e91df3c1f373d95931dd7b (diff) | |
download | ComputeLibrary-b6e49ebb5b83edbf45513978c0c77a8871c6b36c.tar.gz |
Update A510 arm_gemm cpu Kernels
Resolves: COMPMID-4910
Signed-off-by: Ramy Elgammal <ramy.elgammal@arm.com>
Change-Id: I79b4aa51e07ad1fe81d9218ed8a8f34f0ec5ab06
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6803
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Sheri Zhang <sheri.zhang@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/arm_gemm/kernels/a64_interleaved_s8s32_mmla_8x12.hpp')
-rw-r--r-- | src/core/NEON/kernels/arm_gemm/kernels/a64_interleaved_s8s32_mmla_8x12.hpp | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/kernels/a64_interleaved_s8s32_mmla_8x12.hpp b/src/core/NEON/kernels/arm_gemm/kernels/a64_interleaved_s8s32_mmla_8x12.hpp index ff69bc8f53..4cc3ed040a 100644 --- a/src/core/NEON/kernels/arm_gemm/kernels/a64_interleaved_s8s32_mmla_8x12.hpp +++ b/src/core/NEON/kernels/arm_gemm/kernels/a64_interleaved_s8s32_mmla_8x12.hpp @@ -35,6 +35,7 @@ namespace arm_gemm { // Actual kernel implementations void a64_interleaved_s8s32_mmla_8x12( ARGLIST ); +void a64_interleaved_s8s32_mmla_8x12_a510( ARGLIST ); class cls_a64_interleaved_s8s32_mmla_8x12 { @@ -91,7 +92,7 @@ public: case CPUModel::A510: return { 48.22, 2.49, 0.29 }; case CPUModel::V1: - return { 116.76, 4.67, 0.60 }; + return { 75.54, 8.06, 0.63 }; } } @@ -100,9 +101,17 @@ public: // Default to the generic kernel kern_type kernel=a64_interleaved_s8s32_mmla_8x12; - cls_a64_interleaved_s8s32_mmla_8x12(const CPUInfo *) + cls_a64_interleaved_s8s32_mmla_8x12(const CPUInfo *ci) { + switch(ci->get_cpu_model()) { + default: + break; + case CPUModel::A510: + kernel=a64_interleaved_s8s32_mmla_8x12_a510; + break; + } } + }; } // namespace arm_gemm |