commit bdb56f5de6049c50369fea9caf51c4ce9c452f93
author Marat Dukhan <maratek@google.com> Wed Feb 05 21:42:49 2020 -0800
committer XNNPACK Team <xnnpack-github-robot@google.com> Wed Feb 05 21:44:53 2020 -0800
tree 548e82f3be70f0b0dee0a6c995337c8b416d9d82
parent d16d0f0b3518ea33ec1fbc409517e1e8618e5989

FP16 versions of SpMM micro-kernels

PiperOrigin-RevId: 293518537

BUILD.bazel

diff --git a/BUILD.bazel b/BUILD.bazel
index 91d2a03..095b259 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -843,6 +843,14 @@
     "src/f16-gemm/gen/4x8-neonfp16arith-ld64.c",
     "src/f16-gemm/gen/6x8-neonfp16arith-ld64.c",
     "src/f16-gemm/gen/8x8-neonfp16arith-ld64.c",
+    "src/f16-spmm/gen/8x1-neonfp16arith.c",
+    "src/f16-spmm/gen/8x1-neonfp16arith-unroll2.c",
+    "src/f16-spmm/gen/16x1-neonfp16arith.c",
+    "src/f16-spmm/gen/16x1-neonfp16arith-unroll2.c",
+    "src/f16-spmm/gen/24x1-neonfp16arith.c",
+    "src/f16-spmm/gen/24x1-neonfp16arith-unroll2.c",
+    "src/f16-spmm/gen/32x1-neonfp16arith.c",
+    "src/f16-spmm/gen/32x1-neonfp16arith-unroll2.c",
 ]
 
 SSE_UKERNELS = [
@@ -1942,6 +1950,18 @@
 )
 
 xnnpack_benchmark(
+    name = "f16_spmm_bench",
+    srcs = [
+        "bench/f16-spmm.cc",
+        "bench/gemm.h",
+        "bench/google/gemm.h",
+        "src/xnnpack/AlignedAllocator.h",
+    ] + MICROKERNEL_BENCHMARK_HDRS,
+    copts = ["-Wno-unused-function"],
+    deps = MICROKERNEL_BENCHMARK_DEPS,
+)
+
+xnnpack_benchmark(
     name = "f32_igemm_bench",
     srcs = [
         "bench/f32-igemm.cc",
@@ -2312,6 +2332,16 @@
 )
 
 xnnpack_unit_test(
+    name = "f16_spmm_test",
+    srcs = [
+        "test/f16-spmm.cc",
+        "test/spmm-microkernel-tester.h",
+        "src/xnnpack/AlignedAllocator.h",
+    ] + MICROKERNEL_TEST_HDRS,
+    deps = MICROKERNEL_TEST_DEPS,
+)
+
+xnnpack_unit_test(
     name = "f32_argmaxpool_test",
     srcs = [
         "test/f32-argmaxpool.cc",