COMPMID-3831: Android P Floor VTS and CTS tests failing in CpuAcc backend

Expose data type decoupling for building AndroidNN driver through an
environment variable that allows the user to select the data type.

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

1 files changed, 12 insertions, 0 deletions

diff --git a/scripts/arm_compute_library_nn_driver.go b/scripts/arm_compute_library_nn_driver.go
index 553503f8bf..26cb525bcf 100644
--- a/scripts/arm_compute_library_nn_driver.go
+++ b/scripts/arm_compute_library_nn_driver.go
@@ -8,6 +8,7 @@ package arm_compute_library_nn_driver
 import (
     "android/soong/android"
     "android/soong/cc"
+    "strings"
 )
 
 func globalFlags(ctx android.BaseContext) []string {
@@ -21,6 +22,17 @@ func globalFlags(ctx android.BaseContext) []string {
         cppflags = append(cppflags, "-fno-addrsig")
     }
 
+    data_types := strings.Split(ctx.AConfig().GetenvWithDefault("COMPUTE_LIB_DATA_TYPE", "ALL"), ",")
+
+    for _, x := range data_types {
+        if strings.ToUpper(x) == "ALL" || strings.ToUpper(x) == "FP16" {
+            cppflags = append(cppflags, "-DENABLE_FP16_KERNELS")
+        }
+        if strings.ToUpper(x) == "ALL" || strings.ToUpper(x) == "FP32" {
+            cppflags = append(cppflags, "-DENABLE_FP32_KERNELS")
+        }
+    }
+
     return cppflags
 }