diff options
author | Pablo Tello <pablo.tello@arm.com> | 2017-12-15 10:36:21 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:42:33 +0000 |
commit | fef6dae9c2cfe1003ab2abe3a41255e849b1b5eb (patch) | |
tree | 9bff9a89d064d3d4ee5970a4413564481b7a3d63 /src/runtime | |
parent | 340a2722ae6e4dd050c6df669f6c742b5107f12d (diff) | |
download | ComputeLibrary-fef6dae9c2cfe1003ab2abe3a41255e849b1b5eb.tar.gz |
COMPMID-750: Enabled support for U8 and S8 datatypes in NEGEMMLowpAArch64V8P4Kernel
Change-Id: If32cbdc65f2e1441595cae5b4824a9b4357c8bf6
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/113467
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'src/runtime')
-rw-r--r-- | src/runtime/NEON/functions/NEGEMMLowpAssemblyMatrixMultiplyCore.cpp | 6 |
1 files changed, 1 insertions, 5 deletions
diff --git a/src/runtime/NEON/functions/NEGEMMLowpAssemblyMatrixMultiplyCore.cpp b/src/runtime/NEON/functions/NEGEMMLowpAssemblyMatrixMultiplyCore.cpp index 6e03ffa1bc..9b36e81afd 100644 --- a/src/runtime/NEON/functions/NEGEMMLowpAssemblyMatrixMultiplyCore.cpp +++ b/src/runtime/NEON/functions/NEGEMMLowpAssemblyMatrixMultiplyCore.cpp @@ -74,7 +74,7 @@ void NEGEMMLowpAssemblyMatrixMultiplyCore::configure(const ITensor *a, const ITe #endif /* __aarch64__ */ #ifdef ARM_COMPUTE_AARCH64_V8_2 - if(ci.CPU == CPUTarget::A75_DOT) + if(ci.CPU == CPUTarget::A75_DOT || ci.CPU == CPUTarget::A55_DOT) { // Configure matrix multiply kernel GemmInterleaved<gemm_s8_12x8, int8_t, int32_t> gemm(&ci, M, N, K, false, false); @@ -87,10 +87,6 @@ void NEGEMMLowpAssemblyMatrixMultiplyCore::configure(const ITensor *a, const ITe _mm_kernel = std::move(k); _workspace.allocator()->allocate(); } - else if(ci.CPU == CPUTarget::A55_DOT) - { - ARM_COMPUTE_ERROR_ON("WIP"); - } else #elif defined(ARM_COMPUTE_AARCH64_V8A) if(ci.CPU == CPUTarget::A53) |