diff options
author | Michele Di Giorgio <michele.digiorgio@arm.com> | 2020-10-15 11:54:17 +0100 |
---|---|---|
committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2020-10-19 16:08:58 +0000 |
commit | f0a4e609d98f111b6a7d4a2b578d1b7cba64b805 (patch) | |
tree | 74a319a5c41286c22fc5c477e80a44fd12699a5d /scripts/arm_compute_library_nn_driver.go | |
parent | bc6c374f5bc6c17c3e9b5462f5f8c3c5a5e8a13e (diff) | |
download | ComputeLibrary-f0a4e609d98f111b6a7d4a2b578d1b7cba64b805.tar.gz |
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>
Diffstat (limited to 'scripts/arm_compute_library_nn_driver.go')
-rw-r--r-- | scripts/arm_compute_library_nn_driver.go | 12 |
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 } |