Introduce a config setting to selectively disable the hexagon delegate.

tensorflower-gardener · tensorflower-gardener · commit 60fc34b01fdc · 2024-09-10T14:49:03.000-07:00
Useful for building TfLite runtime tools without dependencies on hexagon(enabled by default). To disable hexagon dependencies, simply add `--define EXPLICIT_DISABLE_HEXAGON=1` or `--//tensorflow/lite/delegates/hexagon:explicit_disable_hexagon=true` to your build command.

PiperOrigin-RevId: 673099424
diff --git a/tensorflow/lite/delegates/hexagon/BUILD b/tensorflow/lite/delegates/hexagon/BUILD
@@ -13,6 +13,11 @@
 # limitations under the License.
 # ==============================================================================
 
+load("@bazel_skylib//lib:selects.bzl", "selects")
+load(
+    "@bazel_skylib//rules:common_settings.bzl",
+    "bool_flag",
+)
 load("//tensorflow/lite:build_def.bzl", "tflite_copts", "tflite_linkopts")
 
 package(
@@ -23,6 +28,35 @@ package(
     licenses = ["notice"],
 )
 
+# If --define EXPLICIT_DISABLE_HEXAGON=1 is passed, the Hexagon delegate will be disabled.
+config_setting(
+    name = "explicit_disable_hexagon_define",
+    define_values = {
+        "EXPLICIT_DISABLE_HEXAGON": "1",
+    },
+)
+
+bool_flag(
+    name = "explicit_disable_hexagon",
+    build_setting_default = False,
+)
+
+config_setting(
+    name = "explicit_disable_hexagon_flag",
+    flag_values = {
+        ":explicit_disable_hexagon": "True",
+    },
+)
+
+# Set if either the explicit_disable_hexagon define or the flag is set.
+selects.config_setting_group(
+    name = "explicit_disable_hexagon_setting",
+    match_any = [
+        ":explicit_disable_hexagon_define",
+        ":explicit_disable_hexagon_flag",
+    ],
+)
+
 cc_library(
     name = "hexagon_implementation",
     srcs = ["hexagon_implementation.cc"],
@@ -77,21 +111,31 @@ cc_library(
 
 cc_library(
     name = "hexagon_delegate",
-    srcs = ["hexagon_delegate.cc"],
-    hdrs = ["hexagon_delegate.h"],
+    srcs = select({
+        ":explicit_disable_hexagon_setting": [],
+        "//conditions:default": ["hexagon_delegate.cc"],
+    }),
+    hdrs = select({
+        ":explicit_disable_hexagon_setting": [],
+        "//conditions:default": ["hexagon_delegate.h"],
+    }),
     tags = [
         "manual",
         "nobuilder",
     ],
-    deps = [
-        ":hexagon_delegate_kernel",
-        ":hexagon_implementation",
-        ":utils",
-        "//tensorflow/lite:kernel_api",
-        "//tensorflow/lite:minimal_logging",
-        "//tensorflow/lite/core/c:common",
-        "//tensorflow/lite/delegates/utils:simple_delegate",
-    ] + select({
+    deps = select({
+        ":explicit_disable_hexagon_setting": [],
+        "//conditions:default": [
+            ":hexagon_delegate_kernel",
+            ":hexagon_implementation",
+            ":utils",
+            "//tensorflow/lite:kernel_api",
+            "//tensorflow/lite:minimal_logging",
+            "//tensorflow/lite/core/c:common",
+            "//tensorflow/lite/delegates/utils:simple_delegate",
+        ],
+    }) + select({
+        ":explicit_disable_hexagon_setting": [],
         "//tensorflow:ios": [],
         "//tensorflow:ios_x86_64": [],
         "//tensorflow:macos": [],
