commit c7b183ab741650653289f8ce3bdeb4926521fdbd
author Georgios Pinitas <georgios.pinitas@arm.com> Fri Mar 06 18:12:09 2020 +0000
committer Georgios Pinitas <georgios.pinitas@arm.com> Thu Mar 12 12:12:30 2020 +0000
tree 991e9f20340c91c288d52d8f9a64a3729e4a40b0
parent 6800117df3be825f0ec5c6cc71c4377322f51b99

COMPMID-3160: Add Bfloat16 support in NEGEMMConvolutionLayer

Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: I0e449306c138a562ffc1455e76ec44b2fd059d85
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2860
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>

SConstruct

diff --git a/SConstruct b/SConstruct
index 09df193..a7d2c10 100644
--- a/SConstruct
+++ b/SConstruct
@@ -206,7 +206,7 @@
         env.Append(CXXFLAGS = ['-march=armv8-a'])
 
     if 'v8.6-a' in env['arch']:
-        env.Append(CPPDEFINES = ['V8P6', 'ARM_COMPUTE_FORCE_BF16'])
+        env.Append(CPPDEFINES = ['V8P6', 'V8P6_BF', 'ARM_COMPUTE_FORCE_BF16'])
 
 elif 'x86' in env['arch']:
     if env['estate'] == '32':