diff options
Diffstat (limited to 'BUILD.bazel')
-rw-r--r-- | BUILD.bazel | 23 |
1 files changed, 20 insertions, 3 deletions
diff --git a/BUILD.bazel b/BUILD.bazel index e3ad75abd8..d246336c29 100644 --- a/BUILD.bazel +++ b/BUILD.bazel @@ -72,6 +72,12 @@ bool_flag( visibility = ["//visibility:public"], ) +bool_flag( + name = "enable_sve_validation", + build_setting_default = False, + visibility = ["//visibility:public"], +) + #--------------------------------------------------------------------- # Flag variables config_setting( @@ -116,6 +122,13 @@ config_setting( }, ) +config_setting( + name = "sve_validation_flag", + flag_values = { + ":enable_sve_validation": "true", + }, +) + #--------------------------------------------------------------------- # Common defines used for all targets @@ -138,8 +151,6 @@ cc_library( "DARM_COMPUTE_GRAPH_ENABLED", "ARM_COMPUTE_ENABLE_SVEF32MM", "ARM_COMPUTE_ENABLE_FIXED_FORMAT_KERNELS", - "ENABLE_SVE", - "ARM_COMPUTE_ENABLE_SVE", "_GLIBCXX_USE_NANOSLEEP" ] + select({ "//:debug_flag": [ @@ -243,8 +254,10 @@ cc_library( "//conditions:default": [], }), local_defines = [ + "ENABLE_SVE", + "ARM_COMPUTE_ENABLE_SVE", "ARM_COMPUTE_ENABLE_SVE2", - "ARM_COMPUTE_ENABLE_BF16" + "ARM_COMPUTE_ENABLE_BF16", ], deps = [ "//:common_defines", @@ -292,6 +305,8 @@ cc_library( "//conditions:default": [], }), local_defines = [ + "ENABLE_SVE", + "ARM_COMPUTE_ENABLE_SVE", "ARM_COMPUTE_ENABLE_BF16", ], deps = [ @@ -348,6 +363,8 @@ cc_library( "//conditions:default": [], }), local_defines = [ + "ENABLE_SVE", + "ARM_COMPUTE_ENABLE_SVE", "ARM_COMPUTE_ENABLE_BF16", ], visibility = ["//visibility:public"], |