diff options
Diffstat (limited to 'scripts/arm_compute_library_nn_driver.go')
-rw-r--r-- | scripts/arm_compute_library_nn_driver.go | 9 |
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"), ",") |