diff options
Diffstat (limited to 'scripts/arm_compute_library_nn_driver.go')
-rw-r--r-- | scripts/arm_compute_library_nn_driver.go | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/scripts/arm_compute_library_nn_driver.go b/scripts/arm_compute_library_nn_driver.go index dda77b55df..2aab2d3fe7 100644 --- a/scripts/arm_compute_library_nn_driver.go +++ b/scripts/arm_compute_library_nn_driver.go @@ -46,6 +46,7 @@ func globalFlags(ctx android.BaseContext) []string { if theArch == "armv8-2a" { cppflags = append(cppflags, "-march=armv8.2-a+fp16") cppflags = append(cppflags, "-DARM_COMPUTE_ENABLE_FP16") + cppflags = append(cppflags, "-DENABLE_FP16_KERNELS") } } } @@ -74,9 +75,6 @@ func globalFlags(ctx android.BaseContext) []string { if strings.ToUpper(x) == "ALL" || strings.ToUpper(x) == "QSYMM16" { cppflags = append(cppflags, "-DENABLE_QSYMM16_KERNELS") } - 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") } |