commit 4cea2322fe9c00172f4ea9462bc3d0f6fe7a4a78
author Marat Dukhan <maratek@google.com> Tue Mar 09 09:35:36 2021 -0800
committer XNNPACK Team <xnnpack-github-robot@google.com> Tue Mar 09 09:36:31 2021 -0800
tree 53ec696d0256bc532a33ac53bcb821da9a2b16da
parent 52e061d5bc43f56d67f446190e1f83b52c2b3c25

Built-in end-to-end benchmark on sparse models

PiperOrigin-RevId: 361829000

BUILD.bazel

diff --git a/BUILD.bazel b/BUILD.bazel
index d35b38f..1675b74 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -5333,6 +5333,18 @@
 )
 
 cc_library(
+    name = "fp32_sparse_mobilenet_v1",
+    srcs = ["models/fp32-sparse-mobilenet-v1.cc"],
+    hdrs = ["models/models.h"],
+    copts = xnnpack_std_cxxopts(),
+    linkstatic = True,
+    deps = [
+        ":XNNPACK",
+        "@pthreadpool",
+    ],
+)
+
+cc_library(
     name = "fp16_mobilenet_v1",
     srcs = ["models/fp16-mobilenet-v1.cc"],
     hdrs = ["models/models.h"],
@@ -5394,6 +5406,18 @@
 )
 
 cc_library(
+    name = "fp32_sparse_mobilenet_v2",
+    srcs = ["models/fp32-sparse-mobilenet-v2.cc"],
+    hdrs = ["models/models.h"],
+    copts = xnnpack_std_cxxopts(),
+    linkstatic = True,
+    deps = [
+        ":XNNPACK",
+        "@pthreadpool",
+    ],
+)
+
+cc_library(
     name = "fp16_mobilenet_v2",
     srcs = ["models/fp16-mobilenet-v2.cc"],
     hdrs = ["models/models.h"],
@@ -5419,6 +5443,18 @@
 )
 
 cc_library(
+    name = "fp32_sparse_mobilenet_v3_large",
+    srcs = ["models/fp32-sparse-mobilenet-v3-large.cc"],
+    hdrs = ["models/models.h"],
+    copts = xnnpack_std_cxxopts(),
+    linkstatic = True,
+    deps = [
+        ":XNNPACK",
+        "@pthreadpool",
+    ],
+)
+
+cc_library(
     name = "fp16_mobilenet_v3_large",
     srcs = ["models/fp16-mobilenet-v3-large.cc"],
     hdrs = ["models/models.h"],
@@ -5444,6 +5480,18 @@
 )
 
 cc_library(
+    name = "fp32_sparse_mobilenet_v3_small",
+    srcs = ["models/fp32-sparse-mobilenet-v3-small.cc"],
+    hdrs = ["models/models.h"],
+    copts = xnnpack_std_cxxopts(),
+    linkstatic = True,
+    deps = [
+        ":XNNPACK",
+        "@pthreadpool",
+    ],
+)
+
+cc_library(
     name = "fp16_mobilenet_v3_small",
     srcs = ["models/fp16-mobilenet-v3-small.cc"],
     hdrs = ["models/models.h"],
@@ -5513,6 +5561,10 @@
         ":fp32_mobilenet_v2",
         ":fp32_mobilenet_v3_large",
         ":fp32_mobilenet_v3_small",
+        ":fp32_sparse_mobilenet_v1",
+        ":fp32_sparse_mobilenet_v2",
+        ":fp32_sparse_mobilenet_v3_large",
+        ":fp32_sparse_mobilenet_v3_small",
         ":qs8_mobilenet_v1",
         ":qs8_mobilenet_v2",
         ":qu8_mobilenet_v1",