From 90d15b985dbefd730941cbc79997baa0089dc4f3 Mon Sep 17 00:00:00 2001 From: David Svantesson Date: Thu, 8 Jun 2023 10:05:59 +0000 Subject: Bazel and CMake optional fp16 support Resolves ONCPUML-1274 Signed-off-by: David Svantesson Change-Id: I1d189596cfce5be87a18c8065d683700b3c9960f Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9745 Tested-by: Arm Jenkins Reviewed-by: SiCong Li Comments-Addressed: Arm Jenkins Benchmark: Arm Jenkins --- tests/BUILD.bazel | 32 +++++++++++++++++++------------- 1 file changed, 19 insertions(+), 13 deletions(-) (limited to 'tests/BUILD.bazel') diff --git a/tests/BUILD.bazel b/tests/BUILD.bazel index 13d78958e5..5763938d3c 100644 --- a/tests/BUILD.bazel +++ b/tests/BUILD.bazel @@ -34,9 +34,11 @@ cc_library( "validation/reference/*.h", "validation/**/*.h", ]), - copts = [ - "-march=armv8.2-a+fp16", - ] + select({ + copts = [] + select({ + "//:arch_armv8-a": ["-march=armv8-a"], + "//:arch_armv8.2-a+fp16": ["-march=armv8.2-a+fp16"], + "//conditions:default": ["-march=armv8-a"], + }) + select({ "//:debug_flag": [ "-O0", "-g", @@ -54,7 +56,7 @@ cc_library( }), linkstatic = True, deps = [ - "//:arm_compute_core", + "//:arm_compute", "//:common_defines", "//tests/framework", ], @@ -74,9 +76,11 @@ cc_binary( "datasets/*.h", "instruments/*.h", ]), - copts = [ - "-march=armv8.2-a+fp16", - ] + select({ + copts = [] + select({ + "//:arch_armv8-a": ["-march=armv8-a"], + "//:arch_armv8.2-a+fp16": ["-march=armv8.2-a+fp16"], + "//conditions:default": ["-march=armv8-a"], + }) + select({ "//:debug_flag": [ "-O0", "-g", @@ -95,12 +99,12 @@ cc_binary( linkstatic = True, deps = [ ":validation_framework", - "//:arm_compute_core", + "//:arm_compute", "//:arm_compute_graph", "//:common_defines", "//tests/framework", ], - local_defines = [] + + local_defines = [] + select({ "//:bf16_validation_flag": [ "ARM_COMPUTE_ENABLE_BF16", @@ -125,9 +129,11 @@ cc_binary( "benchmark/NEON/*.cpp", "*.cpp", ]), - copts = [ - "-march=armv8.2-a+fp16", - ] + select({ + copts = [] + select({ + "//:arch_armv8-a": ["-march=armv8-a"], + "//:arch_armv8.2-a+fp16": ["-march=armv8.2-a+fp16"], + "//conditions:default": ["-march=armv8-a"], + }) + select({ "//:debug_flag": [ "-O0", "-g", @@ -147,6 +153,6 @@ cc_binary( deps = [ ":arm_compute_validation", ":validation_framework", - "//:arm_compute_core", + "//:arm_compute", ], ) -- cgit v1.2.1