diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2021-06-07 14:23:57 +0100 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-06-23 12:25:50 +0000 |
| commit | 20fca524baf99402f742ce38c538f2fd07d5fff9 (patch) | |
| tree | b63d98383d1ba22bb3ca59d393e4ab9d47a9c762 /SConstruct | |
| parent | 1d359279e22874121def2ce4bfdb633d94ea5ade (diff) | |
| download | ComputeLibrary-20fca524baf99402f742ce38c538f2fd07d5fff9.tar.gz | |
Create core library using high priority operators
A smaller core library is created using a subset of the operators.
Changed the structure of filelist.json in order to include more
information about the kernels and make the selection easier.
Resolves: COMPMID-4514
Change-Id: I079ca7d8e64346174eebdd13b834e1dd4dc36ca2
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5786
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'SConstruct')
| -rw-r--r-- | SConstruct | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/SConstruct b/SConstruct index f800d9d105..db6e3e0529 100644 --- a/SConstruct +++ b/SConstruct @@ -211,7 +211,7 @@ if 'v7a' in env['arch']: elif 'v8' in env['arch']: if 'sve2' in env['arch']: env.Append(CXXFLAGS = ['-march=armv8.2-a+sve2+fp16+dotprod']) - env.Append(CPPDEFINES = ['SVE2']) + env.Append(CPPDEFINES = ['ARM_COMPUTE_ENABLE_SVE2']) elif 'sve' in env['arch']: env.Append(CXXFLAGS = ['-march=armv8.2-a+sve+fp16+dotprod']) elif 'armv8r64' in env['arch']: @@ -221,10 +221,10 @@ elif 'v8' in env['arch']: else: env.Append(CXXFLAGS = ['-march=armv8-a']) - if 'v8.6-a' in env['arch']: - env.Append(CPPDEFINES = ['MMLA_INT8', 'V8P6', 'V8P6_BF', 'ARM_COMPUTE_FORCE_BF16']) + if 'v8.6-a' in env['arch'] or env['fat_binary']: + env.Append(CPPDEFINES = ['ARM_COMPUTE_ENABLE_I8MM', 'ARM_COMPUTE_ENABLE_BF16']) if "disable_mmla_fp" not in env['custom_options']: - env.Append(CPPDEFINES = ['MMLA_FP32']) + env.Append(CPPDEFINES = ['ARM_COMPUTE_ENABLE_SVEF32MM']) elif 'x86' in env['arch']: if env['estate'] == '32': env.Append(CCFLAGS = ['-m32']) @@ -257,9 +257,9 @@ if 'x86' not in env['arch']: prefix = "aarch64-tizen-linux-gnu-" if 'sve' in env['arch']: - env.Append(CXXFLAGS = ['-DENABLE_SVE']) + env.Append(CXXFLAGS = ['-DENABLE_SVE', '-DARM_COMPUTE_ENABLE_SVE']) else: - env.Append(CXXFLAGS = ['-DENABLE_NEON']) + env.Append(CXXFLAGS = ['-DENABLE_NEON', '-DARM_COMPUTE_ENABLE_NEON']) if env['build'] == 'native': prefix = "" @@ -308,8 +308,8 @@ if env['fat_binary']: if env['arch'] != 'armv8.2-a': print("Currently fat binary is only supported with armv8.2-a") Exit(1) - env.Append(CXXFLAGS = ['-DENABLE_SVE']) - env.Append(CXXFLAGS = ['-DENABLE_NEON']) + env.Append(CXXFLAGS = ['-DENABLE_SVE', '-DARM_COMPUTE_ENABLE_SVE']) + env.Append(CXXFLAGS = ['-DENABLE_NEON', '-DARM_COMPUTE_ENABLE_NEON']) if env['data_type_support']: if any(i in env['data_type_support'] for i in ['all', 'fp16']): |