aboutsummaryrefslogtreecommitdiff
path: root/scripts
diff options
context:
space:
mode:
authorMichele Di Giorgio <michele.digiorgio@arm.com>2021-07-23 12:21:46 +0100
committerNikhil Raj Arm <nikhil.raj@arm.com>2021-07-23 15:23:36 +0000
commitf67034e7f52596eece094b37fb386c946e1b87ee (patch)
tree16f40baae5f2e8768997ba3369dd95ed5cac17d2 /scripts
parente7012a657493f96cdfbf9459ffc0766ef1a704ae (diff)
downloadComputeLibrary-f67034e7f52596eece094b37fb386c946e1b87ee.tar.gz
Fix NN driver build script for Android R
Change-Id: If10bfb27b09915bb1f46380a79a2d51bda342dbd Signed-off-by: Michele Di Giorgio <michele.digiorgio@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5982 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Nikhil Raj Arm <nikhil.raj@arm.com>
Diffstat (limited to 'scripts')
-rw-r--r--scripts/arm_compute_library_nn_driver.go9
1 files changed, 6 insertions, 3 deletions
diff --git a/scripts/arm_compute_library_nn_driver.go b/scripts/arm_compute_library_nn_driver.go
index cf9c99e208..2461838d4f 100644
--- a/scripts/arm_compute_library_nn_driver.go
+++ b/scripts/arm_compute_library_nn_driver.go
@@ -21,9 +21,12 @@ func globalFlags(ctx android.BaseContext) []string {
}
if ctx.AConfig().PlatformVersionName() == "R" || ctx.AConfig().PlatformVersionName() == "11" {
- if ctx.AConfig().Targets[Android][0].Arch.ArchType.String() == "armv8-2a" {
- cppflags = append(cppflags, "-march=armv8.2-a+fp16")
- }
+ for _, a := range ctx.DeviceConfig().Arches() {
+ theArch := a.ArchType.String()
+ if theArch == "armv8-2a" {
+ cppflags = append(cppflags, "-march=armv8.2-a+fp16")
+ }
+ }
}
data_types := strings.Split(ctx.AConfig().GetenvWithDefault("COMPUTE_LIB_DATA_TYPE", "ALL"), ",")