commit c08cdf5ca8ad15811401c8a19c1b2dc3c8937c7b
author Marat Dukhan <maratek@google.com> Mon Dec 09 09:17:51 2019 -0800
committer XNNPACK Team <xnnpack-github-robot@google.com> Mon Dec 09 09:18:25 2019 -0800
tree 9336a417348e2a67a0d335fabe3c746bd5395a64
parent 4c4eb00a382d12e4506c322cf379d93050c2df5d

Randomized end-to-end MobileNet v3 benchmark

PiperOrigin-RevId: 284567328

BUILD.bazel

diff --git a/BUILD.bazel b/BUILD.bazel
index c3f0d06..3f63eef 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -1782,25 +1782,57 @@
     ],
 )
 
+cc_library(
+    name = "mobilenet_v3_large",
+    srcs = ["models/mobilenet-v3-large.cc"],
+    hdrs = ["models/models.h"],
+    linkstatic = True,
+    deps = [
+        ":XNNPACK",
+        "@pthreadpool",
+    ],
+)
+
+cc_library(
+    name = "mobilenet_v3_small",
+    srcs = ["models/mobilenet-v3-small.cc"],
+    hdrs = ["models/models.h"],
+    linkstatic = True,
+    deps = [
+        ":XNNPACK",
+        "@pthreadpool",
+    ],
+)
+
 xnnpack_benchmark(
     name = "f32_dwconv_e2e_bench",
-    srcs = ["bench/f32-dwconv-e2e.cc"] + MICROKERNEL_BENCHMARK_HDRS,
+    srcs = [
+        "bench/f32-dwconv-e2e.cc",
+        "bench/end2end.h",
+    ] + MICROKERNEL_BENCHMARK_HDRS,
     copts = ["-Wno-unused-function"],
     deps = MICROKERNEL_BENCHMARK_DEPS + [
         ":XNNPACK",
         ":mobilenet_v1",
         ":mobilenet_v2",
+        ":mobilenet_v3_large",
+        ":mobilenet_v3_small",
     ],
 )
 
 xnnpack_benchmark(
     name = "f32_gemm_e2e_bench",
-    srcs = ["bench/f32-gemm-e2e.cc"] + MICROKERNEL_BENCHMARK_HDRS,
+    srcs = [
+        "bench/f32-gemm-e2e.cc",
+        "bench/end2end.h",
+    ] + MICROKERNEL_BENCHMARK_HDRS,
     copts = ["-Wno-unused-function"],
     deps = MICROKERNEL_BENCHMARK_DEPS + [
         ":XNNPACK",
         ":mobilenet_v1",
         ":mobilenet_v2",
+        ":mobilenet_v3_large",
+        ":mobilenet_v3_small",
     ],
 )
 
@@ -1812,6 +1844,8 @@
         ":bench_utils",
         ":mobilenet_v1",
         ":mobilenet_v2",
+        ":mobilenet_v3_large",
+        ":mobilenet_v3_small",
         "@pthreadpool",
     ],
 )