Fix driver build error
Resolves COMPMID-6734
Change-Id: I0f0a7a312504b7cca4ed36263843ccd1a190c09e
Signed-off-by: Pablo Marquez Tello <pablo.tello@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10803
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Nikhil Raj Arm <nikhil.raj@arm.com>
Reviewed-by: Jakub Sujak <jakub.sujak@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
diff --git a/scripts/arm_compute_library_nn_driver.go b/scripts/arm_compute_library_nn_driver.go
index dda77b5..2aab2d3 100644
--- a/scripts/arm_compute_library_nn_driver.go
+++ b/scripts/arm_compute_library_nn_driver.go
@@ -46,6 +46,7 @@
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 @@
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")
}