diff --git a/scripts/arm_compute_library_nn_driver.go b/scripts/arm_compute_library_nn_driver.go
index 2461838..689aa3b 100644
--- a/scripts/arm_compute_library_nn_driver.go
+++ b/scripts/arm_compute_library_nn_driver.go
@@ -25,6 +25,7 @@
         theArch := a.ArchType.String()
         if theArch == "armv8-2a" {
           cppflags = append(cppflags, "-march=armv8.2-a+fp16")
+          cppflags = append(cppflags, "-DARM_COMPUTE_ENABLE_FP16")
         }
       }
     }