diff options
Diffstat (limited to 'BUILD.bazel')
-rw-r--r-- | BUILD.bazel | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/BUILD.bazel b/BUILD.bazel index 3a7d941a0e..50340c6c39 100644 --- a/BUILD.bazel +++ b/BUILD.bazel @@ -78,6 +78,12 @@ bool_flag( visibility = ["//visibility:public"], ) +bool_flag( + name = "thread_local_scheduler", + build_setting_default = False, + visibility = ["//visibility:public"], +) + string_flag( name = "arch", build_setting_default = "armv8-a", @@ -152,6 +158,12 @@ config_setting( } ) +config_setting( + name = "thread_local_scheduler_flag", + flag_values = { + ":thread_local_scheduler": "true", + }, +) #--------------------------------------------------------------------- # Common defines used for all targets @@ -196,6 +208,10 @@ cc_library( "//:arch_armv8-a": [], "//:arch_armv8.2-a+fp16": ["ENABLE_FP16_KERNELS", "ARM_COMPUTE_ENABLE_FP16"], "//conditions:default": [], + }) + + select({ + "//:thread_local_scheduler_flag": ["ARM_COMPUTE_THREAD_LOCAL_SCHEDULER"], + "//conditions:default": [], }), visibility = ["//visibility:public"], ) |