commit b939cdb6f284df20dde7054e12cb5f790be708a9
author Marat Dukhan <maratek@google.com> Tue Mar 30 18:51:51 2021 -0700
committer XNNPACK Team <xnnpack-github-robot@google.com> Tue Mar 30 18:52:29 2021 -0700
tree 0b4b6bfd5366952da1b23b67914d2df6518b7ad4
parent 96d95e0fb70e3bd95844102b9f3c3cfa0bd24302

Bazel flag --xnn_enable_qs8=true to include QS8 operators in :xnnpack_for_tflite

PiperOrigin-RevId: 365944188

BUILD.bazel

diff --git a/BUILD.bazel b/BUILD.bazel
index d5ec2be..4451d9a 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -4771,13 +4771,16 @@
         "//conditions:default": [],
     }),
     defines = [
-        "XNN_NO_QS8_OPERATORS",
         "XNN_NO_QU8_OPERATORS",
         "XNN_NO_U8_OPERATORS",
         "XNN_NO_X8_OPERATORS",
         "XNN_NO_F16_OPERATORS",
         "XNN_NO_X16_OPERATORS",
-    ],
+    ] + select({
+        ":xnn_enable_qs8_explicit_true": [],
+        ":xnn_enable_qs8_explicit_false": ["XNN_NO_QS8_OPERATORS"],
+        "//conditions:default": ["XNN_NO_QS8_OPERATORS"],
+    }),
     gcc_copts = xnnpack_gcc_std_copts(),
     includes = ["include"],
     msvc_copts = xnnpack_msvc_std_copts(),
@@ -7354,6 +7357,18 @@
     define_values = {"xnnpack_enable_memopt": "false"},
 )
 
+# Enable QS8 inference in TFLite-specific version
+config_setting(
+    name = "xnn_enable_qs8_explicit_true",
+    define_values = {"xnn_enable_qs8": "true"},
+)
+
+# Disable QS8 inference in TFLite-specific version
+config_setting(
+    name = "xnn_enable_qs8_explicit_false",
+    define_values = {"xnn_enable_qs8": "false"},
+)
+
 # Builds with -c dbg
 config_setting(
     name = "debug_build",