diff options
author | David Svantesson <david.svantesson@arm.com> | 2023-06-08 10:05:59 +0000 |
---|---|---|
committer | ramy.elgammal@arm.com <ramy.elgammal@arm.com> | 2023-06-27 18:31:22 +0100 |
commit | 604a35bdd9cfa8c4221ac146fb29b9166d78d045 (patch) | |
tree | 64e299e25c4dae3c87e34b1ecd363ed112540a6b /tests | |
parent | ba62e9ac2428b8cd84c566d5b37ce8fde14868a2 (diff) | |
download | ComputeLibrary-604a35bdd9cfa8c4221ac146fb29b9166d78d045.tar.gz |
Bazel and CMake optional fp16 support
Resolves ONCPUML-1274
Signed-off-by: David Svantesson <david.svantesson@arm.com>
Change-Id: I1d189596cfce5be87a18c8065d683700b3c9960f
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9745
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests')
-rw-r--r-- | tests/BUILD.bazel | 32 | ||||
-rw-r--r-- | tests/framework/BUILD.bazel | 7 |
2 files changed, 24 insertions, 15 deletions
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", ], ) diff --git a/tests/framework/BUILD.bazel b/tests/framework/BUILD.bazel index 084d12c991..17d5a15a11 100644 --- a/tests/framework/BUILD.bazel +++ b/tests/framework/BUILD.bazel @@ -46,8 +46,11 @@ cc_library( ]), copts = [ "-Wno-overloaded-virtual", - "-march=armv8.2-a+fp16", - ], + ] + select({ + "//:arch_armv8-a": ["-march=armv8-a"], + "//:arch_armv8.2-a+fp16": ["-march=armv8.2-a+fp16"], + "//conditions:default": ["-march=armv8-a"], + }), linkstatic = True, visibility = ["//visibility:public"], deps = [ |