From f0a4e609d98f111b6a7d4a2b578d1b7cba64b805 Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Thu, 15 Oct 2020 11:54:17 +0100 Subject: 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 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4199 Reviewed-by: Georgios Pinitas Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins --- scripts/arm_compute_library_nn_driver.go | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'scripts/arm_compute_library_nn_driver.go') 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 } -- cgit v1.2.1