From eb82fd2aa786715c3b6a941dc6d6deac4ce8e2a0 Mon Sep 17 00:00:00 2001 From: Pablo Tello Date: Fri, 23 Feb 2018 13:43:50 +0000 Subject: COMPMID-881: RSH new arm_gemm interface. Change-Id: I1e2a1a77097d8017c274af3f97eba6964f80f5fa Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/122592 Tested-by: Jenkins Reviewed-by: Anthony Barbier --- SConscript | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'SConscript') diff --git a/SConscript b/SConscript index dab807a47f..cf2dd878e5 100644 --- a/SConscript +++ b/SConscript @@ -194,6 +194,8 @@ if env['neon']: core_files += Glob('src/core/NEON/*.cpp') core_files += Glob('src/core/NEON/kernels/*.cpp') + core_files += Glob('src/core/NEON/kernels/arm_gemm/*.cpp') + # build winograd sources for either v7a / v8a core_files += Glob('src/core/NEON/kernels/convolution/*/*.cpp') core_files += Glob('src/core/NEON/kernels/convolution/winograd/*/*.cpp') @@ -202,10 +204,11 @@ if env['neon']: graph2_files += Glob('src/graph2/backends/NEON/*.cpp') if env['arch'] == "armv7a": - core_files += Glob('src/core/NEON/kernels/arm32/*.cpp') + core_files += Glob('src/core/NEON/kernels/arm_gemm/kernels/a32_*/*.cpp') + if "arm64-v8" in env['arch']: - core_files += Glob('src/core/NEON/kernels/arm64/*.cpp') + core_files += Glob('src/core/NEON/kernels/arm_gemm/kernels/a64_*/*.cpp') runtime_files += Glob('src/runtime/NEON/*.cpp') runtime_files += Glob('src/runtime/NEON/functions/*.cpp') -- cgit v1.2.1