commit c2e2da8f941d96c92dc27fddcfdbe0a9e7616482
author Zhi An Ng <zhin@google.com> Tue Jan 25 16:51:58 2022 -0800
committer XNNPACK Team <xnnpack-github-robot@google.com> Tue Jan 25 16:53:13 2022 -0800
tree 65969552ffda091b92ae6f3f09cb4d241eed8952
parent 4a1c6a8179d1e9932f586b56c9163a62a6778851

Fix conversion script for aarch64 assembly kernels and convert a single F32 GEMM as a test

PiperOrigin-RevId: 424212216

BUILD.bazel

diff --git a/BUILD.bazel b/BUILD.bazel
index 7b10d47..f76f50e 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -7314,6 +7314,10 @@
     "src/qs8-igemm/4x8c4-rndnu-aarch32-neondot-ld64.cc",
 ]
 
+JIT_AARCH64_SRCS = [
+    "src/f32-gemm/6x8-aarch64-neonfma-prfm-cortex-a75.cc",
+]
+
 INTERNAL_MICROKERNEL_HDRS = [
     "src/xnnpack/allocator.h",
     "src/xnnpack/argmaxpool.h",
@@ -9169,6 +9173,7 @@
         "src/xnnpack/assembler.h",
     ],
     aarch32_srcs = JIT_AARCH32_SRCS,
+    aarch64_srcs = JIT_AARCH64_SRCS,
     msvc_copts = xnnpack_msvc_std_copts(),
     deps = [
         ":logging_utils",
@@ -9189,6 +9194,7 @@
         "src/xnnpack/assembler.h",
     ],
     aarch32_srcs = JIT_AARCH32_SRCS,
+    aarch64_srcs = JIT_AARCH64_SRCS,
     copts = [
         "-UNDEBUG",
         "-DXNN_TEST_MODE=1",