commit 6883abb3155373fd89dd7f40067e07e1235ac796
author Zhi An Ng <zhin@google.com> Tue Dec 14 10:13:18 2021 -0800
committer XNNPACK Team <xnnpack-github-robot@google.com> Tue Dec 14 10:14:32 2021 -0800
tree 127805255a2cf6c337a361115e35e99b00613b9b
parent 7bd7ecce4814250889d51c334ec7cd8a4af5637d

JIT memory allocation and integration into Assembler

PiperOrigin-RevId: 416333878

BUILD.bazel

diff --git a/BUILD.bazel b/BUILD.bazel
index 537aa24..940277d 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -8084,11 +8084,38 @@
 )
 
 xnnpack_cc_library(
-    name = "aarch32_assembler",
+    name = "jit",
     srcs = [
         "src/jit/aarch32-assembler.cc",
+        "src/jit/memory.c",
     ],
-    hdrs = INTERNAL_HDRS + ["src/xnnpack/aarch32-assembler.h"],
+    hdrs = INTERNAL_HDRS + [
+        "src/xnnpack/aarch32-assembler.h",
+    ],
+    copts = LOGGING_COPTS,
+    msvc_copts = xnnpack_msvc_std_copts(),
+    deps = [
+        ":logging_utils",
+    ],
+)
+
+xnnpack_cc_library(
+    name = "jit_test_mode",
+    srcs = [
+        "src/jit/aarch32-assembler.cc",
+        "src/jit/memory.c",
+    ],
+    hdrs = INTERNAL_HDRS + [
+        "src/xnnpack/aarch32-assembler.h",
+    ],
+    copts = LOGGING_COPTS + [
+        "-UNDEBUG",
+        "-DXNN_TEST_MODE=1",
+    ],
+    msvc_copts = xnnpack_msvc_std_copts(),
+    deps = [
+        ":logging_utils",
+    ],
 )
 
 xnnpack_cc_library(
@@ -11331,7 +11358,8 @@
         "test/aarch32-assembler.cc",
     ],
     deps = [
-        ":aarch32_assembler",
+        ":XNNPACK",
+        ":jit_test_mode",
     ],
 )