From 8e2f64f214fa3ce5834db966222fa3b804e236a2 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Wed, 28 Jul 2021 13:18:46 +0100 Subject: Create custom flags for enabling fp16 support Resolves: COMPMID-4655 Signed-off-by: Georgios Pinitas Change-Id: I1c1398b39da6da089b288d032074b270d87510c4 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6010 Reviewed-by: Pablo Marquez Tello Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins --- SConstruct | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) (limited to 'SConstruct') diff --git a/SConstruct b/SConstruct index 8dbb68952c..60d220ee2f 100644 --- a/SConstruct +++ b/SConstruct @@ -222,10 +222,13 @@ elif 'v8' in env['arch']: else: env.Append(CXXFLAGS = ['-march=armv8-a']) - if 'v8.6-a' in env['arch'] or env['fat_binary']: + if 'v8.6-a' in env['arch']: env.Append(CPPDEFINES = ['ARM_COMPUTE_ENABLE_I8MM', 'ARM_COMPUTE_ENABLE_BF16']) if "disable_mmla_fp" not in env['custom_options']: env.Append(CPPDEFINES = ['ARM_COMPUTE_ENABLE_SVEF32MM']) + if 'v8.' in env['arch']: + env.Append(CPPDEFINES = ['ARM_COMPUTE_ENABLE_FP16']) + elif 'x86' in env['arch']: if env['estate'] == '32': env.Append(CCFLAGS = ['-m32']) @@ -309,9 +312,10 @@ 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', '-DARM_COMPUTE_ENABLE_SVE', - '-DARM_COMPUTE_ENABLE_BF16', '-DARM_COMPUTE_ENABLE_I8MM', '-DARM_COMPUTE_ENABLE_SVEF32MM']) - env.Append(CXXFLAGS = ['-DENABLE_NEON', '-DARM_COMPUTE_ENABLE_NEON']) + env.Append(CXXFLAGS = ['-DENABLE_NEON', '-DARM_COMPUTE_ENABLE_NEON', + '-DENABLE_SVE', '-DARM_COMPUTE_ENABLE_SVE', + '-DARM_COMPUTE_ENABLE_FP16', '-DARM_COMPUTE_ENABLE_BF16', + '-DARM_COMPUTE_ENABLE_I8MM', '-DARM_COMPUTE_ENABLE_SVEF32MM']) if env['data_type_support']: if any(i in env['data_type_support'] for i in ['all', 'fp16']): -- cgit v1.2.1