diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-07-28 13:18:46 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-07-28 17:25:14 +0000 |
commit | 8e2f64f214fa3ce5834db966222fa3b804e236a2 (patch) | |
tree | a5259b9930a6791ce2d7c01403c4376b12f2080a /scripts/arm_compute_library_nn_driver.go | |
parent | a86433a64ab25c2ea8e274bd2f357a9709636f5b (diff) | |
download | ComputeLibrary-8e2f64f214fa3ce5834db966222fa3b804e236a2.tar.gz |
Create custom flags for enabling fp16 support
Resolves: COMPMID-4655
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: I1c1398b39da6da089b288d032074b270d87510c4
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6010
Reviewed-by: Pablo Marquez Tello <pablo.tello@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'scripts/arm_compute_library_nn_driver.go')
-rw-r--r-- | scripts/arm_compute_library_nn_driver.go | 1 |
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") } } } |