From 8e2f64f214fa3ce5834db966222fa3b804e236a2 Mon Sep 17 00:00:00 2001
From: Georgios Pinitas <georgios.pinitas@arm.com>
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 <georgios.pinitas@arm.com>
Change-Id: I1c1398b39da6da089b288d032074b270d87510c4
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6010
Reviewed-by: Pablo Marquez Tello <pablo.tello@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
---
 scripts/arm_compute_library_nn_driver.go | 1 +
 1 file changed, 1 insertion(+)

(limited to 'scripts')

diff --git a/scripts/arm_compute_library_nn_driver.go b/scripts/arm_compute_library_nn_driver.go
index 2461838d4f..689aa3b8f2 100644
--- a/scripts/arm_compute_library_nn_driver.go
+++ b/scripts/arm_compute_library_nn_driver.go
@@ -25,6 +25,7 @@ func globalFlags(ctx android.BaseContext) []string {
         theArch := a.ArchType.String()
         if theArch == "armv8-2a" {
           cppflags = append(cppflags, "-march=armv8.2-a+fp16")
+          cppflags = append(cppflags, "-DARM_COMPUTE_ENABLE_FP16")
         }
       }
     }
-- 
cgit v1.2.1