aboutsummaryrefslogtreecommitdiff
path: root/scripts
diff options
context:
space:
mode:
Diffstat (limited to 'scripts')
-rw-r--r--scripts/arm_compute_library_nn_driver.go1
1 files changed, 1 insertions, 0 deletions
diff --git a/scripts/arm_compute_library_nn_driver.go b/scripts/arm_compute_library_nn_driver.go
index 2461838d4f..689aa3b8f2 100644
--- a/scripts/arm_compute_library_nn_driver.go
+++ b/scripts/arm_compute_library_nn_driver.go
@@ -25,6 +25,7 @@ func globalFlags(ctx android.BaseContext) []string {
theArch := a.ArchType.String()
if theArch == "armv8-2a" {
cppflags = append(cppflags, "-march=armv8.2-a+fp16")
+ cppflags = append(cppflags, "-DARM_COMPUTE_ENABLE_FP16")
}
}
}