blob: cbd92132608b4af29a65a7ae959d233f96ea4fe4 [file] [log] [blame]
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001# Copyright (c) Facebook, Inc. and its affiliates.
2# All rights reserved.
3#
4# Copyright 2019 Google LLC
5#
6# This source code is licensed under the BSD-style license found in the
7# LICENSE file in the root directory of this source tree.
8
9CMAKE_MINIMUM_REQUIRED(VERSION 3.5 FATAL_ERROR)
10
11INCLUDE(GNUInstallDirs)
12
13# ---[ Project and semantic versioning.
14PROJECT(XNNPACK C CXX ASM)
15
16# ---[ Options.
17SET(XNNPACK_LIBRARY_TYPE "default" CACHE STRING "Type of library (shared, static, or default) to build")
18SET_PROPERTY(CACHE XNNPACK_LIBRARY_TYPE PROPERTY STRINGS default static shared)
19OPTION(XNNPACK_BUILD_TESTS "Build XNNPACK unit tests" ON)
20OPTION(XNNPACK_BUILD_BENCHMARKS "Build XNNPACK benchmarks" ON)
21
22# ---[ CMake options
23IF(XNNPACK_BUILD_TESTS)
24 ENABLE_TESTING()
25ENDIF()
26
27# ---[ Build flags
28IF(NOT CMAKE_SYSTEM_PROCESSOR)
29 IF(IOS)
30 LIST(LENGTH IOS_ARCH IOS_ARCH_COUNT)
31 IF(IOS_ARCH_COUNT GREATER 1)
32 MESSAGE(FATAL_ERROR "Unsupported XNNPACK build with multiple iOS architectures (${IOS_ARCH}). "
33 "Specify a single architecture in IOS_ARCH and re-configure. ")
34 ENDIF()
35 IF(NOT IOS_ARCH MATCHES "^(i386|x86_64|armv7.*|arm64.*)$")
36 MESSAGE(FATAL_ERROR "Unrecognized IOS_ARCH = ${IOS_ARCH}")
37 ENDIF()
38 ELSE()
39 MESSAGE(FATAL_ERROR "CMAKE_SYSTEM_PROCESSOR is not defined")
40 ENDIF()
41ELSEIF(NOT CMAKE_SYSTEM_PROCESSOR MATCHES "^(i[3-6]86|x86_64|armv[5-8].*|aarch64)$")
42 MESSAGE(FATAL_ERROR "Unrecognized CMAKE_SYSTEM_PROCESSOR = ${CMAKE_SYSTEM_PROCESSOR}")
43ENDIF()
44
45IF(NOT CMAKE_SYSTEM_NAME)
46 MESSAGE(FATAL_ERROR "CMAKE_SYSTEM_NAME not defined")
47ELSEIF(NOT CMAKE_SYSTEM_NAME MATCHES "^(Darwin|Linux|Android)$")
48 MESSAGE(FATAL_ERROR "Unrecognized CMAKE_SYSTEM_NAME = ${CMAKE_SYSTEM_NAME}")
49ENDIF()
50
51# ---[ Download deps
52IF(NOT DEFINED CLOG_SOURCE_DIR)
53 MESSAGE(STATUS "Downloading clog to ${CMAKE_BINARY_DIR}/clog-source (define CLOG_SOURCE_DIR to avoid it)")
54 CONFIGURE_FILE(cmake/DownloadCLog.cmake "${CMAKE_BINARY_DIR}/clog-download/CMakeLists.txt")
55 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" -G "${CMAKE_GENERATOR}" .
56 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/clog-download")
57 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" --build .
58 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/clog-download")
59 SET(CLOG_SOURCE_DIR "${CMAKE_BINARY_DIR}/clog-source" CACHE STRING "clog source directory")
60ENDIF()
61
62IF(NOT DEFINED CPUINFO_SOURCE_DIR)
63 MESSAGE(STATUS "Downloading cpuinfo to ${CMAKE_BINARY_DIR}/cpuinfo-source (define CPUINFO_SOURCE_DIR to avoid it)")
64 CONFIGURE_FILE(cmake/DownloadCpuinfo.cmake "${CMAKE_BINARY_DIR}/cpuinfo-download/CMakeLists.txt")
65 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" -G "${CMAKE_GENERATOR}" .
66 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/cpuinfo-download")
67 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" --build .
68 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/cpuinfo-download")
69 SET(CPUINFO_SOURCE_DIR "${CMAKE_BINARY_DIR}/cpuinfo-source" CACHE STRING "cpuinfo source directory")
70ENDIF()
71
72IF(NOT DEFINED FP16_SOURCE_DIR)
73 MESSAGE(STATUS "Downloading FP16 to ${CMAKE_BINARY_DIR}/FP16-source (define FP16_SOURCE_DIR to avoid it)")
74 CONFIGURE_FILE(cmake/DownloadFP16.cmake "${CMAKE_BINARY_DIR}/FP16-download/CMakeLists.txt")
75 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" -G "${CMAKE_GENERATOR}" .
76 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/FP16-download")
77 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" --build .
78 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/FP16-download")
79 SET(FP16_SOURCE_DIR "${CMAKE_BINARY_DIR}/FP16-source" CACHE STRING "FP16 source directory")
80ENDIF()
81
82IF(NOT DEFINED FXDIV_SOURCE_DIR)
83 MESSAGE(STATUS "Downloading FXdiv to ${CMAKE_BINARY_DIR}/FXdiv-source (define FXDIV_SOURCE_DIR to avoid it)")
84 CONFIGURE_FILE(cmake/DownloadFXdiv.cmake "${CMAKE_BINARY_DIR}/FXdiv-download/CMakeLists.txt")
85 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" -G "${CMAKE_GENERATOR}" .
86 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/FXdiv-download")
87 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" --build .
88 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/FXdiv-download")
89 SET(FXDIV_SOURCE_DIR "${CMAKE_BINARY_DIR}/FXdiv-source" CACHE STRING "FXdiv source directory")
90ENDIF()
91
92IF(NOT DEFINED PSIMD_SOURCE_DIR)
93 MESSAGE(STATUS "Downloading PSimd to ${CMAKE_BINARY_DIR}/psimd-source (define PSIMD_SOURCE_DIR to avoid it)")
94 CONFIGURE_FILE(cmake/DownloadPSimd.cmake "${CMAKE_BINARY_DIR}/psimd-download/CMakeLists.txt")
95 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" -G "${CMAKE_GENERATOR}" .
96 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/psimd-download")
97 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" --build .
98 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/psimd-download")
99 SET(PSIMD_SOURCE_DIR "${CMAKE_BINARY_DIR}/psimd-source" CACHE STRING "PSimd source directory")
100ENDIF()
101
102IF(NOT DEFINED PTHREADPOOL_SOURCE_DIR)
103 MESSAGE(STATUS "Downloading pthreadpool to ${CMAKE_BINARY_DIR}/pthreadpool-source (define PTHREADPOOL_SOURCE_DIR to avoid it)")
104 CONFIGURE_FILE(cmake/DownloadPThreadPool.cmake "${CMAKE_BINARY_DIR}/pthreadpool-download/CMakeLists.txt")
105 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" -G "${CMAKE_GENERATOR}" .
106 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/pthreadpool-download")
107 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" --build .
108 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/pthreadpool-download")
109 SET(PTHREADPOOL_SOURCE_DIR "${CMAKE_BINARY_DIR}/pthreadpool-source" CACHE STRING "pthreadpool source directory")
110ENDIF()
111
112IF(XNNPACK_BUILD_TESTS AND NOT DEFINED GOOGLETEST_SOURCE_DIR)
113 MESSAGE(STATUS "Downloading Google Test to ${CMAKE_BINARY_DIR}/googletest-source (define GOOGLETEST_SOURCE_DIR to avoid it)")
114 CONFIGURE_FILE(cmake/DownloadGoogleTest.cmake "${CMAKE_BINARY_DIR}/googletest-download/CMakeLists.txt")
115 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" -G "${CMAKE_GENERATOR}" .
116 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/googletest-download")
117 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" --build .
118 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/googletest-download")
119 SET(GOOGLETEST_SOURCE_DIR "${CMAKE_BINARY_DIR}/googletest-source" CACHE STRING "Google Test source directory")
120ENDIF()
121
122IF(XNNPACK_BUILD_BENCHMARKS AND NOT DEFINED GOOGLEBENCHMARK_SOURCE_DIR)
123 MESSAGE(STATUS "Downloading Google Benchmark to ${CMAKE_BINARY_DIR}/googlebenchmark-source (define GOOGLEBENCHMARK_SOURCE_DIR to avoid it)")
124 CONFIGURE_FILE(cmake/DownloadGoogleBenchmark.cmake "${CMAKE_BINARY_DIR}/googlebenchmark-download/CMakeLists.txt")
125 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" -G "${CMAKE_GENERATOR}" .
126 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/googlebenchmark-download")
127 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" --build .
128 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/googlebenchmark-download")
129 SET(GOOGLEBENCHMARK_SOURCE_DIR "${CMAKE_BINARY_DIR}/googlebenchmark-source" CACHE STRING "Google Benchmark source directory")
130ENDIF()
131
132# ---[ XNNPACK library
133SET(XNNPACK_OPERATOR_SRCS
Marat Dukhanefc47b82019-11-18 09:25:38 -0800134 src/add-nc.c
135 src/argmax-pooling-nhwc.c
136 src/average-pooling-nhwc.c
137 src/channel-pad-nc.c
138 src/channel-shuffle-nc.c
139 src/clamp-nc.c
140 src/convolution-nchw.c
141 src/convolution-nhwc.c
142 src/deconvolution-nhwc.c
143 src/fully-connected-nc.c
144 src/global-average-pooling-ncw.c
145 src/global-average-pooling-nwc.c
146 src/hardswish-nc.c
147 src/leaky-relu-nc.c
148 src/max-pooling-nhwc.c
149 src/multiply-nd.c
150 src/prelu-nc.c
151 src/resize-bilinear-nhwc.c
152 src/sigmoid-nc.c
153 src/softargmax-nc.c
154 src/unpooling-nhwc.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700155
156SET(XNNPACK_COLD_SRCS ${XNNPACK_OPERATOR_SRCS})
157LIST(APPEND XNNPACK_COLD_SRCS
158 src/init.c
Ashkan Aliabadi1f8a2b82019-11-20 11:27:00 -0800159 src/memory.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700160 src/operator-delete.c)
161
162SET(XNNPACK_HOT_SRCS
163 src/indirection.c
164 src/operator-run.c)
165
166SET(XNNPACK_SCALAR_MICROKERNEL_SRCS
Ashkan Aliabadif866a452019-11-20 10:58:51 -0800167 src/f32-argmaxpool/4x-scalar-c1.c
168 src/f32-argmaxpool/9p8x-scalar-c1.c
169 src/f32-argmaxpool/9x-scalar-c1.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700170 src/f32-avgpool/mp9p8q-scalar.c
171 src/f32-avgpool/up9-scalar.c
Marat Dukhan35dacfb2019-11-07 19:18:16 -0800172 src/f32-bilinear/scalar-c1.c
173 src/f32-bilinear/scalar-c2.c
174 src/f32-bilinear/scalar-c4.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700175 src/f32-clamp/scalar.c
Erich Elsen563df5f2019-10-23 08:02:21 -0700176 src/f32-conv-hwc2spchw/3x3s2p1c3x4-scalar-1x1.c
Erich Elsen0cc2c532019-10-15 04:44:18 -0700177 src/f32-dwconv-spchw/3x3p1-scalar.c
Erich Elsenac4de802019-10-16 04:35:30 -0700178 src/f32-dwconv-spchw/3x3s2p1-scalar.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800179 src/f32-dwconv-spchw/5x5p2-scalar.c
Erich Elsen38709a62019-11-08 11:58:45 -0800180 src/f32-dwconv-spchw/5x5s2p2-scalar.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800181 src/f32-dwconv/up1x25-scalar-acc2.c
182 src/f32-dwconv/up1x25-scalar.c
183 src/f32-dwconv/up1x4-scalar-acc2.c
184 src/f32-dwconv/up1x4-scalar.c
185 src/f32-dwconv/up1x9-scalar-acc2.c
186 src/f32-dwconv/up1x9-scalar.c
187 src/f32-dwconv/up2x25-scalar-acc2.c
188 src/f32-dwconv/up2x25-scalar.c
189 src/f32-dwconv/up2x4-scalar-acc2.c
190 src/f32-dwconv/up2x4-scalar.c
191 src/f32-dwconv/up2x9-scalar-acc2.c
192 src/f32-dwconv/up2x9-scalar.c
Erich Elsen34dc2c02019-10-16 05:11:41 -0700193 src/f32-gavgpool-spchw/scalar-x1.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700194 src/f32-gavgpool/mp7p7q-scalar.c
195 src/f32-gavgpool/up7-scalar.c
196 src/f32-gemm/1x4-scalar.c
197 src/f32-gemm/2x4-scalar.c
198 src/f32-gemm/4x2-scalar.c
199 src/f32-gemm/4x4-scalar.c
200 src/f32-gemminc/1x4-scalar.c
201 src/f32-gemminc/2x4-scalar.c
202 src/f32-gemminc/4x4-scalar.c
203 src/f32-hswish/scalar.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800204 src/f32-igemm/1x4-scalar.c
205 src/f32-igemm/2x4-scalar.c
206 src/f32-igemm/4x2-scalar.c
207 src/f32-igemm/4x4-scalar.c
Marat Dukhan329da642019-11-19 21:44:39 -0800208 src/f32-maxpool/9p8x-scalar-c1.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700209 src/f32-pavgpool/mp9p8q-scalar.c
210 src/f32-pavgpool/up9-scalar.c
211 src/f32-ppmm/2x4-scalar.c
212 src/f32-ppmm/3x3-scalar.c
213 src/f32-ppmm/4x2-scalar.c
214 src/f32-ppmm/4x4-scalar.c
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800215 src/f32-prelu/scalar-2x1.c
216 src/f32-prelu/scalar-2x4.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700217 src/f32-rmax/scalar.c
218 src/f32-spmm/1x1-scalar-pipelined.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700219 src/f32-spmm/1x1-scalar.c
220 src/f32-spmm/2x1-scalar-pipelined.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700221 src/f32-spmm/2x1-scalar.c
222 src/f32-spmm/4x1-scalar-pipelined.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700223 src/f32-spmm/4x1-scalar.c
224 src/f32-spmm/8x1-scalar-pipelined.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700225 src/f32-spmm/8x1-scalar.c
Erich Elsenc6afd9b2019-10-24 16:10:53 -0700226 src/f32-spmm/8x2-scalar.c
227 src/f32-spmm/8x4-scalar.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800228 src/f32-vbinary/vadd-scalar-x1.c
229 src/f32-vbinary/vadd-scalar-x2.c
230 src/f32-vbinary/vadd-scalar-x4.c
231 src/f32-vbinary/vaddc-scalar-x1.c
232 src/f32-vbinary/vaddc-scalar-x2.c
233 src/f32-vbinary/vaddc-scalar-x4.c
234 src/f32-vbinary/vmul-scalar-x1.c
235 src/f32-vbinary/vmul-scalar-x2.c
236 src/f32-vbinary/vmul-scalar-x4.c
237 src/f32-vbinary/vmulc-scalar-x1.c
238 src/f32-vbinary/vmulc-scalar-x2.c
239 src/f32-vbinary/vmulc-scalar-x4.c
240 src/f32-vbinary/vrsubc-scalar-x1.c
241 src/f32-vbinary/vrsubc-scalar-x2.c
242 src/f32-vbinary/vrsubc-scalar-x4.c
243 src/f32-vbinary/vsub-scalar-x1.c
244 src/f32-vbinary/vsub-scalar-x2.c
245 src/f32-vbinary/vsub-scalar-x4.c
246 src/f32-vbinary/vsubc-scalar-x1.c
247 src/f32-vbinary/vsubc-scalar-x2.c
248 src/f32-vbinary/vsubc-scalar-x4.c
Marat Dukhan49e6ee92019-11-06 15:55:29 -0800249 src/f32-vmulcaddc/c1-scalar-2x.c
250 src/f32-vmulcaddc/c2-scalar-2x.c
251 src/f32-vmulcaddc/c4-scalar-2x.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700252 src/q8-avgpool/mp9p8q-scalar.c
253 src/q8-avgpool/up9-scalar.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700254 src/q8-dwconv/up1x9-scalar.c
255 src/q8-gavgpool/mp7p7q-scalar.c
256 src/q8-gavgpool/up7-scalar.c
257 src/q8-gemm/2x2-scalar.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800258 src/q8-igemm/2x2-scalar.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700259 src/q8-vadd/scalar.c
260 src/u8-clamp/scalar.c
261 src/u8-lut32norm/scalar.c
Marat Dukhan329da642019-11-19 21:44:39 -0800262 src/u8-maxpool/9p8x-scalar-c1.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700263 src/u8-rmax/scalar.c
264 src/x32-packx/x2-scalar.c
265 src/x32-packx/x3-scalar.c
266 src/x32-packx/x4-scalar.c
267 src/x32-pad/x2-scalar.c
268 src/x32-unpool/scalar.c
269 src/x32-zip/x2-scalar.c
270 src/x32-zip/x3-scalar.c
271 src/x32-zip/x4-scalar.c
272 src/x32-zip/xm-scalar.c
273 src/x8-lut/scalar.c
274 src/x8-zip/x2-scalar.c
275 src/x8-zip/x3-scalar.c
276 src/x8-zip/x4-scalar.c
277 src/x8-zip/xm-scalar.c)
278
279SET(XNNPACK_PSIMD_MICROKERNEL_SRCS
Ashkan Aliabadif866a452019-11-20 10:58:51 -0800280 src/f32-argmaxpool/4x-psimd-c4.c
281 src/f32-argmaxpool/9p8x-psimd-c4.c
282 src/f32-argmaxpool/9x-psimd-c4.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700283 src/f32-avgpool/mp9p8q-psimd.c
284 src/f32-avgpool/up9-psimd.c
Marat Dukhan35dacfb2019-11-07 19:18:16 -0800285 src/f32-bilinear/psimd-c4.c
286 src/f32-bilinear/psimd-c8.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700287 src/f32-clamp/psimd.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800288 src/f32-dwconv/up4x25-psimd-acc2.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800289 src/f32-dwconv/up4x25-psimd.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800290 src/f32-dwconv/up4x4-psimd-acc2.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800291 src/f32-dwconv/up4x4-psimd.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800292 src/f32-dwconv/up4x9-psimd-acc2.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800293 src/f32-dwconv/up4x9-psimd.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800294 src/f32-dwconv/up8x25-psimd-acc2.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800295 src/f32-dwconv/up8x25-psimd.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800296 src/f32-dwconv/up8x4-psimd-acc2.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800297 src/f32-dwconv/up8x4-psimd.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800298 src/f32-dwconv/up8x9-psimd-acc2.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800299 src/f32-dwconv/up8x9-psimd.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700300 src/f32-gavgpool/mp7p7q-psimd.c
301 src/f32-gavgpool/up7-psimd.c
302 src/f32-gemm/1x8-psimd-loadsplat.c
303 src/f32-gemm/1x8-psimd-splat.c
304 src/f32-gemm/1x8s4-psimd.c
305 src/f32-gemm/4x8-psimd-loadsplat.c
306 src/f32-gemm/4x8-psimd-splat.c
307 src/f32-gemm/4x8s4-psimd.c
308 src/f32-gemm/6x8-psimd-loadsplat.c
309 src/f32-gemm/6x8-psimd-splat.c
310 src/f32-gemm/6x8s4-psimd.c
311 src/f32-gemminc/1x8-psimd-loadsplat.c
312 src/f32-gemminc/1x8-psimd-splat.c
313 src/f32-gemminc/1x8s4-psimd.c
314 src/f32-gemminc/4x8-psimd-loadsplat.c
315 src/f32-gemminc/4x8-psimd-splat.c
316 src/f32-gemminc/4x8s4-psimd.c
317 src/f32-gemminc/6x8-psimd-loadsplat.c
318 src/f32-gemminc/6x8-psimd-splat.c
319 src/f32-gemminc/6x8s4-psimd.c
320 src/f32-hswish/psimd.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800321 src/f32-igemm/1x8-psimd-loadsplat.c
322 src/f32-igemm/1x8-psimd-splat.c
323 src/f32-igemm/1x8s4-psimd.c
324 src/f32-igemm/4x2c4-psimd.c
325 src/f32-igemm/4x8-psimd-loadsplat.c
326 src/f32-igemm/4x8-psimd-splat.c
327 src/f32-igemm/4x8s4-psimd.c
328 src/f32-igemm/6x8-psimd-loadsplat.c
329 src/f32-igemm/6x8-psimd-splat.c
330 src/f32-igemm/6x8s4-psimd.c
Marat Dukhan329da642019-11-19 21:44:39 -0800331 src/f32-maxpool/9p8x-psimd-c4.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700332 src/f32-pavgpool/mp9p8q-psimd.c
333 src/f32-pavgpool/up9-psimd.c
334 src/f32-ppmm/4x8-psimd.c
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800335 src/f32-prelu/psimd-2x4.c
336 src/f32-prelu/psimd-2x8.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800337 src/f32-vbinary/vadd-psimd-x4.c
338 src/f32-vbinary/vadd-psimd-x8.c
339 src/f32-vbinary/vaddc-psimd-x4.c
340 src/f32-vbinary/vaddc-psimd-x8.c
341 src/f32-vbinary/vmul-psimd-x4.c
342 src/f32-vbinary/vmul-psimd-x8.c
343 src/f32-vbinary/vmulc-psimd-x4.c
344 src/f32-vbinary/vmulc-psimd-x8.c
345 src/f32-vbinary/vrsubc-psimd-x4.c
346 src/f32-vbinary/vrsubc-psimd-x8.c
347 src/f32-vbinary/vsub-psimd-x4.c
348 src/f32-vbinary/vsub-psimd-x8.c
349 src/f32-vbinary/vsubc-psimd-x4.c
350 src/f32-vbinary/vsubc-psimd-x8.c
Marat Dukhan49e6ee92019-11-06 15:55:29 -0800351 src/f32-vmulcaddc/c4-psimd-2x.c
352 src/f32-vmulcaddc/c8-psimd-2x.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700353 src/x32-packx/x4-psimd.c
354 src/x32-pad/x2-psimd.c
355 src/x32-unpool/psimd.c
356 src/x32-zip/x2-psimd.c
357 src/x32-zip/x3-psimd.c
358 src/x32-zip/x4-psimd.c
359 src/x32-zip/xm-psimd.c)
360
361SET(XNNPACK_NEON_MICROKERNEL_SRCS
362 src/f32-avgpool/mp9p8q-neon.c
363 src/f32-avgpool/up9-neon.c
Marat Dukhan35dacfb2019-11-07 19:18:16 -0800364 src/f32-bilinear/neon-c4.c
365 src/f32-bilinear/neon-c8.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700366 src/f32-clamp/neon.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800367 src/f32-dwconv/up4x9-neon-acc2.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800368 src/f32-dwconv/up4x9-neon.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800369 src/f32-dwconv/up8x9-neon-acc2.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800370 src/f32-dwconv/up8x9-neon.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700371 src/f32-gavgpool-spchw/neon-x4.c
372 src/f32-gavgpool/mp7p7q-neon.c
373 src/f32-gavgpool/up7-neon.c
374 src/f32-gemm/1x8-neon-ld64.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800375 src/f32-gemm/1x8s4-neon.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700376 src/f32-gemm/4x2-neon-ld64.c
377 src/f32-gemm/4x8-neon-ld128.c
378 src/f32-gemm/4x8-neon-ld64.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800379 src/f32-gemm/4x8s4-neon.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700380 src/f32-gemm/5x8-neon-ld64.c
381 src/f32-gemm/6x8-neon-ld64.c
Frank Barcharddf06d802019-11-20 15:53:46 -0800382 src/f32-gemm/6x8s4-neon.c
383 src/f32-gemm/8x8s4-neon.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700384 src/f32-gemminc/1x8-neon-ld64.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800385 src/f32-gemminc/1x8s4-neon.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700386 src/f32-gemminc/4x8-neon-ld128.c
387 src/f32-gemminc/4x8-neon-ld64.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800388 src/f32-gemminc/4x8s4-neon.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700389 src/f32-gemminc/5x8-neon-ld64.c
390 src/f32-gemminc/6x8-neon-ld64.c
Frank Barcharddf06d802019-11-20 15:53:46 -0800391 src/f32-gemminc/6x8s4-neon.c
392 src/f32-gemminc/8x8s4-neon.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700393 src/f32-hswish/neon.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800394 src/f32-igemm/1x8-neon-ld64.c
395 src/f32-igemm/1x8s4-neon.c
396 src/f32-igemm/4x2-neon-ld64.c
397 src/f32-igemm/4x4-neon-ld64.c
398 src/f32-igemm/4x8-neon-ld128.c
399 src/f32-igemm/4x8-neon-ld64.c
400 src/f32-igemm/4x8s4-neon.c
401 src/f32-igemm/6x8-neon-ld64.c
402 src/f32-igemm/6x8s4-neon.c
403 src/f32-igemm/8x8s4-neon.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700404 src/f32-pavgpool/mp9p8q-neon.c
405 src/f32-pavgpool/up9-neon.c
406 src/f32-ppmm/4x8-neon.c
407 src/f32-ppmm/8x8-neon.c
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800408 src/f32-prelu/neon-2x4.c
409 src/f32-prelu/neon-2x8.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700410 src/f32-rmax/neon.c
Marat Dukhan7c6c3b22019-11-18 14:42:50 -0800411 src/f32-sigmoid/neon-frac-p9-p10-nr1recps-x16.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800412 src/f32-vbinary/vadd-neon-x4.c
413 src/f32-vbinary/vadd-neon-x8.c
414 src/f32-vbinary/vaddc-neon-x4.c
415 src/f32-vbinary/vaddc-neon-x8.c
416 src/f32-vbinary/vmul-neon-x4.c
417 src/f32-vbinary/vmul-neon-x8.c
418 src/f32-vbinary/vmulc-neon-x4.c
419 src/f32-vbinary/vmulc-neon-x8.c
420 src/f32-vbinary/vrsubc-neon-x4.c
421 src/f32-vbinary/vrsubc-neon-x8.c
422 src/f32-vbinary/vsub-neon-x4.c
423 src/f32-vbinary/vsub-neon-x8.c
424 src/f32-vbinary/vsubc-neon-x4.c
425 src/f32-vbinary/vsubc-neon-x8.c
Marat Dukhan49e6ee92019-11-06 15:55:29 -0800426 src/f32-vmulcaddc/c4-neon-2x.c
427 src/f32-vmulcaddc/c8-neon-2x.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700428 src/q8-avgpool/mp9p8q-neon.c
429 src/q8-avgpool/up9-neon.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700430 src/q8-dwconv/up8x9-neon.c
431 src/q8-gavgpool/mp7p7q-neon.c
432 src/q8-gavgpool/up7-neon.c
433 src/q8-gemm/4x8-neon.c
434 src/q8-gemm/8x8-neon.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800435 src/q8-igemm/4x8-neon.c
436 src/q8-igemm/8x8-neon.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700437 src/q8-vadd/neon.c
438 src/u8-clamp/neon.c
Marat Dukhan329da642019-11-19 21:44:39 -0800439 src/u8-maxpool/9p8x-neon-c16.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700440 src/u8-rmax/neon.c
441 src/x32-packx/x4-neon-st4.c
442 src/x32-pad/x2-neon.c
443 src/x32-zip/x2-neon.c
444 src/x32-zip/x3-neon.c
445 src/x32-zip/x4-neon.c
446 src/x32-zip/xm-neon.c
447 src/x8-zip/x2-neon.c
448 src/x8-zip/x3-neon.c
449 src/x8-zip/x4-neon.c
450 src/x8-zip/xm-neon.c)
451
452SET(XNNPACK_NEONFMA_MICROKERNEL_SRCS
Marat Dukhan35dacfb2019-11-07 19:18:16 -0800453 src/f32-bilinear/neonfma-c4.c
454 src/f32-bilinear/neonfma-c8.c
Frank Barcharddb45b6a2019-10-09 16:42:45 -0700455 src/f32-igemm/1x8-neonfma-ld64.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700456 src/f32-igemm/4x2-neonfma-ld64.c
457 src/f32-igemm/4x4-neonfma-ld64.c
458 src/f32-igemm/4x8-neonfma-ld128.c
459 src/f32-igemm/4x8-neonfma-ld64.c
460 src/f32-igemm/6x8-neonfma-ld64.c
Frank Barcharddf06d802019-11-20 15:53:46 -0800461 src/f32-igemm/1x8s4-neonfma.c
462 src/f32-igemm/4x8s4-neonfma.c
463 src/f32-igemm/6x8s4-neonfma.c
464 src/f32-igemm/8x8s4-neonfma.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700465 src/f32-dwconv/up4x9-neonfma.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800466 src/f32-dwconv/up4x9-neonfma-acc2.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700467 src/f32-dwconv/up8x9-neonfma.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800468 src/f32-dwconv/up8x9-neonfma-acc2.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700469 src/f32-gemm/1x8-neonfma-ld64.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700470 src/f32-gemm/4x2-neonfma-ld64.c
471 src/f32-gemm/4x8-neonfma-ld128.c
472 src/f32-gemm/4x8-neonfma-ld64.c
473 src/f32-gemm/5x8-neonfma-ld64.c
474 src/f32-gemm/6x8-neonfma-ld64.c
Frank Barcharddf06d802019-11-20 15:53:46 -0800475 src/f32-gemm/1x8s4-neonfma.c
476 src/f32-gemm/4x8s4-neonfma.c
477 src/f32-gemm/6x8s4-neonfma.c
478 src/f32-gemm/8x8s4-neonfma.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700479 src/f32-gemminc/1x8-neonfma-ld64.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700480 src/f32-gemminc/4x8-neonfma-ld128.c
481 src/f32-gemminc/4x8-neonfma-ld64.c
482 src/f32-gemminc/5x8-neonfma-ld64.c
483 src/f32-gemminc/6x8-neonfma-ld64.c
Frank Barcharddf06d802019-11-20 15:53:46 -0800484 src/f32-gemminc/1x8s4-neonfma.c
485 src/f32-gemminc/4x8s4-neonfma.c
486 src/f32-gemminc/6x8s4-neonfma.c
487 src/f32-gemminc/8x8s4-neonfma.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700488 src/f32-hswish/neonfma.c
489 src/f32-ppmm/4x8-neonfma.c
490 src/f32-ppmm/8x8-neonfma.c
Marat Dukhan7c6c3b22019-11-18 14:42:50 -0800491 src/f32-sigmoid/neonfma-p5-nr2fma-x16.c
Marat Dukhan49e6ee92019-11-06 15:55:29 -0800492 src/f32-vmulcaddc/c4-neonfma-2x.c
Marat Dukhan9d501d52019-11-16 02:30:50 -0800493 src/f32-vmulcaddc/c8-neonfma-2x.c
494 src/math/exp-neonfma-lut64-p2.c
495 src/math/exp-neonfma-p5.c
496 src/math/expminus-neonfma-p5.c
Marat Dukhan80bafd22019-11-18 10:16:01 -0800497 src/math/sigmoid-neonfma-p5-nr2fma.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700498
499SET(AARCH64_XNNPACK_NEONFMA_MICROKERNEL_SRCS
500 src/f32-conv-hwc/3x3s2p1c3x4-neonfma-2x2.c
501 src/f32-conv-hwc/3x3s2p1c3x8-neonfma-2x2.c
502 src/f32-conv-hwc2spchw/3x3s2p1c3x4-neonfma-2x2.c
503 src/f32-dwconv-spchw/3x3p1-neonfma.c
504 src/f32-dwconv-spchw/5x5p2-neonfma.c
505 src/f32-dwconv-spchw/3x3s2p1-neonfma.c
506 src/f32-dwconv-spchw/5x5s2p2-neonfma.c
507 src/f32-spmm/12x1-neonfma.c
508 src/f32-spmm/12x2-neonfma.c
509 src/f32-spmm/12x4-neonfma.c
510 src/f32-spmm/16x1-neonfma-pipelined.c
511 src/f32-spmm/16x1-neonfma-unroll2.c
512 src/f32-spmm/16x1-neonfma.c
513 src/f32-spmm/16x2-neonfma.c
514 src/f32-spmm/16x4-neonfma.c
515 src/f32-spmm/4x1-neonfma-pipelined.c
516 src/f32-spmm/4x1-neonfma-unroll2.c
517 src/f32-spmm/4x1-neonfma.c
518 src/f32-spmm/4x2-neonfma.c
519 src/f32-spmm/4x4-neonfma.c
520 src/f32-spmm/8x1-neonfma-pipelined.c
521 src/f32-spmm/8x1-neonfma-unroll2.c
522 src/f32-spmm/8x1-neonfma.c
523 src/f32-spmm/8x2-neonfma.c
524 src/f32-spmm/8x4-neonfma.c)
525
526SET(XNNPACK_AARCH64_NEONFP16ARITH_MICROKERNEL_SRCS
527 src/f16-gemm/4x8-neonfp16arith-ld64.c
528 src/f16-gemm/6x8-neonfp16arith-ld64.c
529 src/f16-gemm/8x8-neonfp16arith-ld64.c)
530
531SET(XNNPACK_SSE_MICROKERNEL_SRCS
532 src/f32-avgpool/mp9p8q-sse.c
533 src/f32-avgpool/up9-sse.c
Marat Dukhan35dacfb2019-11-07 19:18:16 -0800534 src/f32-bilinear/sse-c4.c
535 src/f32-bilinear/sse-c8.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700536 src/f32-clamp/sse.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800537 src/f32-dwconv-spchw/3x3p1-sse.c
538 src/f32-dwconv-spchw/3x3s2p1-sse.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800539 src/f32-dwconv/up4x25-sse-acc2.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800540 src/f32-dwconv/up4x25-sse.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800541 src/f32-dwconv/up4x4-sse-acc2.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800542 src/f32-dwconv/up4x4-sse.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800543 src/f32-dwconv/up4x9-sse-acc2.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800544 src/f32-dwconv/up4x9-sse.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800545 src/f32-dwconv/up8x25-sse-acc2.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800546 src/f32-dwconv/up8x25-sse.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800547 src/f32-dwconv/up8x4-sse-acc2.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800548 src/f32-dwconv/up8x4-sse.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800549 src/f32-dwconv/up8x9-sse-acc2.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800550 src/f32-dwconv/up8x9-sse.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700551 src/f32-gavgpool-spchw/sse-x4.c
552 src/f32-gavgpool/mp7p7q-sse.c
553 src/f32-gavgpool/up7-sse.c
554 src/f32-gemm/1x8-sse-dup.c
555 src/f32-gemm/1x8-sse-load1.c
556 src/f32-gemm/1x8s4-sse.c
557 src/f32-gemm/4x8-sse-dup.c
558 src/f32-gemm/4x8-sse-load1.c
559 src/f32-gemm/4x8s4-sse.c
560 src/f32-gemminc/1x8-sse-dup.c
561 src/f32-gemminc/1x8-sse-load1.c
562 src/f32-gemminc/1x8s4-sse.c
563 src/f32-gemminc/4x8-sse-dup.c
564 src/f32-gemminc/4x8-sse-load1.c
565 src/f32-gemminc/4x8s4-sse.c
566 src/f32-hswish/sse.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800567 src/f32-igemm/1x8-sse-dup.c
568 src/f32-igemm/1x8-sse-load1.c
569 src/f32-igemm/1x8s4-sse.c
570 src/f32-igemm/4x2c4-sse.c
571 src/f32-igemm/4x8-sse-dup.c
572 src/f32-igemm/4x8-sse-load1.c
573 src/f32-igemm/4x8s4-sse.c
Marat Dukhan329da642019-11-19 21:44:39 -0800574 src/f32-maxpool/9p8x-sse-c4.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700575 src/f32-pavgpool/mp9p8q-sse.c
576 src/f32-pavgpool/up9-sse.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700577 src/f32-ppmm/4x8-sse.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700578 src/f32-rmax/sse.c
579 src/f32-spmm/4x1-sse.c
580 src/f32-spmm/8x1-sse.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800581 src/f32-vbinary/vadd-sse-x4.c
582 src/f32-vbinary/vadd-sse-x8.c
583 src/f32-vbinary/vaddc-sse-x4.c
584 src/f32-vbinary/vaddc-sse-x8.c
585 src/f32-vbinary/vmul-sse-x4.c
586 src/f32-vbinary/vmul-sse-x8.c
587 src/f32-vbinary/vmulc-sse-x4.c
588 src/f32-vbinary/vmulc-sse-x8.c
589 src/f32-vbinary/vrsubc-sse-x4.c
590 src/f32-vbinary/vrsubc-sse-x8.c
591 src/f32-vbinary/vsub-sse-x4.c
592 src/f32-vbinary/vsub-sse-x8.c
593 src/f32-vbinary/vsubc-sse-x4.c
594 src/f32-vbinary/vsubc-sse-x8.c
Marat Dukhan49e6ee92019-11-06 15:55:29 -0800595 src/f32-vmulcaddc/c4-sse-2x.c
596 src/f32-vmulcaddc/c8-sse-2x.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700597 src/x32-packx/x4-sse.c)
598
599SET(XNNPACK_SSE2_MICROKERNEL_SRCS
Ashkan Aliabadif866a452019-11-20 10:58:51 -0800600 src/f32-argmaxpool/4x-sse2-c4.c
601 src/f32-argmaxpool/9p8x-sse2-c4.c
602 src/f32-argmaxpool/9x-sse2-c4.c
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800603 src/f32-prelu/sse2-2x4.c
604 src/f32-prelu/sse2-2x8.c
Marat Dukhan7bee7512019-11-18 15:15:48 -0800605 src/f32-sigmoid/sse2-p5-div-x8.c
606 src/f32-sigmoid/sse2-p5-div-x16.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700607 src/q8-avgpool/mp9p8q-sse2.c
608 src/q8-avgpool/up9-sse2.c
609 src/q8-igemm/4x4c2-sse2.c
610 src/q8-dwconv/up8x9-sse2.c
611 src/q8-gavgpool/mp7p7q-sse2.c
612 src/q8-gavgpool/up7-sse2.c
613 src/q8-gemm/2x4c8-sse2.c
614 src/q8-gemm/4x4c2-sse2.c
615 src/q8-vadd/sse2.c
616 src/u8-clamp/sse2.c
Marat Dukhan329da642019-11-19 21:44:39 -0800617 src/u8-maxpool/9p8x-sse2-c16.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700618 src/u8-rmax/sse2.c
619 src/x32-pad/x2-sse2.c
620 src/x32-zip/x2-sse2.c
621 src/x32-zip/x3-sse2.c
622 src/x32-zip/x4-sse2.c
623 src/x32-zip/xm-sse2.c
624 src/x8-zip/x2-sse2.c
625 src/x8-zip/x3-sse2.c
626 src/x8-zip/x4-sse2.c
Marat Dukhan9d501d52019-11-16 02:30:50 -0800627 src/x8-zip/xm-sse2.c
628 src/math/exp-sse2-p5.c
Marat Dukhan80bafd22019-11-18 10:16:01 -0800629 src/math/expminus-sse2-p5.c
630 src/math/sigmoid-sse2-p5-div.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700631
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800632SET(XNNPACK_SSE41_MICROKERNEL_SRCS
633 src/f32-prelu/sse41-2x4.c
Marat Dukhan496e7352019-11-21 15:48:40 -0800634 src/f32-prelu/sse41-2x8.c
635 src/f32-sigmoid/sse41-p5-div-x8.c
636 src/f32-sigmoid/sse41-p5-div-x16.c)
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800637
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700638SET(XNNPACK_AVX_MICROKERNEL_SRCS
Marat Dukhanfda12b82019-11-21 12:27:59 -0800639 src/f32-gemm/1x8-avx-broadcast.c
640 src/f32-gemm/4x8-avx-broadcast.c
641 src/f32-gemm/5x8-avx-broadcast.c
642 src/f32-gemm/6x8-avx-broadcast.c
643 src/f32-gemm/7x8-avx-broadcast.c
644 src/f32-gemminc/1x8-avx-broadcast.c
645 src/f32-gemminc/4x8-avx-broadcast.c
646 src/f32-gemminc/5x8-avx-broadcast.c
647 src/f32-gemminc/6x8-avx-broadcast.c
648 src/f32-gemminc/7x8-avx-broadcast.c
649 src/f32-igemm/1x8-avx-broadcast.c
650 src/f32-igemm/4x8-avx-broadcast.c
651 src/f32-igemm/5x8-avx-broadcast.c
652 src/f32-igemm/6x8-avx-broadcast.c
653 src/f32-igemm/7x8-avx-broadcast.c
Marat Dukhana5977df2019-10-30 22:02:15 -0700654 src/f32-rmax/avx.c
655 src/f32-vscale/avx-unroll32.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700656
Marat Dukhanfda12b82019-11-21 12:27:59 -0800657SET(XNNPACK_FMA3_MICROKERNEL_SRCS
658 src/f32-gemm/1x8-fma3-broadcast.c
659 src/f32-gemm/4x8-fma3-broadcast.c
660 src/f32-gemm/5x8-fma3-broadcast.c
661 src/f32-gemm/6x8-fma3-broadcast.c
662 src/f32-gemm/7x8-fma3-broadcast.c
663 src/f32-gemm/8x8-fma3-broadcast.c
664 src/f32-gemminc/1x8-fma3-broadcast.c
665 src/f32-gemminc/4x8-fma3-broadcast.c
666 src/f32-gemminc/5x8-fma3-broadcast.c
667 src/f32-gemminc/6x8-fma3-broadcast.c
668 src/f32-gemminc/7x8-fma3-broadcast.c
669 src/f32-gemminc/8x8-fma3-broadcast.c
670 src/f32-igemm/1x8-fma3-broadcast.c
671 src/f32-igemm/4x8-fma3-broadcast.c
672 src/f32-igemm/5x8-fma3-broadcast.c
673 src/f32-igemm/6x8-fma3-broadcast.c
674 src/f32-igemm/7x8-fma3-broadcast.c
675 src/f32-igemm/8x8-fma3-broadcast.c)
676
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700677SET(XNNPACK_AVX2_MICROKERNEL_SRCS
Marat Dukhan97579532019-10-18 16:40:39 -0700678 src/f32-raddexpminusmax/avx2-p5-unroll64.c
Marat Dukhan6f8d4d32019-10-25 17:07:09 -0700679 src/f32-raddextexp/avx2-p5-unroll64.c
Marat Dukhan97579532019-10-18 16:40:39 -0700680 src/f32-raddstoreexpminusmax/avx2-p5-unroll64.c
681 src/f32-vscaleexpminusmax/avx2-p5-unroll64.c
Marat Dukhan6f8d4d32019-10-25 17:07:09 -0700682 src/f32-vscaleextexp/avx2-p5-unroll64.c
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700683 src/math/exp-avx2-p5.c
684 src/math/exp-avx2-perm-p3.c
Marat Dukhan515c9772019-10-17 18:07:57 -0700685 src/math/exp-avx2-perm-p4.c
Marat Dukhan9d501d52019-11-16 02:30:50 -0800686 src/math/expminus-avx2-p5.c
687 src/math/extexp-avx2-p5.c)
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700688
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700689SET(XNNPACK_AVX512F_MICROKERNEL_SRCS
Marat Dukhan97579532019-10-18 16:40:39 -0700690 src/f32-raddexpminusmax/avx512f-p5-scalef-unroll128.c
Marat Dukhan6f8d4d32019-10-25 17:07:09 -0700691 src/f32-raddextexp/avx512f-p5-scalef-unroll128.c
Marat Dukhan97579532019-10-18 16:40:39 -0700692 src/f32-raddstoreexpminusmax/avx512f-p5-scalef-unroll128.c
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700693 src/f32-rmax/avx512f.c
Marat Dukhana5977df2019-10-30 22:02:15 -0700694 src/f32-vscale/avx512f-unroll64.c
Marat Dukhan97579532019-10-18 16:40:39 -0700695 src/f32-vscaleexpminusmax/avx512f-p5-scalef-unroll128.c
Marat Dukhan6f8d4d32019-10-25 17:07:09 -0700696 src/f32-vscaleextexp/avx512f-p5-scalef-unroll128.c
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700697 src/math/exp-avx512f-p5-scalef.c
698 src/math/exp-avx512f-p5.c
Marat Dukhanfeb49232019-10-28 11:03:31 -0700699 src/math/exp-avx512f-perm-p3.c
Marat Dukhan9d501d52019-11-16 02:30:50 -0800700 src/math/exp-avx512f-perm2-p2.c
701 src/math/extexp-avx512f-p5.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700702
703SET(XNNPACK_AARCH32_ASM_MICROKERNEL_SRCS
704 src/q8-dwconv/up8x9-aarch32-neon.S)
705
706SET(XNNPACK_AARCH64_ASM_MICROKERNEL_SRCS
707 src/f32-dwconv/up4x9-aarch64-neonfma-cortex-a55.S
708 src/f32-dwconv/up4x9-aarch64-neonfma.S
709 src/f32-gemm/1x12-aarch64-neonfma-cortex-a53.S
Frank Barchard21be34f2019-10-09 19:32:19 -0700710 src/f32-gemm/1x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700711 src/f32-gemm/1x8-aarch64-neonfma-cortex-a57.S
712 src/f32-gemm/1x8-aarch64-neonfma-cortex-a75.S
713 src/f32-gemm/4x12-aarch64-neonfma-cortex-a53.S
Marat Dukhan8e6e9972019-10-31 14:15:31 -0700714 src/f32-gemm/4x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700715 src/f32-gemm/4x8-aarch64-neonfma-cortex-a57.S
716 src/f32-gemm/4x8-aarch64-neonfma-cortex-a75.S
717 src/f32-gemm/4x8-aarch64-neonfma-ld128.S
718 src/f32-gemm/4x8-aarch64-neonfma-ld64.S
719 src/f32-gemm/5x8-aarch64-neonfma-cortex-a75.S
Marat Dukhan8e6e9972019-10-31 14:15:31 -0700720 src/f32-gemm/6x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700721 src/f32-gemm/6x8-aarch64-neonfma-cortex-a57.S
722 src/f32-gemm/6x8-aarch64-neonfma-cortex-a73.S
723 src/f32-gemm/6x8-aarch64-neonfma-cortex-a75.S
724 src/f32-gemm/6x8-aarch64-neonfma-ld128.S
725 src/f32-gemm/6x8-aarch64-neonfma-ld64.S
726 src/f32-gemminc/1x12-aarch64-neonfma-cortex-a53.S
Frank Barchard21be34f2019-10-09 19:32:19 -0700727 src/f32-gemminc/1x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700728 src/f32-gemminc/1x8-aarch64-neonfma-cortex-a57.S
729 src/f32-gemminc/1x8-aarch64-neonfma-cortex-a75.S
730 src/f32-gemminc/4x12-aarch64-neonfma-cortex-a53.S
Marat Dukhan8e6e9972019-10-31 14:15:31 -0700731 src/f32-gemminc/4x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700732 src/f32-gemminc/4x8-aarch64-neonfma-cortex-a57.S
733 src/f32-gemminc/4x8-aarch64-neonfma-cortex-a75.S
734 src/f32-gemminc/4x8-aarch64-neonfma-ld128.S
735 src/f32-gemminc/4x8-aarch64-neonfma-ld64.S
736 src/f32-gemminc/5x8-aarch64-neonfma-cortex-a75.S
Marat Dukhan8e6e9972019-10-31 14:15:31 -0700737 src/f32-gemminc/6x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700738 src/f32-gemminc/6x8-aarch64-neonfma-cortex-a57.S
739 src/f32-gemminc/6x8-aarch64-neonfma-cortex-a73.S
740 src/f32-gemminc/6x8-aarch64-neonfma-cortex-a75.S
741 src/f32-gemminc/6x8-aarch64-neonfma-ld128.S
742 src/f32-gemminc/6x8-aarch64-neonfma-ld64.S
743 src/f32-igemm/1x12-aarch64-neonfma-cortex-a53.S
Frank Barchard21be34f2019-10-09 19:32:19 -0700744 src/f32-igemm/1x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700745 src/f32-igemm/1x8-aarch64-neonfma-cortex-a57.S
746 src/f32-igemm/1x8-aarch64-neonfma-cortex-a75.S
747 src/f32-igemm/4x12-aarch64-neonfma-cortex-a53.S
748 src/f32-igemm/4x8-aarch64-neonfma-cortex-a75.S
749 src/f32-igemm/5x8-aarch64-neonfma-cortex-a75.S
Marat Dukhan8e6e9972019-10-31 14:15:31 -0700750 src/f32-igemm/6x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700751 src/f32-igemm/6x8-aarch64-neonfma-cortex-a57.S
752 src/f32-igemm/6x8-aarch64-neonfma-cortex-a73.S
753 src/f32-igemm/6x8-aarch64-neonfma-cortex-a75.S)
754
755SET(XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SCALAR_MICROKERNEL_SRCS})
756IF(NOT CMAKE_SYSTEM_NAME STREQUAL "Emscripten")
757 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_PSIMD_MICROKERNEL_SRCS})
758ENDIF()
759IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^armv[5-8]" OR IOS_ARCH MATCHES "^armv7")
760 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEON_MICROKERNEL_SRCS})
761 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEONFMA_MICROKERNEL_SRCS})
762 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AARCH32_ASM_MICROKERNEL_SRCS})
763ENDIF()
764IF(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64" OR IOS_ARCH MATCHES "^arm64.*")
765 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEON_MICROKERNEL_SRCS})
766 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEONFMA_MICROKERNEL_SRCS})
767 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${AARCH64_XNNPACK_NEONFMA_MICROKERNEL_SRCS})
768 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AARCH64_NEONFP16ARITH_MICROKERNEL_SRCS})
769 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AARCH64_ASM_MICROKERNEL_SRCS})
770ENDIF()
771IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^(i[3-6]86|x86_64)$" OR IOS_ARCH MATCHES "^(i386|x86_64)$")
772 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SSE_MICROKERNEL_SRCS})
773 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SSE2_MICROKERNEL_SRCS})
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800774 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SSE41_MICROKERNEL_SRCS})
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700775 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AVX_MICROKERNEL_SRCS})
Marat Dukhanfda12b82019-11-21 12:27:59 -0800776 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_FMA3_MICROKERNEL_SRCS})
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700777 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AVX2_MICROKERNEL_SRCS})
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700778 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AVX512F_MICROKERNEL_SRCS})
779ENDIF()
780
781IF(XNNPACK_LIBRARY_TYPE STREQUAL "default")
782 ADD_LIBRARY(XNNPACK ${XNNPACK_COLD_SRCS} ${XNNPACK_HOT_SRCS} ${XNNPACK_MICROKERNEL_SRCS})
783ELSEIF(XNNPACK_LIBRARY_TYPE STREQUAL "shared")
784 ADD_LIBRARY(XNNPACK SHARED ${XNNPACK_COLD_SRCS} ${XNNPACK_HOT_SRCS} ${XNNPACK_MICROKERNEL_SRCS})
785ELSEIF(XNNPACK_LIBRARY_TYPE STREQUAL "static")
786 ADD_LIBRARY(XNNPACK STATIC ${XNNPACK_COLD_SRCS} ${XNNPACK_HOT_SRCS} ${XNNPACK_MICROKERNEL_SRCS})
787ELSE()
788 MESSAGE(FATAL_ERROR "Unsupported XNNPACK library type \"${XNNPACK_LIBRARY_TYPE}\". Must be \"static\", \"shared\", or \"default\"")
789ENDIF()
790SET_TARGET_PROPERTIES(XNNPACK PROPERTIES
791 C_STANDARD 99
792 C_EXTENSIONS YES)
793IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^armv[5-8]" OR IOS_ARCH MATCHES "^armv7")
794 SET_PROPERTY(SOURCE ${XNNPACK_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -marm ")
795 SET_PROPERTY(SOURCE ${XNNPACK_PSIMD_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfpu=neon ")
796 SET_PROPERTY(SOURCE ${XNNPACK_NEON_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfpu=neon ")
797 SET_PROPERTY(SOURCE ${XNNPACK_NEONFMA_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfpu=neon-vfpv4 ")
798 IF(IOS)
799 SET_PROPERTY(SOURCE ${XNNPACK_AARCH32_ASM_UKERNELS} APPEND_STRING PROPERTY COMPILE_FLAGS " -arch ${IOS_ARCH} ")
800 ENDIF()
801ENDIF()
802IF(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64" OR IOS_ARCH MATCHES "^arm64.*")
803 SET_PROPERTY(SOURCE ${XNNPACK_AARCH64_NEONFP16ARITH_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -march=armv8.2-a+fp16 ")
804 IF(IOS)
805 SET_PROPERTY(SOURCE ${XNNPACK_AARCH64_ASM_UKERNELS} APPEND_STRING PROPERTY COMPILE_FLAGS " -arch ${IOS_ARCH} ")
806 ENDIF()
807ENDIF()
808IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^(i[3-6]86|x86_64)$" OR IOS_ARCH MATCHES "^(i386|x86_64)$")
809 SET_PROPERTY(SOURCE ${XNNPACK_SSE_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -msse ")
810 SET_PROPERTY(SOURCE ${XNNPACK_SSE2_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -msse2 ")
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800811 SET_PROPERTY(SOURCE ${XNNPACK_SSE41_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -msse4.1 ")
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700812 SET_PROPERTY(SOURCE ${XNNPACK_AVX_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mavx ")
Marat Dukhanfda12b82019-11-21 12:27:59 -0800813 SET_PROPERTY(SOURCE ${XNNPACK_FMA3_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfma ")
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700814 SET_PROPERTY(SOURCE ${XNNPACK_AVX2_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfma -mavx2 ")
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700815 SET_PROPERTY(SOURCE ${XNNPACK_AVX512F_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mavx512f ")
816ENDIF()
817IF(CMAKE_BUILD_TYPE STREQUAL "Debug")
818 TARGET_COMPILE_DEFINITIONS(XNNPACK PRIVATE XNN_LOG_LEVEL=5)
819ELSE()
820 TARGET_COMPILE_DEFINITIONS(XNNPACK PRIVATE XNN_LOG_LEVEL=0)
821 SET_PROPERTY(SOURCE ${XNNPACK_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -O2 ")
822 SET_PROPERTY(SOURCE ${XNNPACK_HOT_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -O2 ")
823 SET_PROPERTY(SOURCE ${XNNPACK_COLD_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -Os ")
824ENDIF()
825IF(NOT CMAKE_SYSTEM_NAME STREQUAL "Emscripten")
826 SET_PROPERTY(SOURCE ${XNNPACK_PSIMD_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -O3 -ffast-math ")
827ENDIF()
828
829TARGET_INCLUDE_DIRECTORIES(XNNPACK PUBLIC include)
830TARGET_INCLUDE_DIRECTORIES(XNNPACK PRIVATE src)
831SET_TARGET_PROPERTIES(XNNPACK PROPERTIES PUBLIC_HEADER include/XNNPACK.h)
832
833# ---[ Configure clog
834IF(NOT TARGET clog)
835 SET(CLOG_BUILD_TESTS OFF CACHE BOOL "")
836 SET(CLOG_RUNTIME_TYPE "${CPUINFO_RUNTIME_TYPE}" CACHE STRING "")
837 ADD_SUBDIRECTORY(
838 "${CLOG_SOURCE_DIR}/deps/clog"
839 "${CMAKE_BINARY_DIR}/clog")
840 # We build static version of clog but a dynamic library may indirectly depend on it
841 SET_PROPERTY(TARGET clog PROPERTY POSITION_INDEPENDENT_CODE ON)
842ENDIF()
843TARGET_LINK_LIBRARIES(XNNPACK PRIVATE clog)
844
845# ---[ Configure cpuinfo
846IF(NOT TARGET cpuinfo)
847 SET(CPUINFO_BUILD_TOOLS OFF CACHE BOOL "")
848 SET(CPUINFO_BUILD_UNIT_TESTS OFF CACHE BOOL "")
849 SET(CPUINFO_BUILD_MOCK_TESTS OFF CACHE BOOL "")
850 SET(CPUINFO_BUILD_BENCHMARKS OFF CACHE BOOL "")
851 ADD_SUBDIRECTORY(
852 "${CPUINFO_SOURCE_DIR}"
853 "${CMAKE_BINARY_DIR}/cpuinfo")
854ENDIF()
855TARGET_LINK_LIBRARIES(XNNPACK PRIVATE cpuinfo)
856
857# ---[ Configure pthreadpool
858IF(NOT TARGET pthreadpool)
859 SET(PTHREADPOOL_BUILD_TESTS OFF CACHE BOOL "")
860 SET(PTHREADPOOL_BUILD_BENCHMARKS OFF CACHE BOOL "")
Marat Dukhand70028a2019-10-29 09:53:16 -0700861 SET(PTHREADPOOL_ALLOW_DEPRECATED_API OFF CACHE BOOL "")
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700862 ADD_SUBDIRECTORY(
863 "${PTHREADPOOL_SOURCE_DIR}"
864 "${CMAKE_BINARY_DIR}/pthreadpool")
865ENDIF()
866TARGET_LINK_LIBRARIES(XNNPACK PUBLIC pthreadpool)
867
868# ---[ Configure FXdiv
869IF(NOT TARGET fxdiv)
870 SET(FXDIV_BUILD_TESTS OFF CACHE BOOL "")
871 SET(FXDIV_BUILD_BENCHMARKS OFF CACHE BOOL "")
872 ADD_SUBDIRECTORY(
873 "${FXDIV_SOURCE_DIR}"
874 "${CMAKE_BINARY_DIR}/FXdiv")
875ENDIF()
876TARGET_LINK_LIBRARIES(XNNPACK PRIVATE fxdiv)
877
878# ---[ Configure psimd
879IF(NOT TARGET psimd)
880 ADD_SUBDIRECTORY(
881 "${PSIMD_SOURCE_DIR}"
882 "${CMAKE_BINARY_DIR}/psimd")
883ENDIF()
884TARGET_LINK_LIBRARIES(XNNPACK PRIVATE psimd)
885
886# ---[ Configure FP16
887IF(NOT TARGET fp16)
888 SET(FP16_BUILD_TESTS OFF CACHE BOOL "")
889 SET(FP16_BUILD_BENCHMARKS OFF CACHE BOOL "")
890 ADD_SUBDIRECTORY(
891 "${FP16_SOURCE_DIR}"
892 "${CMAKE_BINARY_DIR}/FP16")
893ENDIF()
894TARGET_LINK_LIBRARIES(XNNPACK PRIVATE fp16)
895
896INSTALL(TARGETS XNNPACK
897 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
898 ARCHIVE DESTINATION ${CMAKE_INSTALL_LIBDIR}
899 PUBLIC_HEADER DESTINATION ${CMAKE_INSTALL_INCLUDEDIR})
900
901# ---[ XNNPACK unit tests
902IF(XNNPACK_BUILD_TESTS)
903 # ---[ Build google test
904 IF(NOT TARGET gtest)
905 SET(gtest_force_shared_crt ON CACHE BOOL "" FORCE)
906 ADD_SUBDIRECTORY(
907 "${GOOGLETEST_SOURCE_DIR}"
908 "${CMAKE_BINARY_DIR}/googletest")
909 ENDIF()
910
911 # ---[ Build operator-level unit tests
Marat Dukhanefc47b82019-11-18 09:25:38 -0800912 ADD_EXECUTABLE(add-nc-test test/add-nc.cc)
913 SET_TARGET_PROPERTIES(add-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700914 CXX_STANDARD 11
915 CXX_STANDARD_REQUIRED YES
916 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -0800917 TARGET_INCLUDE_DIRECTORIES(add-nc-test PRIVATE src test)
918 TARGET_LINK_LIBRARIES(add-nc-test PRIVATE XNNPACK gtest gtest_main)
919 ADD_TEST(add-nc-test add-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700920
Marat Dukhanefc47b82019-11-18 09:25:38 -0800921 ADD_EXECUTABLE(argmax-pooling-nhwc-test test/argmax-pooling-nhwc.cc)
922 SET_TARGET_PROPERTIES(argmax-pooling-nhwc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700923 CXX_STANDARD 11
924 CXX_STANDARD_REQUIRED YES
925 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -0800926 TARGET_INCLUDE_DIRECTORIES(argmax-pooling-nhwc-test PRIVATE src test)
927 TARGET_LINK_LIBRARIES(argmax-pooling-nhwc-test PRIVATE XNNPACK gtest gtest_main)
928 ADD_TEST(argmax-pooling-nhwc-test argmax-pooling-nhwc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700929
Marat Dukhanefc47b82019-11-18 09:25:38 -0800930 ADD_EXECUTABLE(average-pooling-nhwc-test test/average-pooling-nhwc.cc)
931 SET_TARGET_PROPERTIES(average-pooling-nhwc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700932 CXX_STANDARD 11
933 CXX_STANDARD_REQUIRED YES
934 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -0800935 TARGET_INCLUDE_DIRECTORIES(average-pooling-nhwc-test PRIVATE src test)
936 TARGET_LINK_LIBRARIES(average-pooling-nhwc-test PRIVATE XNNPACK gtest gtest_main)
937 ADD_TEST(average-pooling-nhwc-test average-pooling-nhwc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700938
Marat Dukhanefc47b82019-11-18 09:25:38 -0800939 ADD_EXECUTABLE(channel-pad-nc-test test/channel-pad-nc.cc)
940 SET_TARGET_PROPERTIES(channel-pad-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700941 CXX_STANDARD 11
942 CXX_STANDARD_REQUIRED YES
943 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -0800944 TARGET_INCLUDE_DIRECTORIES(channel-pad-nc-test PRIVATE src test)
945 TARGET_LINK_LIBRARIES(channel-pad-nc-test PRIVATE XNNPACK gtest gtest_main)
946 ADD_TEST(channel-pad-nc-test channel-pad-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700947
Marat Dukhanefc47b82019-11-18 09:25:38 -0800948 ADD_EXECUTABLE(channel-shuffle-nc-test test/channel-shuffle-nc.cc)
949 SET_TARGET_PROPERTIES(channel-shuffle-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700950 CXX_STANDARD 11
951 CXX_STANDARD_REQUIRED YES
952 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -0800953 TARGET_INCLUDE_DIRECTORIES(channel-shuffle-nc-test PRIVATE src test)
954 TARGET_LINK_LIBRARIES(channel-shuffle-nc-test PRIVATE XNNPACK gtest gtest_main)
955 ADD_TEST(channel-shuffle-nc-test channel-shuffle-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700956
Marat Dukhanefc47b82019-11-18 09:25:38 -0800957 ADD_EXECUTABLE(clamp-nc-test test/clamp-nc.cc)
958 SET_TARGET_PROPERTIES(clamp-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700959 CXX_STANDARD 11
960 CXX_STANDARD_REQUIRED YES
961 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -0800962 TARGET_INCLUDE_DIRECTORIES(clamp-nc-test PRIVATE src test)
963 TARGET_LINK_LIBRARIES(clamp-nc-test PRIVATE XNNPACK gtest gtest_main)
964 ADD_TEST(clamp-nc-test clamp-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700965
Marat Dukhanefc47b82019-11-18 09:25:38 -0800966 ADD_EXECUTABLE(convolution-nhwc-test test/convolution-nhwc.cc)
967 SET_TARGET_PROPERTIES(convolution-nhwc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700968 CXX_STANDARD 11
969 CXX_STANDARD_REQUIRED YES
970 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -0800971 TARGET_INCLUDE_DIRECTORIES(convolution-nhwc-test PRIVATE src test)
972 TARGET_LINK_LIBRARIES(convolution-nhwc-test PRIVATE XNNPACK gtest gtest_main)
973 ADD_TEST(convolution-nhwc-test convolution-nhwc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700974
Marat Dukhanefc47b82019-11-18 09:25:38 -0800975 ADD_EXECUTABLE(convolution-nchw-test test/convolution-nchw.cc)
976 SET_TARGET_PROPERTIES(convolution-nchw-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700977 CXX_STANDARD 11
978 CXX_STANDARD_REQUIRED YES
979 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -0800980 TARGET_INCLUDE_DIRECTORIES(convolution-nchw-test PRIVATE src test)
981 TARGET_LINK_LIBRARIES(convolution-nchw-test PRIVATE XNNPACK gtest gtest_main)
982 ADD_TEST(convolution-nchw-test convolution-nchw-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700983
Marat Dukhanefc47b82019-11-18 09:25:38 -0800984 ADD_EXECUTABLE(deconvolution-nhwc-test test/deconvolution-nhwc.cc)
985 SET_TARGET_PROPERTIES(deconvolution-nhwc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700986 CXX_STANDARD 11
987 CXX_STANDARD_REQUIRED YES
988 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -0800989 TARGET_INCLUDE_DIRECTORIES(deconvolution-nhwc-test PRIVATE src test)
990 TARGET_LINK_LIBRARIES(deconvolution-nhwc-test PRIVATE XNNPACK gtest gtest_main)
991 ADD_TEST(deconvolution-nhwc-test deconvolution-nhwc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700992
Marat Dukhanefc47b82019-11-18 09:25:38 -0800993 ADD_EXECUTABLE(fully-connected-nc-test test/fully-connected-nc.cc)
994 SET_TARGET_PROPERTIES(fully-connected-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700995 CXX_STANDARD 11
996 CXX_STANDARD_REQUIRED YES
997 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -0800998 TARGET_INCLUDE_DIRECTORIES(fully-connected-nc-test PRIVATE src test)
999 TARGET_LINK_LIBRARIES(fully-connected-nc-test PRIVATE XNNPACK gtest gtest_main)
1000 ADD_TEST(fully-connected-nc-test fully-connected-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001001
Marat Dukhanefc47b82019-11-18 09:25:38 -08001002 ADD_EXECUTABLE(global-average-pooling-nwc-test test/global-average-pooling-nwc.cc)
1003 SET_TARGET_PROPERTIES(global-average-pooling-nwc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001004 CXX_STANDARD 11
1005 CXX_STANDARD_REQUIRED YES
1006 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001007 TARGET_INCLUDE_DIRECTORIES(global-average-pooling-nwc-test PRIVATE src test)
1008 TARGET_LINK_LIBRARIES(global-average-pooling-nwc-test PRIVATE XNNPACK gtest gtest_main)
1009 ADD_TEST(global-average-pooling-nwc-test global-average-pooling-nwc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001010
Marat Dukhanefc47b82019-11-18 09:25:38 -08001011 ADD_EXECUTABLE(global-average-pooling-ncw-test test/global-average-pooling-ncw.cc)
1012 SET_TARGET_PROPERTIES(global-average-pooling-ncw-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001013 CXX_STANDARD 11
1014 CXX_STANDARD_REQUIRED YES
1015 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001016 TARGET_INCLUDE_DIRECTORIES(global-average-pooling-ncw-test PRIVATE src test)
1017 TARGET_LINK_LIBRARIES(global-average-pooling-ncw-test PRIVATE XNNPACK gtest gtest_main)
1018 ADD_TEST(global-average-pooling-ncw-test global-average-pooling-ncw-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001019
Marat Dukhanefc47b82019-11-18 09:25:38 -08001020 ADD_EXECUTABLE(hardswish-nc-test test/hardswish-nc.cc)
1021 SET_TARGET_PROPERTIES(hardswish-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001022 CXX_STANDARD 11
1023 CXX_STANDARD_REQUIRED YES
1024 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001025 TARGET_INCLUDE_DIRECTORIES(hardswish-nc-test PRIVATE src test)
1026 TARGET_LINK_LIBRARIES(hardswish-nc-test PRIVATE XNNPACK gtest gtest_main)
1027 ADD_TEST(hardswish-nc-test hardswish-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001028
Marat Dukhanefc47b82019-11-18 09:25:38 -08001029 ADD_EXECUTABLE(leaky-relu-nc-test test/leaky-relu-nc.cc)
1030 SET_TARGET_PROPERTIES(leaky-relu-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001031 CXX_STANDARD 11
1032 CXX_STANDARD_REQUIRED YES
1033 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001034 TARGET_INCLUDE_DIRECTORIES(leaky-relu-nc-test PRIVATE src test)
1035 TARGET_LINK_LIBRARIES(leaky-relu-nc-test PRIVATE XNNPACK gtest gtest_main)
1036 ADD_TEST(leaky-relu-nc-test leaky-relu-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001037
Marat Dukhanefc47b82019-11-18 09:25:38 -08001038 ADD_EXECUTABLE(max-pooling-nhwc-test test/max-pooling-nhwc.cc)
1039 SET_TARGET_PROPERTIES(max-pooling-nhwc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001040 CXX_STANDARD 11
1041 CXX_STANDARD_REQUIRED YES
1042 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001043 TARGET_INCLUDE_DIRECTORIES(max-pooling-nhwc-test PRIVATE src test)
1044 TARGET_LINK_LIBRARIES(max-pooling-nhwc-test PRIVATE XNNPACK gtest gtest_main)
1045 ADD_TEST(max-pooling-nhwc-test max-pooling-nhwc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001046
Marat Dukhanefc47b82019-11-18 09:25:38 -08001047 ADD_EXECUTABLE(multiply-nd-test test/multiply-nd.cc)
1048 SET_TARGET_PROPERTIES(multiply-nd-test PROPERTIES
Marat Dukhan9d501d52019-11-16 02:30:50 -08001049 CXX_STANDARD 11
1050 CXX_STANDARD_REQUIRED YES
1051 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001052 TARGET_INCLUDE_DIRECTORIES(multiply-nd-test PRIVATE src test)
1053 TARGET_LINK_LIBRARIES(multiply-nd-test PRIVATE XNNPACK gtest gtest_main)
1054 ADD_TEST(multiply-nd-test multiply-nd-test)
Marat Dukhan9d501d52019-11-16 02:30:50 -08001055
Marat Dukhanefc47b82019-11-18 09:25:38 -08001056 ADD_EXECUTABLE(prelu-nc-test test/prelu-nc.cc)
1057 SET_TARGET_PROPERTIES(prelu-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001058 CXX_STANDARD 11
1059 CXX_STANDARD_REQUIRED YES
1060 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001061 TARGET_INCLUDE_DIRECTORIES(prelu-nc-test PRIVATE src test)
1062 TARGET_LINK_LIBRARIES(prelu-nc-test PRIVATE XNNPACK gtest gtest_main)
1063 ADD_TEST(prelu-nc-test prelu-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001064
Marat Dukhanefc47b82019-11-18 09:25:38 -08001065 ADD_EXECUTABLE(resize-bilinear-nhwc-test test/resize-bilinear-nhwc.cc)
1066 SET_TARGET_PROPERTIES(resize-bilinear-nhwc-test PROPERTIES
Marat Dukhan69722492019-11-11 19:55:50 -08001067 CXX_STANDARD 11
1068 CXX_STANDARD_REQUIRED YES
1069 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001070 TARGET_INCLUDE_DIRECTORIES(resize-bilinear-nhwc-test PRIVATE src test)
1071 TARGET_LINK_LIBRARIES(resize-bilinear-nhwc-test PRIVATE XNNPACK gtest gtest_main)
1072 ADD_TEST(resize-bilinear-nhwc-test resize-bilinear-nhwc-test)
Marat Dukhan69722492019-11-11 19:55:50 -08001073
Marat Dukhanefc47b82019-11-18 09:25:38 -08001074 ADD_EXECUTABLE(sigmoid-nc-test test/sigmoid-nc.cc)
1075 SET_TARGET_PROPERTIES(sigmoid-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001076 CXX_STANDARD 11
1077 CXX_STANDARD_REQUIRED YES
1078 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001079 TARGET_INCLUDE_DIRECTORIES(sigmoid-nc-test PRIVATE src test)
1080 TARGET_LINK_LIBRARIES(sigmoid-nc-test PRIVATE XNNPACK gtest gtest_main)
1081 ADD_TEST(sigmoid-nc-test sigmoid-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001082
Marat Dukhanefc47b82019-11-18 09:25:38 -08001083 ADD_EXECUTABLE(softargmax-nc-test test/softargmax-nc.cc)
1084 SET_TARGET_PROPERTIES(softargmax-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001085 CXX_STANDARD 11
1086 CXX_STANDARD_REQUIRED YES
1087 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001088 TARGET_INCLUDE_DIRECTORIES(softargmax-nc-test PRIVATE src test)
1089 TARGET_LINK_LIBRARIES(softargmax-nc-test PRIVATE XNNPACK gtest gtest_main)
1090 ADD_TEST(softargmax-nc-test softargmax-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001091
Marat Dukhanefc47b82019-11-18 09:25:38 -08001092 ADD_EXECUTABLE(unpooling-nhwc-test test/unpooling-nhwc.cc)
1093 SET_TARGET_PROPERTIES(unpooling-nhwc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001094 CXX_STANDARD 11
1095 CXX_STANDARD_REQUIRED YES
1096 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001097 TARGET_INCLUDE_DIRECTORIES(unpooling-nhwc-test PRIVATE src test)
1098 TARGET_LINK_LIBRARIES(unpooling-nhwc-test PRIVATE XNNPACK gtest gtest_main)
1099 ADD_TEST(unpooling-nhwc-test unpooling-nhwc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001100
1101 # ---[ Build microkernel-level unit tests
1102 ADD_EXECUTABLE(f16-gemm-test test/f16-gemm.cc)
1103 SET_TARGET_PROPERTIES(f16-gemm-test PROPERTIES
1104 CXX_STANDARD 11
1105 CXX_STANDARD_REQUIRED YES
1106 CXX_EXTENSIONS YES)
1107 TARGET_INCLUDE_DIRECTORIES(f16-gemm-test PRIVATE src test)
1108 TARGET_LINK_LIBRARIES(f16-gemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1109 ADD_TEST(f16-gemm-test f16-gemm-test)
1110
1111 ADD_EXECUTABLE(f32-argmaxpool-test test/f32-argmaxpool.cc)
1112 SET_TARGET_PROPERTIES(f32-argmaxpool-test PROPERTIES
1113 CXX_STANDARD 11
1114 CXX_STANDARD_REQUIRED YES
1115 CXX_EXTENSIONS YES)
1116 TARGET_INCLUDE_DIRECTORIES(f32-argmaxpool-test PRIVATE src test)
1117 TARGET_LINK_LIBRARIES(f32-argmaxpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1118 ADD_TEST(f32-argmaxpool-test f32-argmaxpool-test)
1119
1120 ADD_EXECUTABLE(f32-avgpool-test test/f32-avgpool.cc)
1121 SET_TARGET_PROPERTIES(f32-avgpool-test PROPERTIES
1122 CXX_STANDARD 11
1123 CXX_STANDARD_REQUIRED YES
1124 CXX_EXTENSIONS YES)
1125 TARGET_INCLUDE_DIRECTORIES(f32-avgpool-test PRIVATE src test)
1126 TARGET_LINK_LIBRARIES(f32-avgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1127 ADD_TEST(f32-avgpool-test f32-avgpool-test)
1128
Marat Dukhanc465fc22019-11-07 19:20:28 -08001129 ADD_EXECUTABLE(f32-bilinear-test test/f32-bilinear.cc)
1130 SET_TARGET_PROPERTIES(f32-bilinear-test PROPERTIES
1131 CXX_STANDARD 11
1132 CXX_STANDARD_REQUIRED YES
1133 CXX_EXTENSIONS YES)
1134 TARGET_INCLUDE_DIRECTORIES(f32-bilinear-test PRIVATE src test)
1135 TARGET_LINK_LIBRARIES(f32-bilinear-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1136 ADD_TEST(f32-bilinear-test f32-bilinear-test)
1137
Marat Dukhanefc47b82019-11-18 09:25:38 -08001138 ADD_EXECUTABLE(f32-clamp-nc-test test/f32-clamp.cc)
1139 SET_TARGET_PROPERTIES(f32-clamp-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001140 CXX_STANDARD 11
1141 CXX_STANDARD_REQUIRED YES
1142 CXX_EXTENSIONS YES)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001143 TARGET_INCLUDE_DIRECTORIES(f32-clamp-nc-test PRIVATE src test)
1144 TARGET_LINK_LIBRARIES(f32-clamp-nc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1145 ADD_TEST(f32-clamp-nc-test f32-clamp-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001146
1147 ADD_EXECUTABLE(f32-conv-hwc-test test/f32-conv-hwc.cc)
1148 SET_TARGET_PROPERTIES(f32-conv-hwc-test PROPERTIES
1149 CXX_STANDARD 11
1150 CXX_STANDARD_REQUIRED YES
1151 CXX_EXTENSIONS YES)
1152 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc-test PRIVATE src test)
1153 TARGET_LINK_LIBRARIES(f32-conv-hwc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1154 ADD_TEST(f32-conv-hwc-test f32-conv-hwc-test)
1155
1156 ADD_EXECUTABLE(f32-conv-hwc2spchw-test test/f32-conv-hwc2spchw.cc)
1157 SET_TARGET_PROPERTIES(f32-conv-hwc2spchw-test PROPERTIES
1158 CXX_STANDARD 11
1159 CXX_STANDARD_REQUIRED YES
1160 CXX_EXTENSIONS YES)
1161 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc2spchw-test PRIVATE src test)
1162 TARGET_LINK_LIBRARIES(f32-conv-hwc2spchw-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1163 ADD_TEST(f32-conv-hwc2spchw-test f32-conv-hwc2spchw-test)
1164
1165 ADD_EXECUTABLE(f32-dwconv-spchw-test test/f32-dwconv-spchw.cc)
1166 SET_TARGET_PROPERTIES(f32-dwconv-spchw-test PROPERTIES
1167 CXX_STANDARD 11
1168 CXX_STANDARD_REQUIRED YES
1169 CXX_EXTENSIONS YES)
1170 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-spchw-test PRIVATE src test)
1171 TARGET_LINK_LIBRARIES(f32-dwconv-spchw-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1172 ADD_TEST(f32-dwconv-spchw-test f32-dwconv-spchw-test)
1173
1174 ADD_EXECUTABLE(f32-dwconv-test test/f32-dwconv.cc)
1175 SET_TARGET_PROPERTIES(f32-dwconv-test PROPERTIES
1176 CXX_STANDARD 11
1177 CXX_STANDARD_REQUIRED YES
1178 CXX_EXTENSIONS YES)
1179 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-test PRIVATE src test)
1180 TARGET_LINK_LIBRARIES(f32-dwconv-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1181 ADD_TEST(f32-dwconv-test f32-dwconv-test)
1182
1183 ADD_EXECUTABLE(f32-gavgpool-spchw-test test/f32-gavgpool-spchw.cc)
1184 SET_TARGET_PROPERTIES(f32-gavgpool-spchw-test PROPERTIES
1185 CXX_STANDARD 11
1186 CXX_STANDARD_REQUIRED YES
1187 CXX_EXTENSIONS YES)
1188 TARGET_INCLUDE_DIRECTORIES(f32-gavgpool-spchw-test PRIVATE src test)
1189 TARGET_LINK_LIBRARIES(f32-gavgpool-spchw-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1190 ADD_TEST(f32-gavgpool-spchw-test f32-gavgpool-spchw-test)
1191
1192 ADD_EXECUTABLE(f32-gavgpool-test test/f32-gavgpool.cc)
1193 SET_TARGET_PROPERTIES(f32-gavgpool-test PROPERTIES
1194 CXX_STANDARD 11
1195 CXX_STANDARD_REQUIRED YES
1196 CXX_EXTENSIONS YES)
1197 TARGET_INCLUDE_DIRECTORIES(f32-gavgpool-test PRIVATE src test)
1198 TARGET_LINK_LIBRARIES(f32-gavgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1199 ADD_TEST(f32-gavgpool-test f32-gavgpool-test)
1200
1201 ADD_EXECUTABLE(f32-gemm-test test/f32-gemm.cc)
1202 SET_TARGET_PROPERTIES(f32-gemm-test PROPERTIES
1203 CXX_STANDARD 11
1204 CXX_STANDARD_REQUIRED YES
1205 CXX_EXTENSIONS YES)
1206 TARGET_INCLUDE_DIRECTORIES(f32-gemm-test PRIVATE src test)
1207 TARGET_LINK_LIBRARIES(f32-gemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1208 ADD_TEST(f32-gemm-test f32-gemm-test)
1209
1210 ADD_EXECUTABLE(f32-gemminc-test test/f32-gemminc.cc)
1211 SET_TARGET_PROPERTIES(f32-gemminc-test PROPERTIES
1212 CXX_STANDARD 11
1213 CXX_STANDARD_REQUIRED YES
1214 CXX_EXTENSIONS YES)
1215 TARGET_INCLUDE_DIRECTORIES(f32-gemminc-test PRIVATE src test)
1216 TARGET_LINK_LIBRARIES(f32-gemminc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1217 ADD_TEST(f32-gemminc-test f32-gemminc-test)
1218
1219 ADD_EXECUTABLE(f32-hswish-test test/f32-hswish.cc)
1220 SET_TARGET_PROPERTIES(f32-hswish-test PROPERTIES
1221 CXX_STANDARD 11
1222 CXX_STANDARD_REQUIRED YES
1223 CXX_EXTENSIONS YES)
1224 TARGET_INCLUDE_DIRECTORIES(f32-hswish-test PRIVATE src test)
1225 TARGET_LINK_LIBRARIES(f32-hswish-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1226 ADD_TEST(f32-hswish-test f32-hswish-test)
1227
1228 ADD_EXECUTABLE(f32-igemm-test test/f32-igemm.cc)
1229 SET_TARGET_PROPERTIES(f32-igemm-test PROPERTIES
1230 CXX_STANDARD 11
1231 CXX_STANDARD_REQUIRED YES
1232 CXX_EXTENSIONS YES)
1233 TARGET_INCLUDE_DIRECTORIES(f32-igemm-test PRIVATE src test)
1234 TARGET_LINK_LIBRARIES(f32-igemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1235 ADD_TEST(f32-igemm-test f32-igemm-test)
1236
1237 ADD_EXECUTABLE(f32-maxpool-test test/f32-maxpool.cc)
1238 SET_TARGET_PROPERTIES(f32-maxpool-test PROPERTIES
1239 CXX_STANDARD 11
1240 CXX_STANDARD_REQUIRED YES
1241 CXX_EXTENSIONS YES)
1242 TARGET_INCLUDE_DIRECTORIES(f32-maxpool-test PRIVATE src test)
1243 TARGET_LINK_LIBRARIES(f32-maxpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1244 ADD_TEST(f32-maxpool-test f32-maxpool-test)
1245
1246 ADD_EXECUTABLE(f32-pavgpool-test test/f32-pavgpool.cc)
1247 SET_TARGET_PROPERTIES(f32-pavgpool-test PROPERTIES
1248 CXX_STANDARD 11
1249 CXX_STANDARD_REQUIRED YES
1250 CXX_EXTENSIONS YES)
1251 TARGET_INCLUDE_DIRECTORIES(f32-pavgpool-test PRIVATE src test)
1252 TARGET_LINK_LIBRARIES(f32-pavgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1253 ADD_TEST(f32-pavgpool-test f32-pavgpool-test)
1254
1255 ADD_EXECUTABLE(f32-ppmm-test test/f32-ppmm.cc)
1256 SET_TARGET_PROPERTIES(f32-ppmm-test PROPERTIES
1257 CXX_STANDARD 11
1258 CXX_STANDARD_REQUIRED YES
1259 CXX_EXTENSIONS YES)
1260 TARGET_INCLUDE_DIRECTORIES(f32-ppmm-test PRIVATE src test)
1261 TARGET_LINK_LIBRARIES(f32-ppmm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1262 ADD_TEST(f32-ppmm-test f32-ppmm-test)
1263
Marat Dukhanefc47b82019-11-18 09:25:38 -08001264 ADD_EXECUTABLE(f32-prelu-nc-test test/f32-prelu.cc)
1265 SET_TARGET_PROPERTIES(f32-prelu-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001266 CXX_STANDARD 11
1267 CXX_STANDARD_REQUIRED YES
1268 CXX_EXTENSIONS YES)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001269 TARGET_INCLUDE_DIRECTORIES(f32-prelu-nc-test PRIVATE src test)
1270 TARGET_LINK_LIBRARIES(f32-prelu-nc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1271 ADD_TEST(f32-prelu-nc-test f32-prelu-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001272
Marat Dukhan97579532019-10-18 16:40:39 -07001273 ADD_EXECUTABLE(f32-raddexpminusmax-test test/f32-raddexpminusmax.cc)
1274 SET_TARGET_PROPERTIES(f32-raddexpminusmax-test PROPERTIES
1275 CXX_STANDARD 11
1276 CXX_STANDARD_REQUIRED YES
1277 CXX_EXTENSIONS YES)
1278 TARGET_INCLUDE_DIRECTORIES(f32-raddexpminusmax-test PRIVATE src test)
1279 TARGET_LINK_LIBRARIES(f32-raddexpminusmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1280 ADD_TEST(f32-raddexpminusmax-test f32-raddexpminusmax-test)
1281
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001282 ADD_EXECUTABLE(f32-raddextexp-test test/f32-raddextexp.cc)
1283 SET_TARGET_PROPERTIES(f32-raddextexp-test PROPERTIES
1284 CXX_STANDARD 11
1285 CXX_STANDARD_REQUIRED YES
1286 CXX_EXTENSIONS YES)
1287 TARGET_INCLUDE_DIRECTORIES(f32-raddextexp-test PRIVATE src test)
Marat Dukhana5977df2019-10-30 22:02:15 -07001288 TARGET_LINK_LIBRARIES(f32-raddextexp-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001289 ADD_TEST(f32-raddextexp-test f32-raddextexp-test)
1290
Marat Dukhan97579532019-10-18 16:40:39 -07001291 ADD_EXECUTABLE(f32-raddstoreexpminusmax-test test/f32-raddstoreexpminusmax.cc)
1292 SET_TARGET_PROPERTIES(f32-raddstoreexpminusmax-test PROPERTIES
1293 CXX_STANDARD 11
1294 CXX_STANDARD_REQUIRED YES
1295 CXX_EXTENSIONS YES)
1296 TARGET_INCLUDE_DIRECTORIES(f32-raddstoreexpminusmax-test PRIVATE src test)
1297 TARGET_LINK_LIBRARIES(f32-raddstoreexpminusmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1298 ADD_TEST(f32-raddstoreexpminusmax-test f32-raddstoreexpminusmax-test)
1299
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001300 ADD_EXECUTABLE(f32-rmax-test test/f32-rmax.cc)
1301 SET_TARGET_PROPERTIES(f32-rmax-test PROPERTIES
1302 CXX_STANDARD 11
1303 CXX_STANDARD_REQUIRED YES
1304 CXX_EXTENSIONS YES)
1305 TARGET_INCLUDE_DIRECTORIES(f32-rmax-test PRIVATE src test)
1306 TARGET_LINK_LIBRARIES(f32-rmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1307 ADD_TEST(f32-rmax-test f32-rmax-test)
1308
Marat Dukhan581c1ac2019-11-18 14:57:54 -08001309 ADD_EXECUTABLE(f32-sigmoid-test test/f32-sigmoid.cc)
1310 SET_TARGET_PROPERTIES(f32-sigmoid-test PROPERTIES
1311 CXX_STANDARD 11
1312 CXX_STANDARD_REQUIRED YES
1313 CXX_EXTENSIONS YES)
1314 TARGET_INCLUDE_DIRECTORIES(f32-sigmoid-test PRIVATE src test)
1315 TARGET_LINK_LIBRARIES(f32-sigmoid-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1316 ADD_TEST(f32-sigmoid-test f32-sigmoid-test)
1317
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001318 ADD_EXECUTABLE(f32-spmm-test test/f32-spmm.cc)
1319 SET_TARGET_PROPERTIES(f32-spmm-test PROPERTIES
1320 CXX_STANDARD 11
1321 CXX_STANDARD_REQUIRED YES
1322 CXX_EXTENSIONS YES)
1323 TARGET_INCLUDE_DIRECTORIES(f32-spmm-test PRIVATE src test)
1324 TARGET_LINK_LIBRARIES(f32-spmm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1325 ADD_TEST(f32-spmm-test f32-spmm-test)
1326
Marat Dukhanefc47b82019-11-18 09:25:38 -08001327 ADD_EXECUTABLE(f32-vadd-nc-test test/f32-vadd.cc)
1328 SET_TARGET_PROPERTIES(f32-vadd-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001329 CXX_STANDARD 11
1330 CXX_STANDARD_REQUIRED YES
1331 CXX_EXTENSIONS YES)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001332 TARGET_INCLUDE_DIRECTORIES(f32-vadd-nc-test PRIVATE src test)
1333 TARGET_LINK_LIBRARIES(f32-vadd-nc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1334 ADD_TEST(f32-vadd-nc-test f32-vadd-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001335
Marat Dukhanc76065f2019-11-14 18:13:33 -08001336 ADD_EXECUTABLE(f32-vaddc-test test/f32-vaddc.cc)
1337 SET_TARGET_PROPERTIES(f32-vaddc-test PROPERTIES
1338 CXX_STANDARD 11
1339 CXX_STANDARD_REQUIRED YES
1340 CXX_EXTENSIONS YES)
1341 TARGET_INCLUDE_DIRECTORIES(f32-vaddc-test PRIVATE src test)
1342 TARGET_LINK_LIBRARIES(f32-vaddc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1343 ADD_TEST(f32-vaddc-test f32-vaddc-test)
1344
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001345 ADD_EXECUTABLE(f32-vmul-test test/f32-vmul.cc)
1346 SET_TARGET_PROPERTIES(f32-vmul-test PROPERTIES
1347 CXX_STANDARD 11
1348 CXX_STANDARD_REQUIRED YES
1349 CXX_EXTENSIONS YES)
1350 TARGET_INCLUDE_DIRECTORIES(f32-vmul-test PRIVATE src test)
1351 TARGET_LINK_LIBRARIES(f32-vmul-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1352 ADD_TEST(f32-vmul-test f32-vmul-test)
1353
Marat Dukhanc76065f2019-11-14 18:13:33 -08001354 ADD_EXECUTABLE(f32-vmulc-test test/f32-vmulc.cc)
1355 SET_TARGET_PROPERTIES(f32-vmulc-test PROPERTIES
1356 CXX_STANDARD 11
1357 CXX_STANDARD_REQUIRED YES
1358 CXX_EXTENSIONS YES)
1359 TARGET_INCLUDE_DIRECTORIES(f32-vmulc-test PRIVATE src test)
1360 TARGET_LINK_LIBRARIES(f32-vmulc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1361 ADD_TEST(f32-vmulc-test f32-vmulc-test)
1362
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001363 ADD_EXECUTABLE(f32-vmulcaddc-test test/f32-vmulcaddc.cc)
1364 SET_TARGET_PROPERTIES(f32-vmulcaddc-test PROPERTIES
1365 CXX_STANDARD 11
1366 CXX_STANDARD_REQUIRED YES
1367 CXX_EXTENSIONS YES)
1368 TARGET_INCLUDE_DIRECTORIES(f32-vmulcaddc-test PRIVATE src test)
1369 TARGET_LINK_LIBRARIES(f32-vmulcaddc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1370 ADD_TEST(f32-vmulcaddc-test f32-vmulcaddc-test)
1371
Marat Dukhan97579532019-10-18 16:40:39 -07001372 ADD_EXECUTABLE(f32-vscaleexpminusmax-test test/f32-vscaleexpminusmax.cc)
1373 SET_TARGET_PROPERTIES(f32-vscaleexpminusmax-test PROPERTIES
1374 CXX_STANDARD 11
1375 CXX_STANDARD_REQUIRED YES
1376 CXX_EXTENSIONS YES)
1377 TARGET_INCLUDE_DIRECTORIES(f32-vscaleexpminusmax-test PRIVATE src test)
1378 TARGET_LINK_LIBRARIES(f32-vscaleexpminusmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1379 ADD_TEST(f32-vscaleexpminusmax-test f32-vscaleexpminusmax-test)
1380
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001381 ADD_EXECUTABLE(f32-vscaleextexp-test test/f32-vscaleextexp.cc)
1382 SET_TARGET_PROPERTIES(f32-vscaleexpminusmax-test PROPERTIES
1383 CXX_STANDARD 11
1384 CXX_STANDARD_REQUIRED YES
1385 CXX_EXTENSIONS YES)
1386 TARGET_INCLUDE_DIRECTORIES(f32-vscaleextexp-test PRIVATE src test)
Marat Dukhana5977df2019-10-30 22:02:15 -07001387 TARGET_LINK_LIBRARIES(f32-vscaleextexp-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001388 ADD_TEST(f32-vscaleextexp-test f32-vscaleextexp-test)
1389
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001390 ADD_EXECUTABLE(f32-vsub-test test/f32-vsub.cc)
1391 SET_TARGET_PROPERTIES(f32-vsub-test PROPERTIES
1392 CXX_STANDARD 11
1393 CXX_STANDARD_REQUIRED YES
1394 CXX_EXTENSIONS YES)
1395 TARGET_INCLUDE_DIRECTORIES(f32-vsub-test PRIVATE src test)
1396 TARGET_LINK_LIBRARIES(f32-vsub-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1397 ADD_TEST(f32-vsub-test f32-vsub-test)
1398
Marat Dukhanc76065f2019-11-14 18:13:33 -08001399 ADD_EXECUTABLE(f32-vsubc-test test/f32-vsubc.cc)
1400 SET_TARGET_PROPERTIES(f32-vsubc-test PROPERTIES
1401 CXX_STANDARD 11
1402 CXX_STANDARD_REQUIRED YES
1403 CXX_EXTENSIONS YES)
1404 TARGET_INCLUDE_DIRECTORIES(f32-vsubc-test PRIVATE src test)
1405 TARGET_LINK_LIBRARIES(f32-vsubc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1406 ADD_TEST(f32-vsubc-test f32-vsubc-test)
1407
1408 ADD_EXECUTABLE(f32-vrsubc-test test/f32-vrsubc.cc)
1409 SET_TARGET_PROPERTIES(f32-vrsubc-test PROPERTIES
1410 CXX_STANDARD 11
1411 CXX_STANDARD_REQUIRED YES
1412 CXX_EXTENSIONS YES)
1413 TARGET_INCLUDE_DIRECTORIES(f32-vrsubc-test PRIVATE src test)
1414 TARGET_LINK_LIBRARIES(f32-vrsubc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1415 ADD_TEST(f32-vrsubc-test f32-vrsubc-test)
1416
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001417 ADD_EXECUTABLE(q8-avgpool-test test/q8-avgpool.cc)
1418 SET_TARGET_PROPERTIES(q8-avgpool-test PROPERTIES
1419 CXX_STANDARD 11
1420 CXX_STANDARD_REQUIRED YES
1421 CXX_EXTENSIONS YES)
1422 TARGET_INCLUDE_DIRECTORIES(q8-avgpool-test PRIVATE src test)
1423 TARGET_LINK_LIBRARIES(q8-avgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1424 ADD_TEST(q8-avgpool-test q8-avgpool-test)
1425
1426 ADD_EXECUTABLE(q8-dwconv-test test/q8-dwconv.cc)
1427 SET_TARGET_PROPERTIES(q8-dwconv-test PROPERTIES
1428 CXX_STANDARD 11
1429 CXX_STANDARD_REQUIRED YES
1430 CXX_EXTENSIONS YES)
1431 TARGET_INCLUDE_DIRECTORIES(q8-dwconv-test PRIVATE src test)
1432 TARGET_LINK_LIBRARIES(q8-dwconv-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1433 ADD_TEST(q8-dwconv-test q8-dwconv-test)
1434
1435 ADD_EXECUTABLE(q8-gavgpool-test test/q8-gavgpool.cc)
1436 SET_TARGET_PROPERTIES(q8-gavgpool-test PROPERTIES
1437 CXX_STANDARD 11
1438 CXX_STANDARD_REQUIRED YES
1439 CXX_EXTENSIONS YES)
1440 TARGET_INCLUDE_DIRECTORIES(q8-gavgpool-test PRIVATE src test)
1441 TARGET_LINK_LIBRARIES(q8-gavgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1442 ADD_TEST(q8-gavgpool-test q8-gavgpool-test)
1443
1444 ADD_EXECUTABLE(q8-gemm-test test/q8-gemm.cc)
1445 SET_TARGET_PROPERTIES(q8-gemm-test PROPERTIES
1446 CXX_STANDARD 11
1447 CXX_STANDARD_REQUIRED YES
1448 CXX_EXTENSIONS YES)
1449 TARGET_INCLUDE_DIRECTORIES(q8-gemm-test PRIVATE src test)
1450 TARGET_LINK_LIBRARIES(q8-gemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1451 ADD_TEST(q8-gemm-test q8-gemm-test)
1452
1453 ADD_EXECUTABLE(q8-igemm-test test/q8-igemm.cc)
1454 SET_TARGET_PROPERTIES(q8-igemm-test PROPERTIES
1455 CXX_STANDARD 11
1456 CXX_STANDARD_REQUIRED YES
1457 CXX_EXTENSIONS YES)
1458 TARGET_INCLUDE_DIRECTORIES(q8-igemm-test PRIVATE src test)
1459 TARGET_LINK_LIBRARIES(q8-igemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1460 ADD_TEST(q8-igemm-test q8-igemm-test)
1461
Marat Dukhanefc47b82019-11-18 09:25:38 -08001462 ADD_EXECUTABLE(q8-vadd-nc-test test/q8-vadd.cc)
1463 SET_TARGET_PROPERTIES(q8-vadd-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001464 CXX_STANDARD 11
1465 CXX_STANDARD_REQUIRED YES
1466 CXX_EXTENSIONS YES)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001467 TARGET_INCLUDE_DIRECTORIES(q8-vadd-nc-test PRIVATE src test)
1468 TARGET_LINK_LIBRARIES(q8-vadd-nc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1469 ADD_TEST(q8-vadd-nc-test q8-vadd-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001470
Marat Dukhanefc47b82019-11-18 09:25:38 -08001471 ADD_EXECUTABLE(u8-clamp-nc-test test/u8-clamp.cc)
1472 SET_TARGET_PROPERTIES(u8-clamp-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001473 CXX_STANDARD 11
1474 CXX_STANDARD_REQUIRED YES
1475 CXX_EXTENSIONS YES)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001476 TARGET_INCLUDE_DIRECTORIES(u8-clamp-nc-test PRIVATE src test)
1477 TARGET_LINK_LIBRARIES(u8-clamp-nc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1478 ADD_TEST(u8-clamp-nc-test u8-clamp-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001479
1480 ADD_EXECUTABLE(u8-lut32norm-test test/u8-lut32norm.cc)
1481 SET_TARGET_PROPERTIES(u8-lut32norm-test PROPERTIES
1482 CXX_STANDARD 11
1483 CXX_STANDARD_REQUIRED YES
1484 CXX_EXTENSIONS YES)
1485 TARGET_INCLUDE_DIRECTORIES(u8-lut32norm-test PRIVATE src test)
1486 TARGET_LINK_LIBRARIES(u8-lut32norm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1487 ADD_TEST(u8-lut32norm-test u8-lut32norm-test)
1488
1489 ADD_EXECUTABLE(u8-maxpool-test test/u8-maxpool.cc)
1490 SET_TARGET_PROPERTIES(u8-maxpool-test PROPERTIES
1491 CXX_STANDARD 11
1492 CXX_STANDARD_REQUIRED YES
1493 CXX_EXTENSIONS YES)
1494 TARGET_INCLUDE_DIRECTORIES(u8-maxpool-test PRIVATE src test)
1495 TARGET_LINK_LIBRARIES(u8-maxpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1496 ADD_TEST(u8-maxpool-test u8-maxpool-test)
1497
1498 ADD_EXECUTABLE(u8-rmax-test test/u8-rmax.cc)
1499 SET_TARGET_PROPERTIES(u8-rmax-test PROPERTIES
1500 CXX_STANDARD 11
1501 CXX_STANDARD_REQUIRED YES
1502 CXX_EXTENSIONS YES)
1503 TARGET_INCLUDE_DIRECTORIES(u8-rmax-test PRIVATE src test)
1504 TARGET_LINK_LIBRARIES(u8-rmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1505 ADD_TEST(u8-rmax-test u8-rmax-test)
1506
1507 ADD_EXECUTABLE(x32-packx-test test/x32-packx.cc)
1508 SET_TARGET_PROPERTIES(x32-packx-test PROPERTIES
1509 CXX_STANDARD 11
1510 CXX_STANDARD_REQUIRED YES
1511 CXX_EXTENSIONS YES)
1512 TARGET_INCLUDE_DIRECTORIES(x32-packx-test PRIVATE src test)
1513 TARGET_LINK_LIBRARIES(x32-packx-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1514 ADD_TEST(x32-packx-test x32-packx-test)
1515
1516 ADD_EXECUTABLE(x32-pad-test test/x32-pad.cc)
1517 SET_TARGET_PROPERTIES(x32-pad-test PROPERTIES
1518 CXX_STANDARD 11
1519 CXX_STANDARD_REQUIRED YES
1520 CXX_EXTENSIONS YES)
1521 TARGET_INCLUDE_DIRECTORIES(x32-pad-test PRIVATE src test)
1522 TARGET_LINK_LIBRARIES(x32-pad-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1523 ADD_TEST(x32-pad-test x32-pad-test)
1524
1525 ADD_EXECUTABLE(x32-unpool-test test/x32-unpool.cc)
1526 SET_TARGET_PROPERTIES(x32-unpool-test PROPERTIES
1527 CXX_STANDARD 11
1528 CXX_STANDARD_REQUIRED YES
1529 CXX_EXTENSIONS YES)
1530 TARGET_INCLUDE_DIRECTORIES(x32-unpool-test PRIVATE src test)
1531 TARGET_LINK_LIBRARIES(x32-unpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1532 ADD_TEST(x32-unpool-test x32-unpool-test)
1533
1534 ADD_EXECUTABLE(x32-zip-test test/x32-zip.cc)
1535 SET_TARGET_PROPERTIES(x32-zip-test PROPERTIES
1536 CXX_STANDARD 11
1537 CXX_STANDARD_REQUIRED YES
1538 CXX_EXTENSIONS YES)
1539 TARGET_INCLUDE_DIRECTORIES(x32-zip-test PRIVATE src test)
1540 TARGET_LINK_LIBRARIES(x32-zip-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1541 ADD_TEST(x32-zip-test x32-zip-test)
1542
1543 ADD_EXECUTABLE(x8-lut-test test/x8-lut.cc)
1544 SET_TARGET_PROPERTIES(x8-lut-test PROPERTIES
1545 CXX_STANDARD 11
1546 CXX_STANDARD_REQUIRED YES
1547 CXX_EXTENSIONS YES)
1548 TARGET_INCLUDE_DIRECTORIES(x8-lut-test PRIVATE src test)
1549 TARGET_LINK_LIBRARIES(x8-lut-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1550 ADD_TEST(x8-lut-test x8-lut-test)
1551
1552 ADD_EXECUTABLE(x8-zip-test test/x8-zip.cc)
1553 SET_TARGET_PROPERTIES(x8-zip-test PROPERTIES
1554 CXX_STANDARD 11
1555 CXX_STANDARD_REQUIRED YES
1556 CXX_EXTENSIONS YES)
1557 TARGET_INCLUDE_DIRECTORIES(x8-zip-test PRIVATE src test)
1558 TARGET_LINK_LIBRARIES(x8-zip-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1559 ADD_TEST(x8-zip-test x8-zip-test)
1560ENDIF()
1561
1562# ---[ XNNPACK microbenchmarks
1563IF(XNNPACK_BUILD_BENCHMARKS)
1564 # ---[ Build google benchmark
1565 IF(NOT TARGET benchmark)
1566 SET(BENCHMARK_ENABLE_TESTING OFF CACHE BOOL "")
1567 ADD_SUBDIRECTORY(
1568 "${GOOGLEBENCHMARK_SOURCE_DIR}"
1569 "${CONFU_DEPENDENCIES_BINARY_DIR}/googlebenchmark")
1570 ENDIF()
1571
1572 ADD_LIBRARY(bench-utils bench/utils.cc)
1573 SET_TARGET_PROPERTIES(bench-utils PROPERTIES
1574 CXX_STANDARD 11
1575 CXX_STANDARD_REQUIRED YES
1576 CXX_EXTENSIONS NO)
1577 TARGET_INCLUDE_DIRECTORIES(bench-utils PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
Marat Dukhanbad48fe2019-11-04 10:35:22 -08001578 TARGET_LINK_LIBRARIES(bench-utils PRIVATE benchmark cpuinfo)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001579
Marat Dukhancb10f262019-11-16 15:52:37 -08001580 # ---[ Build accuracy microbenchmarks
1581 ADD_EXECUTABLE(f32-exp-eval eval/f32-exp.cc)
1582 SET_TARGET_PROPERTIES(f32-exp-eval PROPERTIES
1583 CXX_STANDARD 11
1584 CXX_STANDARD_REQUIRED YES
1585 CXX_EXTENSIONS NO)
1586 TARGET_INCLUDE_DIRECTORIES(f32-exp-eval PRIVATE src)
1587 TARGET_LINK_LIBRARIES(f32-exp-eval PRIVATE XNNPACK benchmark fp16)
1588
1589 ADD_EXECUTABLE(f32-expminus-eval eval/f32-expminus.cc)
1590 SET_TARGET_PROPERTIES(f32-expminus-eval PROPERTIES
1591 CXX_STANDARD 11
1592 CXX_STANDARD_REQUIRED YES
1593 CXX_EXTENSIONS NO)
1594 TARGET_INCLUDE_DIRECTORIES(f32-expminus-eval PRIVATE src)
1595 TARGET_LINK_LIBRARIES(f32-expminus-eval PRIVATE XNNPACK benchmark fp16)
1596
1597 ADD_EXECUTABLE(f32-extexp-eval eval/f32-extexp.cc)
1598 SET_TARGET_PROPERTIES(f32-extexp-eval PROPERTIES
1599 CXX_STANDARD 11
1600 CXX_STANDARD_REQUIRED YES
1601 CXX_EXTENSIONS YES)
1602 TARGET_INCLUDE_DIRECTORIES(f32-extexp-eval PRIVATE src)
1603 TARGET_LINK_LIBRARIES(f32-extexp-eval PRIVATE XNNPACK benchmark fp16)
1604
1605 ADD_EXECUTABLE(f32-sigmoid-eval eval/f32-sigmoid.cc)
1606 SET_TARGET_PROPERTIES(f32-sigmoid-eval PROPERTIES
1607 CXX_STANDARD 11
1608 CXX_STANDARD_REQUIRED YES
1609 CXX_EXTENSIONS NO)
1610 TARGET_INCLUDE_DIRECTORIES(f32-sigmoid-eval PRIVATE src)
1611 TARGET_LINK_LIBRARIES(f32-sigmoid-eval PRIVATE XNNPACK benchmark fp16)
1612
Marat Dukhan5f18d262019-10-31 10:24:14 -07001613 # ---[ Build end-to-end microbenchmarks
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001614 ADD_EXECUTABLE(end2end-bench bench/end2end.cc models/mobilenet-v1.cc models/mobilenet-v2.cc)
1615 SET_TARGET_PROPERTIES(end2end-bench PROPERTIES
1616 CXX_STANDARD 11
1617 CXX_STANDARD_REQUIRED YES
1618 CXX_EXTENSIONS NO)
1619 TARGET_INCLUDE_DIRECTORIES(end2end-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
Frank Barcharde72e2872019-10-31 11:12:15 -07001620 TARGET_LINK_LIBRARIES(end2end-bench PRIVATE XNNPACK benchmark bench-utils)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001621
Marat Dukhanef4416e2019-10-31 13:44:40 -07001622 ADD_EXECUTABLE(f32-dwconv-e2e-bench bench/f32-dwconv-e2e.cc models/mobilenet-v1.cc models/mobilenet-v2.cc)
1623 SET_TARGET_PROPERTIES(f32-dwconv-e2e-bench PROPERTIES
1624 CXX_STANDARD 11
1625 CXX_STANDARD_REQUIRED YES
1626 CXX_EXTENSIONS NO)
Marat Dukhanc9d2f3f2019-10-31 14:02:07 -07001627 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-e2e-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}" src)
Marat Dukhanef4416e2019-10-31 13:44:40 -07001628 TARGET_LINK_LIBRARIES(f32-dwconv-e2e-bench PRIVATE XNNPACK benchmark bench-utils)
1629
Marat Dukhan0a5a53f2019-10-31 10:58:31 -07001630 ADD_EXECUTABLE(f32-gemm-e2e-bench bench/f32-gemm-e2e.cc models/mobilenet-v1.cc models/mobilenet-v2.cc)
1631 SET_TARGET_PROPERTIES(f32-gemm-e2e-bench PROPERTIES
Marat Dukhan5f18d262019-10-31 10:24:14 -07001632 CXX_STANDARD 11
1633 CXX_STANDARD_REQUIRED YES
1634 CXX_EXTENSIONS NO)
Marat Dukhanc9d2f3f2019-10-31 14:02:07 -07001635 TARGET_INCLUDE_DIRECTORIES(f32-gemm-e2e-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}" src)
Frank Barchardc712fa42019-10-31 14:00:21 -07001636 TARGET_LINK_LIBRARIES(f32-gemm-e2e-bench PRIVATE XNNPACK benchmark bench-utils)
Marat Dukhan5f18d262019-10-31 10:24:14 -07001637
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001638 # ---[ Build operator-level microbenchmarks
1639 ADD_EXECUTABLE(add-bench bench/add.cc)
1640 SET_TARGET_PROPERTIES(add-bench PROPERTIES
1641 CXX_STANDARD 11
1642 CXX_STANDARD_REQUIRED YES
1643 CXX_EXTENSIONS NO)
1644 TARGET_INCLUDE_DIRECTORIES(add-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1645 TARGET_LINK_LIBRARIES(add-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1646
1647 ADD_EXECUTABLE(average-pooling-bench bench/average-pooling.cc)
1648 SET_TARGET_PROPERTIES(average-pooling-bench PROPERTIES
1649 CXX_STANDARD 11
1650 CXX_STANDARD_REQUIRED YES
1651 CXX_EXTENSIONS NO)
1652 TARGET_INCLUDE_DIRECTORIES(average-pooling-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1653 TARGET_LINK_LIBRARIES(average-pooling-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1654
1655 ADD_EXECUTABLE(channel-shuffle-bench bench/channel-shuffle.cc)
1656 SET_TARGET_PROPERTIES(channel-shuffle-bench PROPERTIES
1657 CXX_STANDARD 11
1658 CXX_STANDARD_REQUIRED YES
1659 CXX_EXTENSIONS NO)
1660 TARGET_INCLUDE_DIRECTORIES(channel-shuffle-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1661 TARGET_LINK_LIBRARIES(channel-shuffle-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1662
1663 ADD_EXECUTABLE(convolution-bench bench/convolution.cc)
1664 SET_TARGET_PROPERTIES(convolution-bench PROPERTIES
1665 CXX_STANDARD 11
1666 CXX_STANDARD_REQUIRED YES
1667 CXX_EXTENSIONS NO)
1668 TARGET_INCLUDE_DIRECTORIES(convolution-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1669 TARGET_LINK_LIBRARIES(convolution-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1670
1671 ADD_EXECUTABLE(deconvolution-bench bench/deconvolution.cc)
1672 SET_TARGET_PROPERTIES(deconvolution-bench PROPERTIES
1673 CXX_STANDARD 11
1674 CXX_STANDARD_REQUIRED YES
1675 CXX_EXTENSIONS NO)
1676 TARGET_INCLUDE_DIRECTORIES(deconvolution-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1677 TARGET_LINK_LIBRARIES(deconvolution-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1678
1679 ADD_EXECUTABLE(global-average-pooling-bench bench/global-average-pooling.cc)
1680 SET_TARGET_PROPERTIES(global-average-pooling-bench PROPERTIES
1681 CXX_STANDARD 11
1682 CXX_STANDARD_REQUIRED YES
1683 CXX_EXTENSIONS NO)
1684 TARGET_INCLUDE_DIRECTORIES(global-average-pooling-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1685 TARGET_LINK_LIBRARIES(global-average-pooling-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1686
1687 ADD_EXECUTABLE(max-pooling-bench bench/max-pooling.cc)
1688 SET_TARGET_PROPERTIES(max-pooling-bench PROPERTIES
1689 CXX_STANDARD 11
1690 CXX_STANDARD_REQUIRED YES
1691 CXX_EXTENSIONS NO)
1692 TARGET_INCLUDE_DIRECTORIES(max-pooling-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1693 TARGET_LINK_LIBRARIES(max-pooling-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1694
Marat Dukhan95b22432019-10-30 16:30:14 -07001695 ADD_EXECUTABLE(prelu-bench bench/prelu.cc)
1696 SET_TARGET_PROPERTIES(prelu-bench PROPERTIES
1697 CXX_STANDARD 11
1698 CXX_STANDARD_REQUIRED YES
1699 CXX_EXTENSIONS NO)
1700 TARGET_INCLUDE_DIRECTORIES(prelu-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1701 TARGET_LINK_LIBRARIES(prelu-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1702
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001703 ADD_EXECUTABLE(sigmoid-bench bench/sigmoid.cc)
1704 SET_TARGET_PROPERTIES(sigmoid-bench PROPERTIES
1705 CXX_STANDARD 11
1706 CXX_STANDARD_REQUIRED YES
1707 CXX_EXTENSIONS NO)
1708 TARGET_INCLUDE_DIRECTORIES(sigmoid-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1709 TARGET_LINK_LIBRARIES(sigmoid-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1710
1711 ADD_EXECUTABLE(softargmax-bench bench/softargmax.cc)
1712 SET_TARGET_PROPERTIES(softargmax-bench PROPERTIES
1713 CXX_STANDARD 11
1714 CXX_STANDARD_REQUIRED YES
1715 CXX_EXTENSIONS NO)
1716 TARGET_INCLUDE_DIRECTORIES(softargmax-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1717 TARGET_LINK_LIBRARIES(softargmax-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1718
1719 # ---[ Build microkernel-level microbenchmarks
1720 ADD_EXECUTABLE(f16-gemm-bench bench/f16-gemm.cc)
1721 SET_TARGET_PROPERTIES(f16-gemm-bench PROPERTIES
1722 CXX_STANDARD 11
1723 CXX_STANDARD_REQUIRED YES
1724 CXX_EXTENSIONS YES)
1725 TARGET_INCLUDE_DIRECTORIES(f16-gemm-bench PRIVATE src)
1726 TARGET_INCLUDE_DIRECTORIES(f16-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1727 TARGET_LINK_LIBRARIES(f16-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1728
1729 ADD_EXECUTABLE(f32-conv-hwc-bench bench/f32-conv-hwc.cc)
1730 SET_TARGET_PROPERTIES(f32-conv-hwc-bench PROPERTIES
1731 CXX_STANDARD 11
1732 CXX_STANDARD_REQUIRED YES
1733 CXX_EXTENSIONS YES)
1734 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc-bench PRIVATE src)
1735 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1736 TARGET_LINK_LIBRARIES(f32-conv-hwc-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1737
1738 ADD_EXECUTABLE(f32-dwconv-spchw-bench bench/f32-dwconv-spchw.cc)
1739 SET_TARGET_PROPERTIES(f32-dwconv-spchw-bench PROPERTIES
1740 CXX_STANDARD 11
1741 CXX_STANDARD_REQUIRED YES
1742 CXX_EXTENSIONS YES)
1743 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-spchw-bench PRIVATE src)
1744 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-spchw-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1745 TARGET_LINK_LIBRARIES(f32-dwconv-spchw-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1746
1747 ADD_EXECUTABLE(f32-dwconv-bench bench/f32-dwconv.cc)
1748 SET_TARGET_PROPERTIES(f32-dwconv-bench PROPERTIES
1749 CXX_STANDARD 11
1750 CXX_STANDARD_REQUIRED YES
1751 CXX_EXTENSIONS YES)
1752 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-bench PRIVATE src)
1753 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1754 TARGET_LINK_LIBRARIES(f32-dwconv-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1755
1756 ADD_EXECUTABLE(f32-gemm-bench bench/f32-gemm.cc)
1757 SET_TARGET_PROPERTIES(f32-gemm-bench PROPERTIES
1758 CXX_STANDARD 11
1759 CXX_STANDARD_REQUIRED YES
1760 CXX_EXTENSIONS YES)
1761 TARGET_INCLUDE_DIRECTORIES(f32-gemm-bench PRIVATE src)
1762 TARGET_INCLUDE_DIRECTORIES(f32-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1763 TARGET_LINK_LIBRARIES(f32-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1764
1765 ADD_EXECUTABLE(f32-igemm-bench bench/f32-igemm.cc)
1766 SET_TARGET_PROPERTIES(f32-igemm-bench PROPERTIES
1767 CXX_STANDARD 11
1768 CXX_STANDARD_REQUIRED YES
1769 CXX_EXTENSIONS YES)
1770 TARGET_INCLUDE_DIRECTORIES(f32-igemm-bench PRIVATE src)
1771 TARGET_INCLUDE_DIRECTORIES(f32-igemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1772 TARGET_LINK_LIBRARIES(f32-igemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1773
1774 ADD_EXECUTABLE(f32-im2col-gemm-bench bench/f32-im2col-gemm.cc src/im2col.c)
1775 SET_TARGET_PROPERTIES(f32-im2col-gemm-bench PROPERTIES
1776 C_STANDARD 99
1777 C_STANDARD_REQUIRED YES
1778 C_EXTENSIONS NO
1779 CXX_STANDARD 11
1780 CXX_STANDARD_REQUIRED YES
1781 CXX_EXTENSIONS YES)
1782 TARGET_INCLUDE_DIRECTORIES(f32-im2col-gemm-bench PRIVATE src)
1783 TARGET_INCLUDE_DIRECTORIES(f32-im2col-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1784 TARGET_LINK_LIBRARIES(f32-im2col-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1785
1786 ADD_EXECUTABLE(f32-rmax-bench bench/f32-rmax.cc)
1787 SET_TARGET_PROPERTIES(f32-rmax-bench PROPERTIES
1788 CXX_STANDARD 11
1789 CXX_STANDARD_REQUIRED YES
1790 CXX_EXTENSIONS YES)
1791 TARGET_INCLUDE_DIRECTORIES(f32-rmax-bench PRIVATE src)
1792 TARGET_INCLUDE_DIRECTORIES(f32-rmax-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1793 TARGET_LINK_LIBRARIES(f32-rmax-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1794
Marat Dukhan581c1ac2019-11-18 14:57:54 -08001795 ADD_EXECUTABLE(f32-sigmoid-bench bench/f32-sigmoid.cc)
1796 SET_TARGET_PROPERTIES(f32-sigmoid-bench PROPERTIES
1797 CXX_STANDARD 11
1798 CXX_STANDARD_REQUIRED YES
1799 CXX_EXTENSIONS YES)
1800 TARGET_INCLUDE_DIRECTORIES(f32-sigmoid-bench PRIVATE src)
1801 TARGET_INCLUDE_DIRECTORIES(f32-sigmoid-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1802 TARGET_LINK_LIBRARIES(f32-sigmoid-bench PRIVATE XNNPACK benchmark bench-utils)
1803
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001804 ADD_EXECUTABLE(f32-spmm-bench bench/f32-spmm.cc)
1805 SET_TARGET_PROPERTIES(f32-spmm-bench PROPERTIES
1806 CXX_STANDARD 11
1807 CXX_STANDARD_REQUIRED YES
1808 CXX_EXTENSIONS YES)
1809 TARGET_INCLUDE_DIRECTORIES(f32-spmm-bench PRIVATE src)
1810 TARGET_INCLUDE_DIRECTORIES(f32-spmm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1811 TARGET_LINK_LIBRARIES(f32-spmm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1812
Ashkan Aliabadi7892d972019-10-24 16:00:52 -07001813 ADD_EXECUTABLE(f32-softargmax-bench bench/f32-softargmax.cc)
1814 SET_TARGET_PROPERTIES(f32-softargmax-bench PROPERTIES
Marat Dukhan4a4a7fa2019-10-21 13:46:14 -07001815 CXX_STANDARD 11
1816 CXX_STANDARD_REQUIRED YES
1817 CXX_EXTENSIONS YES)
Ashkan Aliabadi7892d972019-10-24 16:00:52 -07001818 TARGET_INCLUDE_DIRECTORIES(f32-softargmax-bench PRIVATE src)
1819 TARGET_INCLUDE_DIRECTORIES(f32-softargmax-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1820 TARGET_LINK_LIBRARIES(f32-softargmax-bench PRIVATE XNNPACK fp16 benchmark bench-utils)
Marat Dukhan4a4a7fa2019-10-21 13:46:14 -07001821
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001822 ADD_EXECUTABLE(q8-gemm-bench bench/q8-gemm.cc)
1823 SET_TARGET_PROPERTIES(q8-gemm-bench PROPERTIES
1824 CXX_STANDARD 11
1825 CXX_STANDARD_REQUIRED YES
1826 CXX_EXTENSIONS YES)
1827 TARGET_INCLUDE_DIRECTORIES(q8-gemm-bench PRIVATE src)
1828 TARGET_INCLUDE_DIRECTORIES(q8-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1829 TARGET_LINK_LIBRARIES(q8-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1830ENDIF()