From beabe3bdf47306d0940ddf2ddf52ada6903a0875 Mon Sep 17 00:00:00 2001 From: Moritz Pflanzer Date: Thu, 31 Aug 2017 14:56:32 +0100 Subject: COMPMID-481: Add AArch64 GEMM Change-Id: I34f94f99cb05f0eabafee13c5e623ee779b72360 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/83741 Tested-by: Kaizen Reviewed-by: Anthony Barbier Reviewed-by: Pablo Tello --- src/core/NEON/kernels/NEGEMMInterleave4x4Kernel.cpp | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) (limited to 'src/core/NEON/kernels/NEGEMMInterleave4x4Kernel.cpp') diff --git a/src/core/NEON/kernels/NEGEMMInterleave4x4Kernel.cpp b/src/core/NEON/kernels/NEGEMMInterleave4x4Kernel.cpp index c76c39aa4b..ae5d456141 100644 --- a/src/core/NEON/kernels/NEGEMMInterleave4x4Kernel.cpp +++ b/src/core/NEON/kernels/NEGEMMInterleave4x4Kernel.cpp @@ -85,10 +85,10 @@ void gemm_interleave_16bit_elements(const ITensor *input, ITensor *output, const const uint16x4x4_t data = { { - vld1_u16(reinterpret_cast(in.ptr() + 0 * in_stride)), - vld1_u16(reinterpret_cast(in.ptr() + 1 * in_stride)), - vld1_u16(reinterpret_cast(in.ptr() + 2 * in_stride)), - vld1_u16(reinterpret_cast(in.ptr() + 3 * in_stride)), + vld1_u16(reinterpret_cast(in.ptr() + 0 * in_stride)), + vld1_u16(reinterpret_cast(in.ptr() + 1 * in_stride)), + vld1_u16(reinterpret_cast(in.ptr() + 2 * in_stride)), + vld1_u16(reinterpret_cast(in.ptr() + 3 * in_stride)), } }; vst4_u16(reinterpret_cast(out.ptr()), data); @@ -113,10 +113,10 @@ void gemm_interleave_32bit_elements(const ITensor *input, ITensor *output, const const uint32x4x4_t data = { { - vld1q_u32(reinterpret_cast(in.ptr() + 0 * in_stride)), - vld1q_u32(reinterpret_cast(in.ptr() + 1 * in_stride)), - vld1q_u32(reinterpret_cast(in.ptr() + 2 * in_stride)), - vld1q_u32(reinterpret_cast(in.ptr() + 3 * in_stride)) + vld1q_u32(reinterpret_cast(in.ptr() + 0 * in_stride)), + vld1q_u32(reinterpret_cast(in.ptr() + 1 * in_stride)), + vld1q_u32(reinterpret_cast(in.ptr() + 2 * in_stride)), + vld1q_u32(reinterpret_cast(in.ptr() + 3 * in_stride)) } }; vst4q_u32(reinterpret_cast(out.ptr()), data); -- cgit v1.2.1