blob: 9b18e290cfd56c561f6fa86a16368519eed99e09 [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
134 src/add.c
135 src/argmax-pooling.c
136 src/average-pooling.c
137 src/channel-pad.c
138 src/channel-shuffle.c
139 src/clamp.c
140 src/convolution-spnchw.c
141 src/convolution.c
142 src/deconvolution.c
143 src/fully-connected.c
144 src/global-average-pooling-spnchw.c
145 src/global-average-pooling.c
146 src/hardswish.c
147 src/leaky-relu.c
148 src/max-pooling.c
149 src/prelu.c
150 src/sigmoid.c
151 src/softargmax.c
152 src/unpooling.c)
153
154SET(XNNPACK_COLD_SRCS ${XNNPACK_OPERATOR_SRCS})
155LIST(APPEND XNNPACK_COLD_SRCS
156 src/init.c
157 src/operator-delete.c)
158
159SET(XNNPACK_HOT_SRCS
160 src/indirection.c
161 src/operator-run.c)
162
163SET(XNNPACK_SCALAR_MICROKERNEL_SRCS
164 src/f32-argmaxpool/mp9p8q-scalar.c
165 src/f32-argmaxpool/up4-scalar.c
166 src/f32-argmaxpool/up9-scalar.c
167 src/f32-avgpool/mp9p8q-scalar.c
168 src/f32-avgpool/up9-scalar.c
169 src/f32-clamp/scalar.c
170 src/f32-igemm/1x4-scalar.c
171 src/f32-igemm/2x4-scalar.c
172 src/f32-igemm/4x2-scalar.c
173 src/f32-igemm/4x4-scalar.c
174 src/f32-dwconv/up1x25-scalar.c
175 src/f32-dwconv/up1x4-scalar.c
176 src/f32-dwconv/up1x9-scalar.c
177 src/f32-gavgpool/mp7p7q-scalar.c
178 src/f32-gavgpool/up7-scalar.c
179 src/f32-gemm/1x4-scalar.c
180 src/f32-gemm/2x4-scalar.c
181 src/f32-gemm/4x2-scalar.c
182 src/f32-gemm/4x4-scalar.c
183 src/f32-gemminc/1x4-scalar.c
184 src/f32-gemminc/2x4-scalar.c
185 src/f32-gemminc/4x4-scalar.c
186 src/f32-hswish/scalar.c
187 src/f32-maxpool/9p8q-scalar.c
188 src/f32-pavgpool/mp9p8q-scalar.c
189 src/f32-pavgpool/up9-scalar.c
190 src/f32-ppmm/2x4-scalar.c
191 src/f32-ppmm/3x3-scalar.c
192 src/f32-ppmm/4x2-scalar.c
193 src/f32-ppmm/4x4-scalar.c
194 src/f32-prelu/x4-scalar.c
195 src/f32-rmax/scalar.c
196 src/f32-spmm/1x1-scalar-pipelined.c
197 src/f32-spmm/1x1-scalar-unroll2.c
198 src/f32-spmm/1x1-scalar.c
199 src/f32-spmm/2x1-scalar-pipelined.c
200 src/f32-spmm/2x1-scalar-unroll2.c
201 src/f32-spmm/2x1-scalar.c
202 src/f32-spmm/4x1-scalar-pipelined.c
203 src/f32-spmm/4x1-scalar-unroll2.c
204 src/f32-spmm/4x1-scalar.c
205 src/f32-spmm/8x1-scalar-pipelined.c
206 src/f32-spmm/8x1-scalar-unroll2.c
207 src/f32-spmm/8x1-scalar.c
208 src/f32-vadd/scalar.c
209 src/f32-vmul/scalar.c
210 src/f32-vmulcaddc/c1-scalar-x2.c
211 src/f32-vsub/scalar.c
212 src/q8-avgpool/mp9p8q-scalar.c
213 src/q8-avgpool/up9-scalar.c
214 src/q8-igemm/2x2-scalar.c
215 src/q8-dwconv/up1x9-scalar.c
216 src/q8-gavgpool/mp7p7q-scalar.c
217 src/q8-gavgpool/up7-scalar.c
218 src/q8-gemm/2x2-scalar.c
219 src/q8-vadd/scalar.c
220 src/u8-clamp/scalar.c
221 src/u8-lut32norm/scalar.c
222 src/u8-maxpool/9p8q-scalar.c
223 src/u8-rmax/scalar.c
224 src/x32-packx/x2-scalar.c
225 src/x32-packx/x3-scalar.c
226 src/x32-packx/x4-scalar.c
227 src/x32-pad/x2-scalar.c
228 src/x32-unpool/scalar.c
229 src/x32-zip/x2-scalar.c
230 src/x32-zip/x3-scalar.c
231 src/x32-zip/x4-scalar.c
232 src/x32-zip/xm-scalar.c
233 src/x8-lut/scalar.c
234 src/x8-zip/x2-scalar.c
235 src/x8-zip/x3-scalar.c
236 src/x8-zip/x4-scalar.c
237 src/x8-zip/xm-scalar.c)
238
239SET(XNNPACK_PSIMD_MICROKERNEL_SRCS
240 src/f32-argmaxpool/mp9p8q-psimd.c
241 src/f32-argmaxpool/up4-psimd.c
242 src/f32-argmaxpool/up9-psimd.c
243 src/f32-avgpool/mp9p8q-psimd.c
244 src/f32-avgpool/up9-psimd.c
245 src/f32-clamp/psimd.c
246 src/f32-igemm/1x8-psimd-loadsplat.c
247 src/f32-igemm/1x8-psimd-splat.c
248 src/f32-igemm/1x8s4-psimd.c
249 src/f32-igemm/4x2c4-psimd.c
250 src/f32-igemm/4x8-psimd-loadsplat.c
251 src/f32-igemm/4x8-psimd-splat.c
252 src/f32-igemm/4x8s4-psimd.c
253 src/f32-igemm/6x8-psimd-loadsplat.c
254 src/f32-igemm/6x8-psimd-splat.c
255 src/f32-igemm/6x8s4-psimd.c
256 src/f32-dwconv/up4x25-psimd.c
257 src/f32-dwconv/up4x4-psimd.c
258 src/f32-dwconv/up4x9-psimd.c
259 src/f32-gavgpool/mp7p7q-psimd.c
260 src/f32-gavgpool/up7-psimd.c
261 src/f32-gemm/1x8-psimd-loadsplat.c
262 src/f32-gemm/1x8-psimd-splat.c
263 src/f32-gemm/1x8s4-psimd.c
264 src/f32-gemm/4x8-psimd-loadsplat.c
265 src/f32-gemm/4x8-psimd-splat.c
266 src/f32-gemm/4x8s4-psimd.c
267 src/f32-gemm/6x8-psimd-loadsplat.c
268 src/f32-gemm/6x8-psimd-splat.c
269 src/f32-gemm/6x8s4-psimd.c
270 src/f32-gemminc/1x8-psimd-loadsplat.c
271 src/f32-gemminc/1x8-psimd-splat.c
272 src/f32-gemminc/1x8s4-psimd.c
273 src/f32-gemminc/4x8-psimd-loadsplat.c
274 src/f32-gemminc/4x8-psimd-splat.c
275 src/f32-gemminc/4x8s4-psimd.c
276 src/f32-gemminc/6x8-psimd-loadsplat.c
277 src/f32-gemminc/6x8-psimd-splat.c
278 src/f32-gemminc/6x8s4-psimd.c
279 src/f32-hswish/psimd.c
280 src/f32-maxpool/9p8q-psimd.c
281 src/f32-pavgpool/mp9p8q-psimd.c
282 src/f32-pavgpool/up9-psimd.c
283 src/f32-ppmm/4x8-psimd.c
284 src/f32-prelu/x4-psimd.c
285 src/f32-vadd/psimd.c
286 src/f32-vmul/psimd.c
287 src/f32-vmulcaddc/c4-psimd-x2.c
288 src/f32-vsub/psimd.c
289 src/x32-packx/x4-psimd.c
290 src/x32-pad/x2-psimd.c
291 src/x32-unpool/psimd.c
292 src/x32-zip/x2-psimd.c
293 src/x32-zip/x3-psimd.c
294 src/x32-zip/x4-psimd.c
295 src/x32-zip/xm-psimd.c)
296
297SET(XNNPACK_NEON_MICROKERNEL_SRCS
298 src/f32-avgpool/mp9p8q-neon.c
299 src/f32-avgpool/up9-neon.c
300 src/f32-clamp/neon.c
301 src/f32-igemm/1x8-neon-ld64.c
302 src/f32-igemm/4x12-neon-ld64.c
303 src/f32-igemm/4x2-neon-ld64.c
304 src/f32-igemm/4x4-neon-ld64.c
305 src/f32-igemm/4x8-neon-ld128.c
306 src/f32-igemm/4x8-neon-ld64.c
307 src/f32-igemm/6x8-neon-ld64.c
308 src/f32-dwconv/up4x9-neon.c
309 src/f32-gavgpool-spchw/neon-x4.c
310 src/f32-gavgpool/mp7p7q-neon.c
311 src/f32-gavgpool/up7-neon.c
312 src/f32-gemm/1x8-neon-ld64.c
313 src/f32-gemm/4x12-neon-ld64.c
314 src/f32-gemm/4x2-neon-ld64.c
315 src/f32-gemm/4x8-neon-ld128.c
316 src/f32-gemm/4x8-neon-ld64.c
317 src/f32-gemm/5x8-neon-ld64.c
318 src/f32-gemm/6x8-neon-ld64.c
319 src/f32-gemminc/1x8-neon-ld64.c
320 src/f32-gemminc/4x12-neon-ld64.c
321 src/f32-gemminc/4x8-neon-ld128.c
322 src/f32-gemminc/4x8-neon-ld64.c
323 src/f32-gemminc/5x8-neon-ld64.c
324 src/f32-gemminc/6x8-neon-ld64.c
325 src/f32-hswish/neon.c
326 src/f32-pavgpool/mp9p8q-neon.c
327 src/f32-pavgpool/up9-neon.c
328 src/f32-ppmm/4x8-neon.c
329 src/f32-ppmm/8x8-neon.c
330 src/f32-rmax/neon.c
331 src/f32-vmulcaddc/c4-neon-x2.c
332 src/q8-avgpool/mp9p8q-neon.c
333 src/q8-avgpool/up9-neon.c
334 src/q8-igemm/4x8-neon.c
335 src/q8-igemm/8x8-neon.c
336 src/q8-dwconv/up8x9-neon.c
337 src/q8-gavgpool/mp7p7q-neon.c
338 src/q8-gavgpool/up7-neon.c
339 src/q8-gemm/4x8-neon.c
340 src/q8-gemm/8x8-neon.c
341 src/q8-vadd/neon.c
342 src/u8-clamp/neon.c
343 src/u8-maxpool/9p8q-neon.c
344 src/u8-rmax/neon.c
345 src/x32-packx/x4-neon-st4.c
346 src/x32-pad/x2-neon.c
347 src/x32-zip/x2-neon.c
348 src/x32-zip/x3-neon.c
349 src/x32-zip/x4-neon.c
350 src/x32-zip/xm-neon.c
351 src/x8-zip/x2-neon.c
352 src/x8-zip/x3-neon.c
353 src/x8-zip/x4-neon.c
354 src/x8-zip/xm-neon.c)
355
356SET(XNNPACK_NEONFMA_MICROKERNEL_SRCS
Frank Barcharddb45b6a2019-10-09 16:42:45 -0700357 src/f32-igemm/1x8-neonfma-ld64.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700358 src/f32-igemm/4x12-neonfma-ld64.c
359 src/f32-igemm/4x2-neonfma-ld64.c
360 src/f32-igemm/4x4-neonfma-ld64.c
361 src/f32-igemm/4x8-neonfma-ld128.c
362 src/f32-igemm/4x8-neonfma-ld64.c
363 src/f32-igemm/6x8-neonfma-ld64.c
364 src/f32-dwconv/up4x9-neonfma.c
365 src/f32-dwconv/up8x9-neonfma.c
366 src/f32-gemm/1x8-neonfma-ld64.c
367 src/f32-gemm/4x12-neonfma-ld64.c
368 src/f32-gemm/4x2-neonfma-ld64.c
369 src/f32-gemm/4x8-neonfma-ld128.c
370 src/f32-gemm/4x8-neonfma-ld64.c
371 src/f32-gemm/5x8-neonfma-ld64.c
372 src/f32-gemm/6x8-neonfma-ld64.c
373 src/f32-gemminc/1x8-neonfma-ld64.c
374 src/f32-gemminc/4x12-neonfma-ld64.c
375 src/f32-gemminc/4x8-neonfma-ld128.c
376 src/f32-gemminc/4x8-neonfma-ld64.c
377 src/f32-gemminc/5x8-neonfma-ld64.c
378 src/f32-gemminc/6x8-neonfma-ld64.c
379 src/f32-hswish/neonfma.c
380 src/f32-ppmm/4x8-neonfma.c
381 src/f32-ppmm/8x8-neonfma.c
382 src/f32-vmulcaddc/c4-neonfma-x2.c)
383
384SET(AARCH64_XNNPACK_NEONFMA_MICROKERNEL_SRCS
385 src/f32-conv-hwc/3x3s2p1c3x4-neonfma-2x2.c
386 src/f32-conv-hwc/3x3s2p1c3x8-neonfma-2x2.c
387 src/f32-conv-hwc2spchw/3x3s2p1c3x4-neonfma-2x2.c
388 src/f32-dwconv-spchw/3x3p1-neonfma.c
389 src/f32-dwconv-spchw/5x5p2-neonfma.c
390 src/f32-dwconv-spchw/3x3s2p1-neonfma.c
391 src/f32-dwconv-spchw/5x5s2p2-neonfma.c
392 src/f32-spmm/12x1-neonfma.c
393 src/f32-spmm/12x2-neonfma.c
394 src/f32-spmm/12x4-neonfma.c
395 src/f32-spmm/16x1-neonfma-pipelined.c
396 src/f32-spmm/16x1-neonfma-unroll2.c
397 src/f32-spmm/16x1-neonfma.c
398 src/f32-spmm/16x2-neonfma.c
399 src/f32-spmm/16x4-neonfma.c
400 src/f32-spmm/4x1-neonfma-pipelined.c
401 src/f32-spmm/4x1-neonfma-unroll2.c
402 src/f32-spmm/4x1-neonfma.c
403 src/f32-spmm/4x2-neonfma.c
404 src/f32-spmm/4x4-neonfma.c
405 src/f32-spmm/8x1-neonfma-pipelined.c
406 src/f32-spmm/8x1-neonfma-unroll2.c
407 src/f32-spmm/8x1-neonfma.c
408 src/f32-spmm/8x2-neonfma.c
409 src/f32-spmm/8x4-neonfma.c)
410
411SET(XNNPACK_AARCH64_NEONFP16ARITH_MICROKERNEL_SRCS
412 src/f16-gemm/4x8-neonfp16arith-ld64.c
413 src/f16-gemm/6x8-neonfp16arith-ld64.c
414 src/f16-gemm/8x8-neonfp16arith-ld64.c)
415
416SET(XNNPACK_SSE_MICROKERNEL_SRCS
417 src/f32-avgpool/mp9p8q-sse.c
418 src/f32-avgpool/up9-sse.c
419 src/f32-clamp/sse.c
420 src/f32-igemm/1x8-sse-dup.c
421 src/f32-igemm/1x8-sse-load1.c
422 src/f32-igemm/1x8s4-sse.c
423 src/f32-igemm/4x2c4-sse.c
424 src/f32-igemm/4x8-sse-dup.c
425 src/f32-igemm/4x8-sse-load1.c
426 src/f32-igemm/4x8s4-sse.c
427 src/f32-dwconv/up4x25-sse.c
428 src/f32-dwconv/up4x4-sse.c
429 src/f32-dwconv/up4x9-sse.c
430 src/f32-gavgpool-spchw/sse-x4.c
431 src/f32-gavgpool/mp7p7q-sse.c
432 src/f32-gavgpool/up7-sse.c
433 src/f32-gemm/1x8-sse-dup.c
434 src/f32-gemm/1x8-sse-load1.c
435 src/f32-gemm/1x8s4-sse.c
436 src/f32-gemm/4x8-sse-dup.c
437 src/f32-gemm/4x8-sse-load1.c
438 src/f32-gemm/4x8s4-sse.c
439 src/f32-gemminc/1x8-sse-dup.c
440 src/f32-gemminc/1x8-sse-load1.c
441 src/f32-gemminc/1x8s4-sse.c
442 src/f32-gemminc/4x8-sse-dup.c
443 src/f32-gemminc/4x8-sse-load1.c
444 src/f32-gemminc/4x8s4-sse.c
445 src/f32-hswish/sse.c
446 src/f32-maxpool/9p8q-sse.c
447 src/f32-pavgpool/mp9p8q-sse.c
448 src/f32-pavgpool/up9-sse.c
449 src/f32-dwconv-spchw/3x3p1-sse.c
450 src/f32-dwconv-spchw/3x3s2p1-sse.c
451 src/f32-ppmm/4x8-sse.c
452 src/f32-prelu/x4-sse.c
453 src/f32-rmax/sse.c
454 src/f32-spmm/4x1-sse.c
455 src/f32-spmm/8x1-sse.c
456 src/f32-vadd/sse.c
457 src/f32-vmul/sse.c
458 src/f32-vmulcaddc/c4-sse-x2.c
459 src/f32-vsub/sse.c
460 src/x32-packx/x4-sse.c)
461
462SET(XNNPACK_SSE2_MICROKERNEL_SRCS
463 src/f32-argmaxpool/mp9p8q-sse2.c
464 src/f32-argmaxpool/up4-sse2.c
465 src/f32-argmaxpool/up9-sse2.c
466 src/q8-avgpool/mp9p8q-sse2.c
467 src/q8-avgpool/up9-sse2.c
468 src/q8-igemm/4x4c2-sse2.c
469 src/q8-dwconv/up8x9-sse2.c
470 src/q8-gavgpool/mp7p7q-sse2.c
471 src/q8-gavgpool/up7-sse2.c
472 src/q8-gemm/2x4c8-sse2.c
473 src/q8-gemm/4x4c2-sse2.c
474 src/q8-vadd/sse2.c
475 src/u8-clamp/sse2.c
476 src/u8-maxpool/9p8q-sse2.c
477 src/u8-rmax/sse2.c
478 src/x32-pad/x2-sse2.c
479 src/x32-zip/x2-sse2.c
480 src/x32-zip/x3-sse2.c
481 src/x32-zip/x4-sse2.c
482 src/x32-zip/xm-sse2.c
483 src/x8-zip/x2-sse2.c
484 src/x8-zip/x3-sse2.c
485 src/x8-zip/x4-sse2.c
486 src/x8-zip/xm-sse2.c)
487
488SET(XNNPACK_AVX_MICROKERNEL_SRCS
489 src/f32-rmax/avx.c)
490
491SET(XNNPACK_AVX512F_MICROKERNEL_SRCS
492 src/f32-rmax/avx512f.c)
493
494SET(XNNPACK_AARCH32_ASM_MICROKERNEL_SRCS
495 src/q8-dwconv/up8x9-aarch32-neon.S)
496
497SET(XNNPACK_AARCH64_ASM_MICROKERNEL_SRCS
498 src/f32-dwconv/up4x9-aarch64-neonfma-cortex-a55.S
499 src/f32-dwconv/up4x9-aarch64-neonfma.S
500 src/f32-gemm/1x12-aarch64-neonfma-cortex-a53.S
Frank Barchard21be34f2019-10-09 19:32:19 -0700501 src/f32-gemm/1x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700502 src/f32-gemm/1x8-aarch64-neonfma-cortex-a57.S
503 src/f32-gemm/1x8-aarch64-neonfma-cortex-a75.S
504 src/f32-gemm/4x12-aarch64-neonfma-cortex-a53.S
505 src/f32-gemm/4x8-aarch64-neonfma-cortex-a57.S
506 src/f32-gemm/4x8-aarch64-neonfma-cortex-a75.S
507 src/f32-gemm/4x8-aarch64-neonfma-ld128.S
508 src/f32-gemm/4x8-aarch64-neonfma-ld64.S
509 src/f32-gemm/5x8-aarch64-neonfma-cortex-a75.S
510 src/f32-gemm/6x8-aarch64-neonfma-cortex-a57.S
511 src/f32-gemm/6x8-aarch64-neonfma-cortex-a73.S
512 src/f32-gemm/6x8-aarch64-neonfma-cortex-a75.S
513 src/f32-gemm/6x8-aarch64-neonfma-ld128.S
514 src/f32-gemm/6x8-aarch64-neonfma-ld64.S
515 src/f32-gemminc/1x12-aarch64-neonfma-cortex-a53.S
Frank Barchard21be34f2019-10-09 19:32:19 -0700516 src/f32-gemminc/1x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700517 src/f32-gemminc/1x8-aarch64-neonfma-cortex-a57.S
518 src/f32-gemminc/1x8-aarch64-neonfma-cortex-a75.S
519 src/f32-gemminc/4x12-aarch64-neonfma-cortex-a53.S
520 src/f32-gemminc/4x8-aarch64-neonfma-cortex-a57.S
521 src/f32-gemminc/4x8-aarch64-neonfma-cortex-a75.S
522 src/f32-gemminc/4x8-aarch64-neonfma-ld128.S
523 src/f32-gemminc/4x8-aarch64-neonfma-ld64.S
524 src/f32-gemminc/5x8-aarch64-neonfma-cortex-a75.S
525 src/f32-gemminc/6x8-aarch64-neonfma-cortex-a57.S
526 src/f32-gemminc/6x8-aarch64-neonfma-cortex-a73.S
527 src/f32-gemminc/6x8-aarch64-neonfma-cortex-a75.S
528 src/f32-gemminc/6x8-aarch64-neonfma-ld128.S
529 src/f32-gemminc/6x8-aarch64-neonfma-ld64.S
530 src/f32-igemm/1x12-aarch64-neonfma-cortex-a53.S
Frank Barchard21be34f2019-10-09 19:32:19 -0700531 src/f32-igemm/1x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700532 src/f32-igemm/1x8-aarch64-neonfma-cortex-a57.S
533 src/f32-igemm/1x8-aarch64-neonfma-cortex-a75.S
534 src/f32-igemm/4x12-aarch64-neonfma-cortex-a53.S
535 src/f32-igemm/4x8-aarch64-neonfma-cortex-a75.S
536 src/f32-igemm/5x8-aarch64-neonfma-cortex-a75.S
537 src/f32-igemm/6x8-aarch64-neonfma-cortex-a57.S
538 src/f32-igemm/6x8-aarch64-neonfma-cortex-a73.S
539 src/f32-igemm/6x8-aarch64-neonfma-cortex-a75.S)
540
541SET(XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SCALAR_MICROKERNEL_SRCS})
542IF(NOT CMAKE_SYSTEM_NAME STREQUAL "Emscripten")
543 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_PSIMD_MICROKERNEL_SRCS})
544ENDIF()
545IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^armv[5-8]" OR IOS_ARCH MATCHES "^armv7")
546 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEON_MICROKERNEL_SRCS})
547 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEONFMA_MICROKERNEL_SRCS})
548 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AARCH32_ASM_MICROKERNEL_SRCS})
549ENDIF()
550IF(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64" OR IOS_ARCH MATCHES "^arm64.*")
551 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEON_MICROKERNEL_SRCS})
552 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEONFMA_MICROKERNEL_SRCS})
553 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${AARCH64_XNNPACK_NEONFMA_MICROKERNEL_SRCS})
554 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AARCH64_NEONFP16ARITH_MICROKERNEL_SRCS})
555 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AARCH64_ASM_MICROKERNEL_SRCS})
556ENDIF()
557IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^(i[3-6]86|x86_64)$" OR IOS_ARCH MATCHES "^(i386|x86_64)$")
558 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SSE_MICROKERNEL_SRCS})
559 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SSE2_MICROKERNEL_SRCS})
560 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AVX_MICROKERNEL_SRCS})
561 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AVX512F_MICROKERNEL_SRCS})
562ENDIF()
563
564IF(XNNPACK_LIBRARY_TYPE STREQUAL "default")
565 ADD_LIBRARY(XNNPACK ${XNNPACK_COLD_SRCS} ${XNNPACK_HOT_SRCS} ${XNNPACK_MICROKERNEL_SRCS})
566ELSEIF(XNNPACK_LIBRARY_TYPE STREQUAL "shared")
567 ADD_LIBRARY(XNNPACK SHARED ${XNNPACK_COLD_SRCS} ${XNNPACK_HOT_SRCS} ${XNNPACK_MICROKERNEL_SRCS})
568ELSEIF(XNNPACK_LIBRARY_TYPE STREQUAL "static")
569 ADD_LIBRARY(XNNPACK STATIC ${XNNPACK_COLD_SRCS} ${XNNPACK_HOT_SRCS} ${XNNPACK_MICROKERNEL_SRCS})
570ELSE()
571 MESSAGE(FATAL_ERROR "Unsupported XNNPACK library type \"${XNNPACK_LIBRARY_TYPE}\". Must be \"static\", \"shared\", or \"default\"")
572ENDIF()
573SET_TARGET_PROPERTIES(XNNPACK PROPERTIES
574 C_STANDARD 99
575 C_EXTENSIONS YES)
576IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^armv[5-8]" OR IOS_ARCH MATCHES "^armv7")
577 SET_PROPERTY(SOURCE ${XNNPACK_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -marm ")
578 SET_PROPERTY(SOURCE ${XNNPACK_PSIMD_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfpu=neon ")
579 SET_PROPERTY(SOURCE ${XNNPACK_NEON_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfpu=neon ")
580 SET_PROPERTY(SOURCE ${XNNPACK_NEONFMA_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfpu=neon-vfpv4 ")
581 IF(IOS)
582 SET_PROPERTY(SOURCE ${XNNPACK_AARCH32_ASM_UKERNELS} APPEND_STRING PROPERTY COMPILE_FLAGS " -arch ${IOS_ARCH} ")
583 ENDIF()
584ENDIF()
585IF(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64" OR IOS_ARCH MATCHES "^arm64.*")
586 SET_PROPERTY(SOURCE ${XNNPACK_AARCH64_NEONFP16ARITH_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -march=armv8.2-a+fp16 ")
587 IF(IOS)
588 SET_PROPERTY(SOURCE ${XNNPACK_AARCH64_ASM_UKERNELS} APPEND_STRING PROPERTY COMPILE_FLAGS " -arch ${IOS_ARCH} ")
589 ENDIF()
590ENDIF()
591IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^(i[3-6]86|x86_64)$" OR IOS_ARCH MATCHES "^(i386|x86_64)$")
592 SET_PROPERTY(SOURCE ${XNNPACK_SSE_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -msse ")
593 SET_PROPERTY(SOURCE ${XNNPACK_SSE2_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -msse2 ")
594 SET_PROPERTY(SOURCE ${XNNPACK_AVX_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mavx ")
595 SET_PROPERTY(SOURCE ${XNNPACK_AVX512F_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mavx512f ")
596ENDIF()
597IF(CMAKE_BUILD_TYPE STREQUAL "Debug")
598 TARGET_COMPILE_DEFINITIONS(XNNPACK PRIVATE XNN_LOG_LEVEL=5)
599ELSE()
600 TARGET_COMPILE_DEFINITIONS(XNNPACK PRIVATE XNN_LOG_LEVEL=0)
601 SET_PROPERTY(SOURCE ${XNNPACK_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -O2 ")
602 SET_PROPERTY(SOURCE ${XNNPACK_HOT_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -O2 ")
603 SET_PROPERTY(SOURCE ${XNNPACK_COLD_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -Os ")
604ENDIF()
605IF(NOT CMAKE_SYSTEM_NAME STREQUAL "Emscripten")
606 SET_PROPERTY(SOURCE ${XNNPACK_PSIMD_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -O3 -ffast-math ")
607ENDIF()
608
609TARGET_INCLUDE_DIRECTORIES(XNNPACK PUBLIC include)
610TARGET_INCLUDE_DIRECTORIES(XNNPACK PRIVATE src)
611SET_TARGET_PROPERTIES(XNNPACK PROPERTIES PUBLIC_HEADER include/XNNPACK.h)
612
613# ---[ Configure clog
614IF(NOT TARGET clog)
615 SET(CLOG_BUILD_TESTS OFF CACHE BOOL "")
616 SET(CLOG_RUNTIME_TYPE "${CPUINFO_RUNTIME_TYPE}" CACHE STRING "")
617 ADD_SUBDIRECTORY(
618 "${CLOG_SOURCE_DIR}/deps/clog"
619 "${CMAKE_BINARY_DIR}/clog")
620 # We build static version of clog but a dynamic library may indirectly depend on it
621 SET_PROPERTY(TARGET clog PROPERTY POSITION_INDEPENDENT_CODE ON)
622ENDIF()
623TARGET_LINK_LIBRARIES(XNNPACK PRIVATE clog)
624
625# ---[ Configure cpuinfo
626IF(NOT TARGET cpuinfo)
627 SET(CPUINFO_BUILD_TOOLS OFF CACHE BOOL "")
628 SET(CPUINFO_BUILD_UNIT_TESTS OFF CACHE BOOL "")
629 SET(CPUINFO_BUILD_MOCK_TESTS OFF CACHE BOOL "")
630 SET(CPUINFO_BUILD_BENCHMARKS OFF CACHE BOOL "")
631 ADD_SUBDIRECTORY(
632 "${CPUINFO_SOURCE_DIR}"
633 "${CMAKE_BINARY_DIR}/cpuinfo")
634ENDIF()
635TARGET_LINK_LIBRARIES(XNNPACK PRIVATE cpuinfo)
636
637# ---[ Configure pthreadpool
638IF(NOT TARGET pthreadpool)
639 SET(PTHREADPOOL_BUILD_TESTS OFF CACHE BOOL "")
640 SET(PTHREADPOOL_BUILD_BENCHMARKS OFF CACHE BOOL "")
641 ADD_SUBDIRECTORY(
642 "${PTHREADPOOL_SOURCE_DIR}"
643 "${CMAKE_BINARY_DIR}/pthreadpool")
644ENDIF()
645TARGET_LINK_LIBRARIES(XNNPACK PUBLIC pthreadpool)
646
647# ---[ Configure FXdiv
648IF(NOT TARGET fxdiv)
649 SET(FXDIV_BUILD_TESTS OFF CACHE BOOL "")
650 SET(FXDIV_BUILD_BENCHMARKS OFF CACHE BOOL "")
651 ADD_SUBDIRECTORY(
652 "${FXDIV_SOURCE_DIR}"
653 "${CMAKE_BINARY_DIR}/FXdiv")
654ENDIF()
655TARGET_LINK_LIBRARIES(XNNPACK PRIVATE fxdiv)
656
657# ---[ Configure psimd
658IF(NOT TARGET psimd)
659 ADD_SUBDIRECTORY(
660 "${PSIMD_SOURCE_DIR}"
661 "${CMAKE_BINARY_DIR}/psimd")
662ENDIF()
663TARGET_LINK_LIBRARIES(XNNPACK PRIVATE psimd)
664
665# ---[ Configure FP16
666IF(NOT TARGET fp16)
667 SET(FP16_BUILD_TESTS OFF CACHE BOOL "")
668 SET(FP16_BUILD_BENCHMARKS OFF CACHE BOOL "")
669 ADD_SUBDIRECTORY(
670 "${FP16_SOURCE_DIR}"
671 "${CMAKE_BINARY_DIR}/FP16")
672ENDIF()
673TARGET_LINK_LIBRARIES(XNNPACK PRIVATE fp16)
674
675INSTALL(TARGETS XNNPACK
676 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
677 ARCHIVE DESTINATION ${CMAKE_INSTALL_LIBDIR}
678 PUBLIC_HEADER DESTINATION ${CMAKE_INSTALL_INCLUDEDIR})
679
680# ---[ XNNPACK unit tests
681IF(XNNPACK_BUILD_TESTS)
682 # ---[ Build google test
683 IF(NOT TARGET gtest)
684 SET(gtest_force_shared_crt ON CACHE BOOL "" FORCE)
685 ADD_SUBDIRECTORY(
686 "${GOOGLETEST_SOURCE_DIR}"
687 "${CMAKE_BINARY_DIR}/googletest")
688 ENDIF()
689
690 # ---[ Build operator-level unit tests
691 ADD_EXECUTABLE(add-test test/add.cc)
692 SET_TARGET_PROPERTIES(add-test PROPERTIES
693 CXX_STANDARD 11
694 CXX_STANDARD_REQUIRED YES
695 CXX_EXTENSIONS NO)
696 TARGET_INCLUDE_DIRECTORIES(add-test PRIVATE src test)
697 TARGET_LINK_LIBRARIES(add-test PRIVATE XNNPACK gtest gtest_main)
698 ADD_TEST(add-test add-test)
699
700 ADD_EXECUTABLE(argmax-pooling-test test/argmax-pooling.cc)
701 SET_TARGET_PROPERTIES(argmax-pooling-test PROPERTIES
702 CXX_STANDARD 11
703 CXX_STANDARD_REQUIRED YES
704 CXX_EXTENSIONS NO)
705 TARGET_INCLUDE_DIRECTORIES(argmax-pooling-test PRIVATE src test)
706 TARGET_LINK_LIBRARIES(argmax-pooling-test PRIVATE XNNPACK gtest gtest_main)
707 ADD_TEST(argmax-pooling-test argmax-pooling-test)
708
709 ADD_EXECUTABLE(average-pooling-test test/average-pooling.cc)
710 SET_TARGET_PROPERTIES(average-pooling-test PROPERTIES
711 CXX_STANDARD 11
712 CXX_STANDARD_REQUIRED YES
713 CXX_EXTENSIONS NO)
714 TARGET_INCLUDE_DIRECTORIES(average-pooling-test PRIVATE src test)
715 TARGET_LINK_LIBRARIES(average-pooling-test PRIVATE XNNPACK gtest gtest_main)
716 ADD_TEST(average-pooling-test average-pooling-test)
717
718 ADD_EXECUTABLE(channel-pad-test test/channel-pad.cc)
719 SET_TARGET_PROPERTIES(channel-pad-test PROPERTIES
720 CXX_STANDARD 11
721 CXX_STANDARD_REQUIRED YES
722 CXX_EXTENSIONS NO)
723 TARGET_INCLUDE_DIRECTORIES(channel-pad-test PRIVATE src test)
724 TARGET_LINK_LIBRARIES(channel-pad-test PRIVATE XNNPACK gtest gtest_main)
725 ADD_TEST(channel-pad-test channel-pad-test)
726
727 ADD_EXECUTABLE(channel-shuffle-test test/channel-shuffle.cc)
728 SET_TARGET_PROPERTIES(channel-shuffle-test PROPERTIES
729 CXX_STANDARD 11
730 CXX_STANDARD_REQUIRED YES
731 CXX_EXTENSIONS NO)
732 TARGET_INCLUDE_DIRECTORIES(channel-shuffle-test PRIVATE src test)
733 TARGET_LINK_LIBRARIES(channel-shuffle-test PRIVATE XNNPACK gtest gtest_main)
734 ADD_TEST(channel-shuffle-test channel-shuffle-test)
735
736 ADD_EXECUTABLE(clamp-test test/clamp.cc)
737 SET_TARGET_PROPERTIES(clamp-test PROPERTIES
738 CXX_STANDARD 11
739 CXX_STANDARD_REQUIRED YES
740 CXX_EXTENSIONS NO)
741 TARGET_INCLUDE_DIRECTORIES(clamp-test PRIVATE src test)
742 TARGET_LINK_LIBRARIES(clamp-test PRIVATE XNNPACK gtest gtest_main)
743 ADD_TEST(clamp-test clamp-test)
744
745 ADD_EXECUTABLE(convolution-test test/convolution.cc)
746 SET_TARGET_PROPERTIES(convolution-test PROPERTIES
747 CXX_STANDARD 11
748 CXX_STANDARD_REQUIRED YES
749 CXX_EXTENSIONS NO)
750 TARGET_INCLUDE_DIRECTORIES(convolution-test PRIVATE src test)
751 TARGET_LINK_LIBRARIES(convolution-test PRIVATE XNNPACK gtest gtest_main)
752 ADD_TEST(convolution-test convolution-test)
753
754 ADD_EXECUTABLE(convolution-spnchw-test test/convolution-spnchw.cc)
755 SET_TARGET_PROPERTIES(convolution-spnchw-test PROPERTIES
756 CXX_STANDARD 11
757 CXX_STANDARD_REQUIRED YES
758 CXX_EXTENSIONS NO)
759 TARGET_INCLUDE_DIRECTORIES(convolution-spnchw-test PRIVATE src test)
760 TARGET_LINK_LIBRARIES(convolution-spnchw-test PRIVATE XNNPACK gtest gtest_main)
761 ADD_TEST(convolution-spnchw-test convolution-spnchw-test)
762
763 ADD_EXECUTABLE(deconvolution-test test/deconvolution.cc)
764 SET_TARGET_PROPERTIES(deconvolution-test PROPERTIES
765 CXX_STANDARD 11
766 CXX_STANDARD_REQUIRED YES
767 CXX_EXTENSIONS NO)
768 TARGET_INCLUDE_DIRECTORIES(deconvolution-test PRIVATE src test)
769 TARGET_LINK_LIBRARIES(deconvolution-test PRIVATE XNNPACK gtest gtest_main)
770 ADD_TEST(deconvolution-test deconvolution-test)
771
772 ADD_EXECUTABLE(fully-connected-test test/fully-connected.cc)
773 SET_TARGET_PROPERTIES(fully-connected-test PROPERTIES
774 CXX_STANDARD 11
775 CXX_STANDARD_REQUIRED YES
776 CXX_EXTENSIONS NO)
777 TARGET_INCLUDE_DIRECTORIES(fully-connected-test PRIVATE src test)
778 TARGET_LINK_LIBRARIES(fully-connected-test PRIVATE XNNPACK gtest gtest_main)
779 ADD_TEST(fully-connected-test fully-connected-test)
780
781 ADD_EXECUTABLE(global-average-pooling-test test/global-average-pooling.cc)
782 SET_TARGET_PROPERTIES(global-average-pooling-test PROPERTIES
783 CXX_STANDARD 11
784 CXX_STANDARD_REQUIRED YES
785 CXX_EXTENSIONS NO)
786 TARGET_INCLUDE_DIRECTORIES(global-average-pooling-test PRIVATE src test)
787 TARGET_LINK_LIBRARIES(global-average-pooling-test PRIVATE XNNPACK gtest gtest_main)
788 ADD_TEST(global-average-pooling-test global-average-pooling-test)
789
790 ADD_EXECUTABLE(global-average-pooling-spnchw-test test/global-average-pooling-spnchw.cc)
791 SET_TARGET_PROPERTIES(global-average-pooling-spnchw-test PROPERTIES
792 CXX_STANDARD 11
793 CXX_STANDARD_REQUIRED YES
794 CXX_EXTENSIONS NO)
795 TARGET_INCLUDE_DIRECTORIES(global-average-pooling-spnchw-test PRIVATE src test)
796 TARGET_LINK_LIBRARIES(global-average-pooling-spnchw-test PRIVATE XNNPACK gtest gtest_main)
797 ADD_TEST(global-average-pooling-spnchw-test global-average-pooling-spnchw-test)
798
799 ADD_EXECUTABLE(hardswish-test test/hardswish.cc)
800 SET_TARGET_PROPERTIES(hardswish-test PROPERTIES
801 CXX_STANDARD 11
802 CXX_STANDARD_REQUIRED YES
803 CXX_EXTENSIONS NO)
804 TARGET_INCLUDE_DIRECTORIES(hardswish-test PRIVATE src test)
805 TARGET_LINK_LIBRARIES(hardswish-test PRIVATE XNNPACK gtest gtest_main)
806 ADD_TEST(hardswish-test hardswish-test)
807
808 ADD_EXECUTABLE(leaky-relu-test test/leaky-relu.cc)
809 SET_TARGET_PROPERTIES(leaky-relu-test PROPERTIES
810 CXX_STANDARD 11
811 CXX_STANDARD_REQUIRED YES
812 CXX_EXTENSIONS NO)
813 TARGET_INCLUDE_DIRECTORIES(leaky-relu-test PRIVATE src test)
814 TARGET_LINK_LIBRARIES(leaky-relu-test PRIVATE XNNPACK gtest gtest_main)
815 ADD_TEST(leaky-relu-test leaky-relu-test)
816
817 ADD_EXECUTABLE(max-pooling-test test/max-pooling.cc)
818 SET_TARGET_PROPERTIES(max-pooling-test PROPERTIES
819 CXX_STANDARD 11
820 CXX_STANDARD_REQUIRED YES
821 CXX_EXTENSIONS NO)
822 TARGET_INCLUDE_DIRECTORIES(max-pooling-test PRIVATE src test)
823 TARGET_LINK_LIBRARIES(max-pooling-test PRIVATE XNNPACK gtest gtest_main)
824 ADD_TEST(max-pooling-test max-pooling-test)
825
826 ADD_EXECUTABLE(prelu-test test/prelu.cc)
827 SET_TARGET_PROPERTIES(prelu-test PROPERTIES
828 CXX_STANDARD 11
829 CXX_STANDARD_REQUIRED YES
830 CXX_EXTENSIONS NO)
831 TARGET_INCLUDE_DIRECTORIES(prelu-test PRIVATE src test)
832 TARGET_LINK_LIBRARIES(prelu-test PRIVATE XNNPACK gtest gtest_main)
833 ADD_TEST(prelu-test prelu-test)
834
835 ADD_EXECUTABLE(sigmoid-test test/sigmoid.cc)
836 SET_TARGET_PROPERTIES(sigmoid-test PROPERTIES
837 CXX_STANDARD 11
838 CXX_STANDARD_REQUIRED YES
839 CXX_EXTENSIONS NO)
840 TARGET_INCLUDE_DIRECTORIES(sigmoid-test PRIVATE src test)
841 TARGET_LINK_LIBRARIES(sigmoid-test PRIVATE XNNPACK gtest gtest_main)
842 ADD_TEST(sigmoid-test sigmoid-test)
843
844 ADD_EXECUTABLE(softargmax-test test/softargmax.cc)
845 SET_TARGET_PROPERTIES(softargmax-test PROPERTIES
846 CXX_STANDARD 11
847 CXX_STANDARD_REQUIRED YES
848 CXX_EXTENSIONS NO)
849 TARGET_INCLUDE_DIRECTORIES(softargmax-test PRIVATE src test)
850 TARGET_LINK_LIBRARIES(softargmax-test PRIVATE XNNPACK gtest gtest_main)
851 ADD_TEST(softargmax-test softargmax-test)
852
853 ADD_EXECUTABLE(unpooling-test test/unpooling.cc)
854 SET_TARGET_PROPERTIES(unpooling-test PROPERTIES
855 CXX_STANDARD 11
856 CXX_STANDARD_REQUIRED YES
857 CXX_EXTENSIONS NO)
858 TARGET_INCLUDE_DIRECTORIES(unpooling-test PRIVATE src test)
859 TARGET_LINK_LIBRARIES(unpooling-test PRIVATE XNNPACK gtest gtest_main)
860 ADD_TEST(unpooling-test unpooling-test)
861
862 # ---[ Build microkernel-level unit tests
863 ADD_EXECUTABLE(f16-gemm-test test/f16-gemm.cc)
864 SET_TARGET_PROPERTIES(f16-gemm-test PROPERTIES
865 CXX_STANDARD 11
866 CXX_STANDARD_REQUIRED YES
867 CXX_EXTENSIONS YES)
868 TARGET_INCLUDE_DIRECTORIES(f16-gemm-test PRIVATE src test)
869 TARGET_LINK_LIBRARIES(f16-gemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
870 ADD_TEST(f16-gemm-test f16-gemm-test)
871
872 ADD_EXECUTABLE(f32-argmaxpool-test test/f32-argmaxpool.cc)
873 SET_TARGET_PROPERTIES(f32-argmaxpool-test PROPERTIES
874 CXX_STANDARD 11
875 CXX_STANDARD_REQUIRED YES
876 CXX_EXTENSIONS YES)
877 TARGET_INCLUDE_DIRECTORIES(f32-argmaxpool-test PRIVATE src test)
878 TARGET_LINK_LIBRARIES(f32-argmaxpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
879 ADD_TEST(f32-argmaxpool-test f32-argmaxpool-test)
880
881 ADD_EXECUTABLE(f32-avgpool-test test/f32-avgpool.cc)
882 SET_TARGET_PROPERTIES(f32-avgpool-test PROPERTIES
883 CXX_STANDARD 11
884 CXX_STANDARD_REQUIRED YES
885 CXX_EXTENSIONS YES)
886 TARGET_INCLUDE_DIRECTORIES(f32-avgpool-test PRIVATE src test)
887 TARGET_LINK_LIBRARIES(f32-avgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
888 ADD_TEST(f32-avgpool-test f32-avgpool-test)
889
890 ADD_EXECUTABLE(f32-clamp-test test/f32-clamp.cc)
891 SET_TARGET_PROPERTIES(f32-clamp-test PROPERTIES
892 CXX_STANDARD 11
893 CXX_STANDARD_REQUIRED YES
894 CXX_EXTENSIONS YES)
895 TARGET_INCLUDE_DIRECTORIES(f32-clamp-test PRIVATE src test)
896 TARGET_LINK_LIBRARIES(f32-clamp-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
897 ADD_TEST(f32-clamp-test f32-clamp-test)
898
899 ADD_EXECUTABLE(f32-conv-hwc-test test/f32-conv-hwc.cc)
900 SET_TARGET_PROPERTIES(f32-conv-hwc-test PROPERTIES
901 CXX_STANDARD 11
902 CXX_STANDARD_REQUIRED YES
903 CXX_EXTENSIONS YES)
904 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc-test PRIVATE src test)
905 TARGET_LINK_LIBRARIES(f32-conv-hwc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
906 ADD_TEST(f32-conv-hwc-test f32-conv-hwc-test)
907
908 ADD_EXECUTABLE(f32-conv-hwc2spchw-test test/f32-conv-hwc2spchw.cc)
909 SET_TARGET_PROPERTIES(f32-conv-hwc2spchw-test PROPERTIES
910 CXX_STANDARD 11
911 CXX_STANDARD_REQUIRED YES
912 CXX_EXTENSIONS YES)
913 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc2spchw-test PRIVATE src test)
914 TARGET_LINK_LIBRARIES(f32-conv-hwc2spchw-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
915 ADD_TEST(f32-conv-hwc2spchw-test f32-conv-hwc2spchw-test)
916
917 ADD_EXECUTABLE(f32-dwconv-spchw-test test/f32-dwconv-spchw.cc)
918 SET_TARGET_PROPERTIES(f32-dwconv-spchw-test PROPERTIES
919 CXX_STANDARD 11
920 CXX_STANDARD_REQUIRED YES
921 CXX_EXTENSIONS YES)
922 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-spchw-test PRIVATE src test)
923 TARGET_LINK_LIBRARIES(f32-dwconv-spchw-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
924 ADD_TEST(f32-dwconv-spchw-test f32-dwconv-spchw-test)
925
926 ADD_EXECUTABLE(f32-dwconv-test test/f32-dwconv.cc)
927 SET_TARGET_PROPERTIES(f32-dwconv-test PROPERTIES
928 CXX_STANDARD 11
929 CXX_STANDARD_REQUIRED YES
930 CXX_EXTENSIONS YES)
931 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-test PRIVATE src test)
932 TARGET_LINK_LIBRARIES(f32-dwconv-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
933 ADD_TEST(f32-dwconv-test f32-dwconv-test)
934
935 ADD_EXECUTABLE(f32-gavgpool-spchw-test test/f32-gavgpool-spchw.cc)
936 SET_TARGET_PROPERTIES(f32-gavgpool-spchw-test PROPERTIES
937 CXX_STANDARD 11
938 CXX_STANDARD_REQUIRED YES
939 CXX_EXTENSIONS YES)
940 TARGET_INCLUDE_DIRECTORIES(f32-gavgpool-spchw-test PRIVATE src test)
941 TARGET_LINK_LIBRARIES(f32-gavgpool-spchw-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
942 ADD_TEST(f32-gavgpool-spchw-test f32-gavgpool-spchw-test)
943
944 ADD_EXECUTABLE(f32-gavgpool-test test/f32-gavgpool.cc)
945 SET_TARGET_PROPERTIES(f32-gavgpool-test PROPERTIES
946 CXX_STANDARD 11
947 CXX_STANDARD_REQUIRED YES
948 CXX_EXTENSIONS YES)
949 TARGET_INCLUDE_DIRECTORIES(f32-gavgpool-test PRIVATE src test)
950 TARGET_LINK_LIBRARIES(f32-gavgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
951 ADD_TEST(f32-gavgpool-test f32-gavgpool-test)
952
953 ADD_EXECUTABLE(f32-gemm-test test/f32-gemm.cc)
954 SET_TARGET_PROPERTIES(f32-gemm-test PROPERTIES
955 CXX_STANDARD 11
956 CXX_STANDARD_REQUIRED YES
957 CXX_EXTENSIONS YES)
958 TARGET_INCLUDE_DIRECTORIES(f32-gemm-test PRIVATE src test)
959 TARGET_LINK_LIBRARIES(f32-gemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
960 ADD_TEST(f32-gemm-test f32-gemm-test)
961
962 ADD_EXECUTABLE(f32-gemminc-test test/f32-gemminc.cc)
963 SET_TARGET_PROPERTIES(f32-gemminc-test PROPERTIES
964 CXX_STANDARD 11
965 CXX_STANDARD_REQUIRED YES
966 CXX_EXTENSIONS YES)
967 TARGET_INCLUDE_DIRECTORIES(f32-gemminc-test PRIVATE src test)
968 TARGET_LINK_LIBRARIES(f32-gemminc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
969 ADD_TEST(f32-gemminc-test f32-gemminc-test)
970
971 ADD_EXECUTABLE(f32-hswish-test test/f32-hswish.cc)
972 SET_TARGET_PROPERTIES(f32-hswish-test PROPERTIES
973 CXX_STANDARD 11
974 CXX_STANDARD_REQUIRED YES
975 CXX_EXTENSIONS YES)
976 TARGET_INCLUDE_DIRECTORIES(f32-hswish-test PRIVATE src test)
977 TARGET_LINK_LIBRARIES(f32-hswish-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
978 ADD_TEST(f32-hswish-test f32-hswish-test)
979
980 ADD_EXECUTABLE(f32-igemm-test test/f32-igemm.cc)
981 SET_TARGET_PROPERTIES(f32-igemm-test PROPERTIES
982 CXX_STANDARD 11
983 CXX_STANDARD_REQUIRED YES
984 CXX_EXTENSIONS YES)
985 TARGET_INCLUDE_DIRECTORIES(f32-igemm-test PRIVATE src test)
986 TARGET_LINK_LIBRARIES(f32-igemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
987 ADD_TEST(f32-igemm-test f32-igemm-test)
988
989 ADD_EXECUTABLE(f32-maxpool-test test/f32-maxpool.cc)
990 SET_TARGET_PROPERTIES(f32-maxpool-test PROPERTIES
991 CXX_STANDARD 11
992 CXX_STANDARD_REQUIRED YES
993 CXX_EXTENSIONS YES)
994 TARGET_INCLUDE_DIRECTORIES(f32-maxpool-test PRIVATE src test)
995 TARGET_LINK_LIBRARIES(f32-maxpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
996 ADD_TEST(f32-maxpool-test f32-maxpool-test)
997
998 ADD_EXECUTABLE(f32-pavgpool-test test/f32-pavgpool.cc)
999 SET_TARGET_PROPERTIES(f32-pavgpool-test PROPERTIES
1000 CXX_STANDARD 11
1001 CXX_STANDARD_REQUIRED YES
1002 CXX_EXTENSIONS YES)
1003 TARGET_INCLUDE_DIRECTORIES(f32-pavgpool-test PRIVATE src test)
1004 TARGET_LINK_LIBRARIES(f32-pavgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1005 ADD_TEST(f32-pavgpool-test f32-pavgpool-test)
1006
1007 ADD_EXECUTABLE(f32-ppmm-test test/f32-ppmm.cc)
1008 SET_TARGET_PROPERTIES(f32-ppmm-test PROPERTIES
1009 CXX_STANDARD 11
1010 CXX_STANDARD_REQUIRED YES
1011 CXX_EXTENSIONS YES)
1012 TARGET_INCLUDE_DIRECTORIES(f32-ppmm-test PRIVATE src test)
1013 TARGET_LINK_LIBRARIES(f32-ppmm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1014 ADD_TEST(f32-ppmm-test f32-ppmm-test)
1015
1016 ADD_EXECUTABLE(f32-prelu-test test/f32-prelu.cc)
1017 SET_TARGET_PROPERTIES(f32-prelu-test PROPERTIES
1018 CXX_STANDARD 11
1019 CXX_STANDARD_REQUIRED YES
1020 CXX_EXTENSIONS YES)
1021 TARGET_INCLUDE_DIRECTORIES(f32-prelu-test PRIVATE src test)
1022 TARGET_LINK_LIBRARIES(f32-prelu-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1023 ADD_TEST(f32-prelu-test f32-prelu-test)
1024
1025 ADD_EXECUTABLE(f32-rmax-test test/f32-rmax.cc)
1026 SET_TARGET_PROPERTIES(f32-rmax-test PROPERTIES
1027 CXX_STANDARD 11
1028 CXX_STANDARD_REQUIRED YES
1029 CXX_EXTENSIONS YES)
1030 TARGET_INCLUDE_DIRECTORIES(f32-rmax-test PRIVATE src test)
1031 TARGET_LINK_LIBRARIES(f32-rmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1032 ADD_TEST(f32-rmax-test f32-rmax-test)
1033
1034 ADD_EXECUTABLE(f32-spmm-test test/f32-spmm.cc)
1035 SET_TARGET_PROPERTIES(f32-spmm-test PROPERTIES
1036 CXX_STANDARD 11
1037 CXX_STANDARD_REQUIRED YES
1038 CXX_EXTENSIONS YES)
1039 TARGET_INCLUDE_DIRECTORIES(f32-spmm-test PRIVATE src test)
1040 TARGET_LINK_LIBRARIES(f32-spmm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1041 ADD_TEST(f32-spmm-test f32-spmm-test)
1042
1043 ADD_EXECUTABLE(f32-vadd-test test/f32-vadd.cc)
1044 SET_TARGET_PROPERTIES(f32-vadd-test PROPERTIES
1045 CXX_STANDARD 11
1046 CXX_STANDARD_REQUIRED YES
1047 CXX_EXTENSIONS YES)
1048 TARGET_INCLUDE_DIRECTORIES(f32-vadd-test PRIVATE src test)
1049 TARGET_LINK_LIBRARIES(f32-vadd-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1050 ADD_TEST(f32-vadd-test f32-vadd-test)
1051
1052 ADD_EXECUTABLE(f32-vmul-test test/f32-vmul.cc)
1053 SET_TARGET_PROPERTIES(f32-vmul-test PROPERTIES
1054 CXX_STANDARD 11
1055 CXX_STANDARD_REQUIRED YES
1056 CXX_EXTENSIONS YES)
1057 TARGET_INCLUDE_DIRECTORIES(f32-vmul-test PRIVATE src test)
1058 TARGET_LINK_LIBRARIES(f32-vmul-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1059 ADD_TEST(f32-vmul-test f32-vmul-test)
1060
1061 ADD_EXECUTABLE(f32-vmulcaddc-test test/f32-vmulcaddc.cc)
1062 SET_TARGET_PROPERTIES(f32-vmulcaddc-test PROPERTIES
1063 CXX_STANDARD 11
1064 CXX_STANDARD_REQUIRED YES
1065 CXX_EXTENSIONS YES)
1066 TARGET_INCLUDE_DIRECTORIES(f32-vmulcaddc-test PRIVATE src test)
1067 TARGET_LINK_LIBRARIES(f32-vmulcaddc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1068 ADD_TEST(f32-vmulcaddc-test f32-vmulcaddc-test)
1069
1070 ADD_EXECUTABLE(f32-vsub-test test/f32-vsub.cc)
1071 SET_TARGET_PROPERTIES(f32-vsub-test PROPERTIES
1072 CXX_STANDARD 11
1073 CXX_STANDARD_REQUIRED YES
1074 CXX_EXTENSIONS YES)
1075 TARGET_INCLUDE_DIRECTORIES(f32-vsub-test PRIVATE src test)
1076 TARGET_LINK_LIBRARIES(f32-vsub-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1077 ADD_TEST(f32-vsub-test f32-vsub-test)
1078
1079 ADD_EXECUTABLE(q8-avgpool-test test/q8-avgpool.cc)
1080 SET_TARGET_PROPERTIES(q8-avgpool-test PROPERTIES
1081 CXX_STANDARD 11
1082 CXX_STANDARD_REQUIRED YES
1083 CXX_EXTENSIONS YES)
1084 TARGET_INCLUDE_DIRECTORIES(q8-avgpool-test PRIVATE src test)
1085 TARGET_LINK_LIBRARIES(q8-avgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1086 ADD_TEST(q8-avgpool-test q8-avgpool-test)
1087
1088 ADD_EXECUTABLE(q8-dwconv-test test/q8-dwconv.cc)
1089 SET_TARGET_PROPERTIES(q8-dwconv-test PROPERTIES
1090 CXX_STANDARD 11
1091 CXX_STANDARD_REQUIRED YES
1092 CXX_EXTENSIONS YES)
1093 TARGET_INCLUDE_DIRECTORIES(q8-dwconv-test PRIVATE src test)
1094 TARGET_LINK_LIBRARIES(q8-dwconv-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1095 ADD_TEST(q8-dwconv-test q8-dwconv-test)
1096
1097 ADD_EXECUTABLE(q8-gavgpool-test test/q8-gavgpool.cc)
1098 SET_TARGET_PROPERTIES(q8-gavgpool-test PROPERTIES
1099 CXX_STANDARD 11
1100 CXX_STANDARD_REQUIRED YES
1101 CXX_EXTENSIONS YES)
1102 TARGET_INCLUDE_DIRECTORIES(q8-gavgpool-test PRIVATE src test)
1103 TARGET_LINK_LIBRARIES(q8-gavgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1104 ADD_TEST(q8-gavgpool-test q8-gavgpool-test)
1105
1106 ADD_EXECUTABLE(q8-gemm-test test/q8-gemm.cc)
1107 SET_TARGET_PROPERTIES(q8-gemm-test PROPERTIES
1108 CXX_STANDARD 11
1109 CXX_STANDARD_REQUIRED YES
1110 CXX_EXTENSIONS YES)
1111 TARGET_INCLUDE_DIRECTORIES(q8-gemm-test PRIVATE src test)
1112 TARGET_LINK_LIBRARIES(q8-gemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1113 ADD_TEST(q8-gemm-test q8-gemm-test)
1114
1115 ADD_EXECUTABLE(q8-igemm-test test/q8-igemm.cc)
1116 SET_TARGET_PROPERTIES(q8-igemm-test PROPERTIES
1117 CXX_STANDARD 11
1118 CXX_STANDARD_REQUIRED YES
1119 CXX_EXTENSIONS YES)
1120 TARGET_INCLUDE_DIRECTORIES(q8-igemm-test PRIVATE src test)
1121 TARGET_LINK_LIBRARIES(q8-igemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1122 ADD_TEST(q8-igemm-test q8-igemm-test)
1123
1124 ADD_EXECUTABLE(q8-vadd-test test/q8-vadd.cc)
1125 SET_TARGET_PROPERTIES(q8-vadd-test PROPERTIES
1126 CXX_STANDARD 11
1127 CXX_STANDARD_REQUIRED YES
1128 CXX_EXTENSIONS YES)
1129 TARGET_INCLUDE_DIRECTORIES(q8-vadd-test PRIVATE src test)
1130 TARGET_LINK_LIBRARIES(q8-vadd-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1131 ADD_TEST(q8-vadd-test q8-vadd-test)
1132
1133 ADD_EXECUTABLE(u8-clamp-test test/u8-clamp.cc)
1134 SET_TARGET_PROPERTIES(u8-clamp-test PROPERTIES
1135 CXX_STANDARD 11
1136 CXX_STANDARD_REQUIRED YES
1137 CXX_EXTENSIONS YES)
1138 TARGET_INCLUDE_DIRECTORIES(u8-clamp-test PRIVATE src test)
1139 TARGET_LINK_LIBRARIES(u8-clamp-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1140 ADD_TEST(u8-clamp-test u8-clamp-test)
1141
1142 ADD_EXECUTABLE(u8-lut32norm-test test/u8-lut32norm.cc)
1143 SET_TARGET_PROPERTIES(u8-lut32norm-test PROPERTIES
1144 CXX_STANDARD 11
1145 CXX_STANDARD_REQUIRED YES
1146 CXX_EXTENSIONS YES)
1147 TARGET_INCLUDE_DIRECTORIES(u8-lut32norm-test PRIVATE src test)
1148 TARGET_LINK_LIBRARIES(u8-lut32norm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1149 ADD_TEST(u8-lut32norm-test u8-lut32norm-test)
1150
1151 ADD_EXECUTABLE(u8-maxpool-test test/u8-maxpool.cc)
1152 SET_TARGET_PROPERTIES(u8-maxpool-test PROPERTIES
1153 CXX_STANDARD 11
1154 CXX_STANDARD_REQUIRED YES
1155 CXX_EXTENSIONS YES)
1156 TARGET_INCLUDE_DIRECTORIES(u8-maxpool-test PRIVATE src test)
1157 TARGET_LINK_LIBRARIES(u8-maxpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1158 ADD_TEST(u8-maxpool-test u8-maxpool-test)
1159
1160 ADD_EXECUTABLE(u8-rmax-test test/u8-rmax.cc)
1161 SET_TARGET_PROPERTIES(u8-rmax-test PROPERTIES
1162 CXX_STANDARD 11
1163 CXX_STANDARD_REQUIRED YES
1164 CXX_EXTENSIONS YES)
1165 TARGET_INCLUDE_DIRECTORIES(u8-rmax-test PRIVATE src test)
1166 TARGET_LINK_LIBRARIES(u8-rmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1167 ADD_TEST(u8-rmax-test u8-rmax-test)
1168
1169 ADD_EXECUTABLE(x32-packx-test test/x32-packx.cc)
1170 SET_TARGET_PROPERTIES(x32-packx-test PROPERTIES
1171 CXX_STANDARD 11
1172 CXX_STANDARD_REQUIRED YES
1173 CXX_EXTENSIONS YES)
1174 TARGET_INCLUDE_DIRECTORIES(x32-packx-test PRIVATE src test)
1175 TARGET_LINK_LIBRARIES(x32-packx-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1176 ADD_TEST(x32-packx-test x32-packx-test)
1177
1178 ADD_EXECUTABLE(x32-pad-test test/x32-pad.cc)
1179 SET_TARGET_PROPERTIES(x32-pad-test PROPERTIES
1180 CXX_STANDARD 11
1181 CXX_STANDARD_REQUIRED YES
1182 CXX_EXTENSIONS YES)
1183 TARGET_INCLUDE_DIRECTORIES(x32-pad-test PRIVATE src test)
1184 TARGET_LINK_LIBRARIES(x32-pad-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1185 ADD_TEST(x32-pad-test x32-pad-test)
1186
1187 ADD_EXECUTABLE(x32-unpool-test test/x32-unpool.cc)
1188 SET_TARGET_PROPERTIES(x32-unpool-test PROPERTIES
1189 CXX_STANDARD 11
1190 CXX_STANDARD_REQUIRED YES
1191 CXX_EXTENSIONS YES)
1192 TARGET_INCLUDE_DIRECTORIES(x32-unpool-test PRIVATE src test)
1193 TARGET_LINK_LIBRARIES(x32-unpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1194 ADD_TEST(x32-unpool-test x32-unpool-test)
1195
1196 ADD_EXECUTABLE(x32-zip-test test/x32-zip.cc)
1197 SET_TARGET_PROPERTIES(x32-zip-test PROPERTIES
1198 CXX_STANDARD 11
1199 CXX_STANDARD_REQUIRED YES
1200 CXX_EXTENSIONS YES)
1201 TARGET_INCLUDE_DIRECTORIES(x32-zip-test PRIVATE src test)
1202 TARGET_LINK_LIBRARIES(x32-zip-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1203 ADD_TEST(x32-zip-test x32-zip-test)
1204
1205 ADD_EXECUTABLE(x8-lut-test test/x8-lut.cc)
1206 SET_TARGET_PROPERTIES(x8-lut-test PROPERTIES
1207 CXX_STANDARD 11
1208 CXX_STANDARD_REQUIRED YES
1209 CXX_EXTENSIONS YES)
1210 TARGET_INCLUDE_DIRECTORIES(x8-lut-test PRIVATE src test)
1211 TARGET_LINK_LIBRARIES(x8-lut-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1212 ADD_TEST(x8-lut-test x8-lut-test)
1213
1214 ADD_EXECUTABLE(x8-zip-test test/x8-zip.cc)
1215 SET_TARGET_PROPERTIES(x8-zip-test PROPERTIES
1216 CXX_STANDARD 11
1217 CXX_STANDARD_REQUIRED YES
1218 CXX_EXTENSIONS YES)
1219 TARGET_INCLUDE_DIRECTORIES(x8-zip-test PRIVATE src test)
1220 TARGET_LINK_LIBRARIES(x8-zip-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1221 ADD_TEST(x8-zip-test x8-zip-test)
1222ENDIF()
1223
1224# ---[ XNNPACK microbenchmarks
1225IF(XNNPACK_BUILD_BENCHMARKS)
1226 # ---[ Build google benchmark
1227 IF(NOT TARGET benchmark)
1228 SET(BENCHMARK_ENABLE_TESTING OFF CACHE BOOL "")
1229 ADD_SUBDIRECTORY(
1230 "${GOOGLEBENCHMARK_SOURCE_DIR}"
1231 "${CONFU_DEPENDENCIES_BINARY_DIR}/googlebenchmark")
1232 ENDIF()
1233
1234 ADD_LIBRARY(bench-utils bench/utils.cc)
1235 SET_TARGET_PROPERTIES(bench-utils PROPERTIES
1236 CXX_STANDARD 11
1237 CXX_STANDARD_REQUIRED YES
1238 CXX_EXTENSIONS NO)
1239 TARGET_INCLUDE_DIRECTORIES(bench-utils PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1240 TARGET_LINK_LIBRARIES(bench-utils PRIVATE cpuinfo)
1241
1242 # ---[ Build end-to-end microbenchmark
1243 ADD_EXECUTABLE(end2end-bench bench/end2end.cc models/mobilenet-v1.cc models/mobilenet-v2.cc)
1244 SET_TARGET_PROPERTIES(end2end-bench PROPERTIES
1245 CXX_STANDARD 11
1246 CXX_STANDARD_REQUIRED YES
1247 CXX_EXTENSIONS NO)
1248 TARGET_INCLUDE_DIRECTORIES(end2end-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1249 TARGET_LINK_LIBRARIES(end2end-bench PRIVATE XNNPACK benchmark)
1250
1251 # ---[ Build operator-level microbenchmarks
1252 ADD_EXECUTABLE(add-bench bench/add.cc)
1253 SET_TARGET_PROPERTIES(add-bench PROPERTIES
1254 CXX_STANDARD 11
1255 CXX_STANDARD_REQUIRED YES
1256 CXX_EXTENSIONS NO)
1257 TARGET_INCLUDE_DIRECTORIES(add-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1258 TARGET_LINK_LIBRARIES(add-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1259
1260 ADD_EXECUTABLE(average-pooling-bench bench/average-pooling.cc)
1261 SET_TARGET_PROPERTIES(average-pooling-bench PROPERTIES
1262 CXX_STANDARD 11
1263 CXX_STANDARD_REQUIRED YES
1264 CXX_EXTENSIONS NO)
1265 TARGET_INCLUDE_DIRECTORIES(average-pooling-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1266 TARGET_LINK_LIBRARIES(average-pooling-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1267
1268 ADD_EXECUTABLE(channel-shuffle-bench bench/channel-shuffle.cc)
1269 SET_TARGET_PROPERTIES(channel-shuffle-bench PROPERTIES
1270 CXX_STANDARD 11
1271 CXX_STANDARD_REQUIRED YES
1272 CXX_EXTENSIONS NO)
1273 TARGET_INCLUDE_DIRECTORIES(channel-shuffle-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1274 TARGET_LINK_LIBRARIES(channel-shuffle-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1275
1276 ADD_EXECUTABLE(convolution-bench bench/convolution.cc)
1277 SET_TARGET_PROPERTIES(convolution-bench PROPERTIES
1278 CXX_STANDARD 11
1279 CXX_STANDARD_REQUIRED YES
1280 CXX_EXTENSIONS NO)
1281 TARGET_INCLUDE_DIRECTORIES(convolution-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1282 TARGET_LINK_LIBRARIES(convolution-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1283
1284 ADD_EXECUTABLE(deconvolution-bench bench/deconvolution.cc)
1285 SET_TARGET_PROPERTIES(deconvolution-bench PROPERTIES
1286 CXX_STANDARD 11
1287 CXX_STANDARD_REQUIRED YES
1288 CXX_EXTENSIONS NO)
1289 TARGET_INCLUDE_DIRECTORIES(deconvolution-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1290 TARGET_LINK_LIBRARIES(deconvolution-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1291
1292 ADD_EXECUTABLE(global-average-pooling-bench bench/global-average-pooling.cc)
1293 SET_TARGET_PROPERTIES(global-average-pooling-bench PROPERTIES
1294 CXX_STANDARD 11
1295 CXX_STANDARD_REQUIRED YES
1296 CXX_EXTENSIONS NO)
1297 TARGET_INCLUDE_DIRECTORIES(global-average-pooling-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1298 TARGET_LINK_LIBRARIES(global-average-pooling-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1299
1300 ADD_EXECUTABLE(max-pooling-bench bench/max-pooling.cc)
1301 SET_TARGET_PROPERTIES(max-pooling-bench PROPERTIES
1302 CXX_STANDARD 11
1303 CXX_STANDARD_REQUIRED YES
1304 CXX_EXTENSIONS NO)
1305 TARGET_INCLUDE_DIRECTORIES(max-pooling-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1306 TARGET_LINK_LIBRARIES(max-pooling-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1307
1308 ADD_EXECUTABLE(sigmoid-bench bench/sigmoid.cc)
1309 SET_TARGET_PROPERTIES(sigmoid-bench PROPERTIES
1310 CXX_STANDARD 11
1311 CXX_STANDARD_REQUIRED YES
1312 CXX_EXTENSIONS NO)
1313 TARGET_INCLUDE_DIRECTORIES(sigmoid-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1314 TARGET_LINK_LIBRARIES(sigmoid-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1315
1316 ADD_EXECUTABLE(softargmax-bench bench/softargmax.cc)
1317 SET_TARGET_PROPERTIES(softargmax-bench PROPERTIES
1318 CXX_STANDARD 11
1319 CXX_STANDARD_REQUIRED YES
1320 CXX_EXTENSIONS NO)
1321 TARGET_INCLUDE_DIRECTORIES(softargmax-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1322 TARGET_LINK_LIBRARIES(softargmax-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1323
1324 # ---[ Build microkernel-level microbenchmarks
1325 ADD_EXECUTABLE(f16-gemm-bench bench/f16-gemm.cc)
1326 SET_TARGET_PROPERTIES(f16-gemm-bench PROPERTIES
1327 CXX_STANDARD 11
1328 CXX_STANDARD_REQUIRED YES
1329 CXX_EXTENSIONS YES)
1330 TARGET_INCLUDE_DIRECTORIES(f16-gemm-bench PRIVATE src)
1331 TARGET_INCLUDE_DIRECTORIES(f16-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1332 TARGET_LINK_LIBRARIES(f16-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1333
1334 ADD_EXECUTABLE(f32-conv-hwc-bench bench/f32-conv-hwc.cc)
1335 SET_TARGET_PROPERTIES(f32-conv-hwc-bench PROPERTIES
1336 CXX_STANDARD 11
1337 CXX_STANDARD_REQUIRED YES
1338 CXX_EXTENSIONS YES)
1339 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc-bench PRIVATE src)
1340 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1341 TARGET_LINK_LIBRARIES(f32-conv-hwc-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1342
1343 ADD_EXECUTABLE(f32-dwconv-spchw-bench bench/f32-dwconv-spchw.cc)
1344 SET_TARGET_PROPERTIES(f32-dwconv-spchw-bench PROPERTIES
1345 CXX_STANDARD 11
1346 CXX_STANDARD_REQUIRED YES
1347 CXX_EXTENSIONS YES)
1348 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-spchw-bench PRIVATE src)
1349 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-spchw-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1350 TARGET_LINK_LIBRARIES(f32-dwconv-spchw-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1351
1352 ADD_EXECUTABLE(f32-dwconv-bench bench/f32-dwconv.cc)
1353 SET_TARGET_PROPERTIES(f32-dwconv-bench PROPERTIES
1354 CXX_STANDARD 11
1355 CXX_STANDARD_REQUIRED YES
1356 CXX_EXTENSIONS YES)
1357 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-bench PRIVATE src)
1358 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1359 TARGET_LINK_LIBRARIES(f32-dwconv-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1360
1361 ADD_EXECUTABLE(f32-gemm-bench bench/f32-gemm.cc)
1362 SET_TARGET_PROPERTIES(f32-gemm-bench PROPERTIES
1363 CXX_STANDARD 11
1364 CXX_STANDARD_REQUIRED YES
1365 CXX_EXTENSIONS YES)
1366 TARGET_INCLUDE_DIRECTORIES(f32-gemm-bench PRIVATE src)
1367 TARGET_INCLUDE_DIRECTORIES(f32-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1368 TARGET_LINK_LIBRARIES(f32-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1369
1370 ADD_EXECUTABLE(f32-igemm-bench bench/f32-igemm.cc)
1371 SET_TARGET_PROPERTIES(f32-igemm-bench PROPERTIES
1372 CXX_STANDARD 11
1373 CXX_STANDARD_REQUIRED YES
1374 CXX_EXTENSIONS YES)
1375 TARGET_INCLUDE_DIRECTORIES(f32-igemm-bench PRIVATE src)
1376 TARGET_INCLUDE_DIRECTORIES(f32-igemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1377 TARGET_LINK_LIBRARIES(f32-igemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1378
1379 ADD_EXECUTABLE(f32-im2col-gemm-bench bench/f32-im2col-gemm.cc src/im2col.c)
1380 SET_TARGET_PROPERTIES(f32-im2col-gemm-bench PROPERTIES
1381 C_STANDARD 99
1382 C_STANDARD_REQUIRED YES
1383 C_EXTENSIONS NO
1384 CXX_STANDARD 11
1385 CXX_STANDARD_REQUIRED YES
1386 CXX_EXTENSIONS YES)
1387 TARGET_INCLUDE_DIRECTORIES(f32-im2col-gemm-bench PRIVATE src)
1388 TARGET_INCLUDE_DIRECTORIES(f32-im2col-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1389 TARGET_LINK_LIBRARIES(f32-im2col-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1390
1391 ADD_EXECUTABLE(f32-rmax-bench bench/f32-rmax.cc)
1392 SET_TARGET_PROPERTIES(f32-rmax-bench PROPERTIES
1393 CXX_STANDARD 11
1394 CXX_STANDARD_REQUIRED YES
1395 CXX_EXTENSIONS YES)
1396 TARGET_INCLUDE_DIRECTORIES(f32-rmax-bench PRIVATE src)
1397 TARGET_INCLUDE_DIRECTORIES(f32-rmax-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1398 TARGET_LINK_LIBRARIES(f32-rmax-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1399
1400 ADD_EXECUTABLE(f32-spmm-bench bench/f32-spmm.cc)
1401 SET_TARGET_PROPERTIES(f32-spmm-bench PROPERTIES
1402 CXX_STANDARD 11
1403 CXX_STANDARD_REQUIRED YES
1404 CXX_EXTENSIONS YES)
1405 TARGET_INCLUDE_DIRECTORIES(f32-spmm-bench PRIVATE src)
1406 TARGET_INCLUDE_DIRECTORIES(f32-spmm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1407 TARGET_LINK_LIBRARIES(f32-spmm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1408
1409 ADD_EXECUTABLE(q8-gemm-bench bench/q8-gemm.cc)
1410 SET_TARGET_PROPERTIES(q8-gemm-bench PROPERTIES
1411 CXX_STANDARD 11
1412 CXX_STANDARD_REQUIRED YES
1413 CXX_EXTENSIONS YES)
1414 TARGET_INCLUDE_DIRECTORIES(q8-gemm-bench PRIVATE src)
1415 TARGET_INCLUDE_DIRECTORIES(q8-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1416 TARGET_LINK_LIBRARIES(q8-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1417ENDIF()