commit 25764d89c87ea1335d7599d4dd50f4b9b5a379a4
author Zhi An Ng <zhin@google.com> Fri Jan 07 11:27:36 2022 -0800
committer XNNPACK Team <xnnpack-github-robot@google.com> Fri Jan 07 11:28:34 2022 -0800
tree 72e2e79d68246b1fa999034393d7cac31d663621
parent 0e0f7261e29511bb58c1116124307ef7c73adee2

Add JIT microkernels to bench/f32-gemm

PiperOrigin-RevId: 420333896

BUILD.bazel

diff --git a/BUILD.bazel b/BUILD.bazel
index eecf0d2..ba03874 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -24,6 +24,7 @@
     ":bench_microkernels",
     ":bench_utils",
     ":enable_assembly",
+    ":enable_jit",
     "@cpuinfo",
     "@FP16",
     "@pthreadpool",
@@ -8799,6 +8800,15 @@
     }),
 )
 
+cc_library(
+    name = "enable_jit",
+    defines = select({
+        ":xnn_enable_jit_explicit_true": ["XNN_ENABLE_JIT=1"],
+        ":xnn_enable_jit_explicit_false": ["XNN_ENABLE_JIT=0"],
+        "//conditions:default": ["XNN_ENABLE_JIT=0"],
+    }),
+)
+
 xnnpack_cc_library(
     name = "operators",
     srcs = OPERATOR_SRCS + [
@@ -9428,7 +9438,10 @@
         "src/xnnpack/AlignedAllocator.h",
     ] + WEIGHTS_PACK_HDRS + MICROKERNEL_BENCHMARK_HDRS,
     copts = xnnpack_optional_ruy_copts(),
-    deps = MICROKERNEL_BENCHMARK_DEPS + [":packing"] + xnnpack_optional_ruy_deps(),
+    deps = MICROKERNEL_BENCHMARK_DEPS + [
+        ":packing",
+        ":jit",
+    ] + xnnpack_optional_ruy_deps(),
 )
 
 xnnpack_benchmark(
@@ -12284,6 +12297,18 @@
     define_values = {"xnn_enable_qu8": "false"},
 )
 
+# Enables usage of JIT kernels.
+config_setting(
+    name = "xnn_enable_jit_explicit_true",
+    define_values = {"xnn_enable_jit": "true"},
+)
+
+# Disables usage of JIT kernels.
+config_setting(
+    name = "xnn_enable_jit_explicit_false",
+    define_values = {"xnn_enable_jit": "false"},
+)
+
 # Target Chrome M87 instructions in WAsm SIMD build
 config_setting(
     name = "xnn_wasmsimd_version_m87",