aboutsummaryrefslogtreecommitdiff
path: root/SConscript
diff options
context:
space:
mode:
Diffstat (limited to 'SConscript')
-rw-r--r--SConscript5
1 files changed, 2 insertions, 3 deletions
diff --git a/SConscript b/SConscript
index ed22f6eefe..0b7729cced 100644
--- a/SConscript
+++ b/SConscript
@@ -216,11 +216,10 @@ if env['neon']:
graph_files += Glob('src/graph/backends/NEON/*.cpp')
- if env['arch'] == "armv7a":
+ if env['estate'] == '32':
core_files += Glob('src/core/NEON/kernels/arm_gemm/kernels/a32_*/*.cpp')
-
- if "arm64-v8" in env['arch']:
+ if env['estate'] == '64':
core_files += Glob('src/core/NEON/kernels/arm_gemm/kernels/a64_*/*.cpp')
if "sve" in env['arch']:
core_files += Glob('src/core/NEON/kernels/arm_gemm/kernels/sve_*/*.cpp')