commit 16b734cd279975d4ca70062efb0d507288909c99
author Zhi An Ng <zhin@google.com> Thu Jan 06 13:54:40 2022 -0800
committer Marat Dukhan <maratek@google.com> Thu Jan 06 15:52:08 2022 -0800
tree cbdc422d6d43a4f4a0fabfa9d38745346a84f861
parent ddc49c1b748e547c16056754d287d31efaf2c877

Add more QC8 GEMM/IGEMM JIT microkernels.

PiperOrigin-RevId: 420137137

BUILD.bazel

diff --git a/BUILD.bazel b/BUILD.bazel
index 99c7fae..75e907a 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -7024,14 +7024,16 @@
     "src/f32-gemm/4x8-aarch32-neon-cortex-a53.cc",
     "src/f32-gemm/4x8-aarch32-neon-cortex-a55.cc",
     "src/f32-gemm/4x8-aarch32-neon-cortex-a7.cc",
-    "src/f32-gemm/4x8-aarch32-neon-ld64.cc",
     "src/f32-gemm/4x8-aarch32-neon-cortex-a75.cc",
+    "src/f32-gemm/4x8-aarch32-neon-ld64.cc",
     "src/f32-igemm/4x8-aarch32-neon-cortex-a53.cc",
     "src/f32-igemm/4x8-aarch32-neon-cortex-a55.cc",
     "src/f32-igemm/4x8-aarch32-neon-cortex-a7.cc",
-    "src/f32-igemm/4x8-aarch32-neon-ld64.cc",
     "src/f32-igemm/4x8-aarch32-neon-cortex-a75.cc",
+    "src/f32-igemm/4x8-aarch32-neon-ld64.cc",
+    "src/qc8-gemm/4x8-fp32-aarch32-neonv8-mlal-lane-ld64.cc",
     "src/qc8-gemm/4x8c4-fp32-aarch32-neondot-ld64.cc",
+    "src/qc8-igemm/4x8-fp32-aarch32-neonv8-mlal-lane-ld64.cc",
     "src/qc8-igemm/4x8c4-fp32-aarch32-neondot-ld64.cc",
     "src/qs8-gemm/4x8-rndnu-aarch32-neon-mlal-lane-ld64.cc",
     "src/qs8-gemm/4x8c4-rndnu-aarch32-neondot-ld64.cc",
@@ -11313,7 +11315,10 @@
         "src/xnnpack/AlignedAllocator.h",
     ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
     shard_count = 10,
-    deps = MICROKERNEL_TEST_DEPS + [":packing"],
+    deps = MICROKERNEL_TEST_DEPS + [
+        ":packing",
+        ":jit",
+    ],
 )
 
 xnnpack_unit_test(
@@ -11325,7 +11330,10 @@
         "src/xnnpack/AlignedAllocator.h",
     ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
     shard_count = 10,
-    deps = MICROKERNEL_TEST_DEPS + [":packing"],
+    deps = MICROKERNEL_TEST_DEPS + [
+        ":packing",
+        ":jit",
+    ],
 )
 
 xnnpack_unit_test(