blob: eb0d1901eb0c1f830db37a06fa5e784931df6a8e [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
Erich Elsen563df5f2019-10-23 08:02:21 -0700170 src/f32-conv-hwc2spchw/3x3s2p1c3x4-scalar-1x1.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700171 src/f32-igemm/1x4-scalar.c
172 src/f32-igemm/2x4-scalar.c
173 src/f32-igemm/4x2-scalar.c
174 src/f32-igemm/4x4-scalar.c
175 src/f32-dwconv/up1x25-scalar.c
176 src/f32-dwconv/up1x4-scalar.c
177 src/f32-dwconv/up1x9-scalar.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800178 src/f32-dwconv/up1x25-scalar-acc2.c
179 src/f32-dwconv/up1x4-scalar-acc2.c
180 src/f32-dwconv/up1x9-scalar-acc2.c
181 src/f32-dwconv/up2x25-scalar.c
182 src/f32-dwconv/up2x4-scalar.c
183 src/f32-dwconv/up2x9-scalar.c
184 src/f32-dwconv/up2x25-scalar-acc2.c
185 src/f32-dwconv/up2x4-scalar-acc2.c
186 src/f32-dwconv/up2x9-scalar-acc2.c
Erich Elsen0cc2c532019-10-15 04:44:18 -0700187 src/f32-dwconv-spchw/3x3p1-scalar.c
Erich Elsenac4de802019-10-16 04:35:30 -0700188 src/f32-dwconv-spchw/3x3s2p1-scalar.c
Erich Elsen34dc2c02019-10-16 05:11:41 -0700189 src/f32-gavgpool-spchw/scalar-x1.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700190 src/f32-gavgpool/mp7p7q-scalar.c
191 src/f32-gavgpool/up7-scalar.c
192 src/f32-gemm/1x4-scalar.c
193 src/f32-gemm/2x4-scalar.c
194 src/f32-gemm/4x2-scalar.c
195 src/f32-gemm/4x4-scalar.c
196 src/f32-gemminc/1x4-scalar.c
197 src/f32-gemminc/2x4-scalar.c
198 src/f32-gemminc/4x4-scalar.c
199 src/f32-hswish/scalar.c
200 src/f32-maxpool/9p8q-scalar.c
201 src/f32-pavgpool/mp9p8q-scalar.c
202 src/f32-pavgpool/up9-scalar.c
203 src/f32-ppmm/2x4-scalar.c
204 src/f32-ppmm/3x3-scalar.c
205 src/f32-ppmm/4x2-scalar.c
206 src/f32-ppmm/4x4-scalar.c
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800207 src/f32-prelu/scalar-2x1.c
208 src/f32-prelu/scalar-2x4.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700209 src/f32-rmax/scalar.c
210 src/f32-spmm/1x1-scalar-pipelined.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700211 src/f32-spmm/1x1-scalar.c
212 src/f32-spmm/2x1-scalar-pipelined.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700213 src/f32-spmm/2x1-scalar.c
214 src/f32-spmm/4x1-scalar-pipelined.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700215 src/f32-spmm/4x1-scalar.c
216 src/f32-spmm/8x1-scalar-pipelined.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700217 src/f32-spmm/8x1-scalar.c
Erich Elsenc6afd9b2019-10-24 16:10:53 -0700218 src/f32-spmm/8x2-scalar.c
219 src/f32-spmm/8x4-scalar.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700220 src/f32-vadd/scalar.c
221 src/f32-vmul/scalar.c
Marat Dukhan49e6ee92019-11-06 15:55:29 -0800222 src/f32-vmulcaddc/c1-scalar-2x.c
223 src/f32-vmulcaddc/c2-scalar-2x.c
224 src/f32-vmulcaddc/c4-scalar-2x.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700225 src/f32-vsub/scalar.c
226 src/q8-avgpool/mp9p8q-scalar.c
227 src/q8-avgpool/up9-scalar.c
228 src/q8-igemm/2x2-scalar.c
229 src/q8-dwconv/up1x9-scalar.c
230 src/q8-gavgpool/mp7p7q-scalar.c
231 src/q8-gavgpool/up7-scalar.c
232 src/q8-gemm/2x2-scalar.c
233 src/q8-vadd/scalar.c
234 src/u8-clamp/scalar.c
235 src/u8-lut32norm/scalar.c
236 src/u8-maxpool/9p8q-scalar.c
237 src/u8-rmax/scalar.c
238 src/x32-packx/x2-scalar.c
239 src/x32-packx/x3-scalar.c
240 src/x32-packx/x4-scalar.c
241 src/x32-pad/x2-scalar.c
242 src/x32-unpool/scalar.c
243 src/x32-zip/x2-scalar.c
244 src/x32-zip/x3-scalar.c
245 src/x32-zip/x4-scalar.c
246 src/x32-zip/xm-scalar.c
247 src/x8-lut/scalar.c
248 src/x8-zip/x2-scalar.c
249 src/x8-zip/x3-scalar.c
250 src/x8-zip/x4-scalar.c
251 src/x8-zip/xm-scalar.c)
252
253SET(XNNPACK_PSIMD_MICROKERNEL_SRCS
254 src/f32-argmaxpool/mp9p8q-psimd.c
255 src/f32-argmaxpool/up4-psimd.c
256 src/f32-argmaxpool/up9-psimd.c
257 src/f32-avgpool/mp9p8q-psimd.c
258 src/f32-avgpool/up9-psimd.c
259 src/f32-clamp/psimd.c
260 src/f32-igemm/1x8-psimd-loadsplat.c
261 src/f32-igemm/1x8-psimd-splat.c
262 src/f32-igemm/1x8s4-psimd.c
263 src/f32-igemm/4x2c4-psimd.c
264 src/f32-igemm/4x8-psimd-loadsplat.c
265 src/f32-igemm/4x8-psimd-splat.c
266 src/f32-igemm/4x8s4-psimd.c
267 src/f32-igemm/6x8-psimd-loadsplat.c
268 src/f32-igemm/6x8-psimd-splat.c
269 src/f32-igemm/6x8s4-psimd.c
270 src/f32-dwconv/up4x25-psimd.c
271 src/f32-dwconv/up4x4-psimd.c
272 src/f32-dwconv/up4x9-psimd.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800273 src/f32-dwconv/up4x25-psimd-acc2.c
274 src/f32-dwconv/up4x4-psimd-acc2.c
275 src/f32-dwconv/up4x9-psimd-acc2.c
276 src/f32-dwconv/up8x25-psimd.c
277 src/f32-dwconv/up8x4-psimd.c
278 src/f32-dwconv/up8x9-psimd.c
279 src/f32-dwconv/up8x25-psimd-acc2.c
280 src/f32-dwconv/up8x4-psimd-acc2.c
281 src/f32-dwconv/up8x9-psimd-acc2.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700282 src/f32-gavgpool/mp7p7q-psimd.c
283 src/f32-gavgpool/up7-psimd.c
284 src/f32-gemm/1x8-psimd-loadsplat.c
285 src/f32-gemm/1x8-psimd-splat.c
286 src/f32-gemm/1x8s4-psimd.c
287 src/f32-gemm/4x8-psimd-loadsplat.c
288 src/f32-gemm/4x8-psimd-splat.c
289 src/f32-gemm/4x8s4-psimd.c
290 src/f32-gemm/6x8-psimd-loadsplat.c
291 src/f32-gemm/6x8-psimd-splat.c
292 src/f32-gemm/6x8s4-psimd.c
293 src/f32-gemminc/1x8-psimd-loadsplat.c
294 src/f32-gemminc/1x8-psimd-splat.c
295 src/f32-gemminc/1x8s4-psimd.c
296 src/f32-gemminc/4x8-psimd-loadsplat.c
297 src/f32-gemminc/4x8-psimd-splat.c
298 src/f32-gemminc/4x8s4-psimd.c
299 src/f32-gemminc/6x8-psimd-loadsplat.c
300 src/f32-gemminc/6x8-psimd-splat.c
301 src/f32-gemminc/6x8s4-psimd.c
302 src/f32-hswish/psimd.c
303 src/f32-maxpool/9p8q-psimd.c
304 src/f32-pavgpool/mp9p8q-psimd.c
305 src/f32-pavgpool/up9-psimd.c
306 src/f32-ppmm/4x8-psimd.c
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800307 src/f32-prelu/psimd-2x4.c
308 src/f32-prelu/psimd-2x8.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700309 src/f32-vadd/psimd.c
310 src/f32-vmul/psimd.c
Marat Dukhan49e6ee92019-11-06 15:55:29 -0800311 src/f32-vmulcaddc/c4-psimd-2x.c
312 src/f32-vmulcaddc/c8-psimd-2x.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700313 src/f32-vsub/psimd.c
314 src/x32-packx/x4-psimd.c
315 src/x32-pad/x2-psimd.c
316 src/x32-unpool/psimd.c
317 src/x32-zip/x2-psimd.c
318 src/x32-zip/x3-psimd.c
319 src/x32-zip/x4-psimd.c
320 src/x32-zip/xm-psimd.c)
321
322SET(XNNPACK_NEON_MICROKERNEL_SRCS
323 src/f32-avgpool/mp9p8q-neon.c
324 src/f32-avgpool/up9-neon.c
325 src/f32-clamp/neon.c
326 src/f32-igemm/1x8-neon-ld64.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700327 src/f32-igemm/4x2-neon-ld64.c
328 src/f32-igemm/4x4-neon-ld64.c
329 src/f32-igemm/4x8-neon-ld128.c
330 src/f32-igemm/4x8-neon-ld64.c
331 src/f32-igemm/6x8-neon-ld64.c
332 src/f32-dwconv/up4x9-neon.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800333 src/f32-dwconv/up4x9-neon-acc2.c
334 src/f32-dwconv/up8x9-neon.c
335 src/f32-dwconv/up8x9-neon-acc2.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700336 src/f32-gavgpool-spchw/neon-x4.c
337 src/f32-gavgpool/mp7p7q-neon.c
338 src/f32-gavgpool/up7-neon.c
339 src/f32-gemm/1x8-neon-ld64.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700340 src/f32-gemm/4x2-neon-ld64.c
341 src/f32-gemm/4x8-neon-ld128.c
342 src/f32-gemm/4x8-neon-ld64.c
343 src/f32-gemm/5x8-neon-ld64.c
344 src/f32-gemm/6x8-neon-ld64.c
345 src/f32-gemminc/1x8-neon-ld64.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700346 src/f32-gemminc/4x8-neon-ld128.c
347 src/f32-gemminc/4x8-neon-ld64.c
348 src/f32-gemminc/5x8-neon-ld64.c
349 src/f32-gemminc/6x8-neon-ld64.c
350 src/f32-hswish/neon.c
351 src/f32-pavgpool/mp9p8q-neon.c
352 src/f32-pavgpool/up9-neon.c
353 src/f32-ppmm/4x8-neon.c
354 src/f32-ppmm/8x8-neon.c
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800355 src/f32-prelu/neon-2x4.c
356 src/f32-prelu/neon-2x8.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700357 src/f32-rmax/neon.c
Marat Dukhan49e6ee92019-11-06 15:55:29 -0800358 src/f32-vmulcaddc/c4-neon-2x.c
359 src/f32-vmulcaddc/c8-neon-2x.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700360 src/q8-avgpool/mp9p8q-neon.c
361 src/q8-avgpool/up9-neon.c
362 src/q8-igemm/4x8-neon.c
363 src/q8-igemm/8x8-neon.c
364 src/q8-dwconv/up8x9-neon.c
365 src/q8-gavgpool/mp7p7q-neon.c
366 src/q8-gavgpool/up7-neon.c
367 src/q8-gemm/4x8-neon.c
368 src/q8-gemm/8x8-neon.c
369 src/q8-vadd/neon.c
370 src/u8-clamp/neon.c
371 src/u8-maxpool/9p8q-neon.c
372 src/u8-rmax/neon.c
373 src/x32-packx/x4-neon-st4.c
374 src/x32-pad/x2-neon.c
375 src/x32-zip/x2-neon.c
376 src/x32-zip/x3-neon.c
377 src/x32-zip/x4-neon.c
378 src/x32-zip/xm-neon.c
379 src/x8-zip/x2-neon.c
380 src/x8-zip/x3-neon.c
381 src/x8-zip/x4-neon.c
382 src/x8-zip/xm-neon.c)
383
384SET(XNNPACK_NEONFMA_MICROKERNEL_SRCS
Frank Barcharddb45b6a2019-10-09 16:42:45 -0700385 src/f32-igemm/1x8-neonfma-ld64.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700386 src/f32-igemm/4x2-neonfma-ld64.c
387 src/f32-igemm/4x4-neonfma-ld64.c
388 src/f32-igemm/4x8-neonfma-ld128.c
389 src/f32-igemm/4x8-neonfma-ld64.c
390 src/f32-igemm/6x8-neonfma-ld64.c
391 src/f32-dwconv/up4x9-neonfma.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800392 src/f32-dwconv/up4x9-neonfma-acc2.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700393 src/f32-dwconv/up8x9-neonfma.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800394 src/f32-dwconv/up8x9-neonfma-acc2.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700395 src/f32-gemm/1x8-neonfma-ld64.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700396 src/f32-gemm/4x2-neonfma-ld64.c
397 src/f32-gemm/4x8-neonfma-ld128.c
398 src/f32-gemm/4x8-neonfma-ld64.c
399 src/f32-gemm/5x8-neonfma-ld64.c
400 src/f32-gemm/6x8-neonfma-ld64.c
401 src/f32-gemminc/1x8-neonfma-ld64.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700402 src/f32-gemminc/4x8-neonfma-ld128.c
403 src/f32-gemminc/4x8-neonfma-ld64.c
404 src/f32-gemminc/5x8-neonfma-ld64.c
405 src/f32-gemminc/6x8-neonfma-ld64.c
406 src/f32-hswish/neonfma.c
407 src/f32-ppmm/4x8-neonfma.c
408 src/f32-ppmm/8x8-neonfma.c
Marat Dukhan49e6ee92019-11-06 15:55:29 -0800409 src/f32-vmulcaddc/c4-neonfma-2x.c
410 src/f32-vmulcaddc/c8-neonfma-2x.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700411
412SET(AARCH64_XNNPACK_NEONFMA_MICROKERNEL_SRCS
413 src/f32-conv-hwc/3x3s2p1c3x4-neonfma-2x2.c
414 src/f32-conv-hwc/3x3s2p1c3x8-neonfma-2x2.c
415 src/f32-conv-hwc2spchw/3x3s2p1c3x4-neonfma-2x2.c
416 src/f32-dwconv-spchw/3x3p1-neonfma.c
417 src/f32-dwconv-spchw/5x5p2-neonfma.c
418 src/f32-dwconv-spchw/3x3s2p1-neonfma.c
419 src/f32-dwconv-spchw/5x5s2p2-neonfma.c
420 src/f32-spmm/12x1-neonfma.c
421 src/f32-spmm/12x2-neonfma.c
422 src/f32-spmm/12x4-neonfma.c
423 src/f32-spmm/16x1-neonfma-pipelined.c
424 src/f32-spmm/16x1-neonfma-unroll2.c
425 src/f32-spmm/16x1-neonfma.c
426 src/f32-spmm/16x2-neonfma.c
427 src/f32-spmm/16x4-neonfma.c
428 src/f32-spmm/4x1-neonfma-pipelined.c
429 src/f32-spmm/4x1-neonfma-unroll2.c
430 src/f32-spmm/4x1-neonfma.c
431 src/f32-spmm/4x2-neonfma.c
432 src/f32-spmm/4x4-neonfma.c
433 src/f32-spmm/8x1-neonfma-pipelined.c
434 src/f32-spmm/8x1-neonfma-unroll2.c
435 src/f32-spmm/8x1-neonfma.c
436 src/f32-spmm/8x2-neonfma.c
437 src/f32-spmm/8x4-neonfma.c)
438
439SET(XNNPACK_AARCH64_NEONFP16ARITH_MICROKERNEL_SRCS
440 src/f16-gemm/4x8-neonfp16arith-ld64.c
441 src/f16-gemm/6x8-neonfp16arith-ld64.c
442 src/f16-gemm/8x8-neonfp16arith-ld64.c)
443
444SET(XNNPACK_SSE_MICROKERNEL_SRCS
445 src/f32-avgpool/mp9p8q-sse.c
446 src/f32-avgpool/up9-sse.c
447 src/f32-clamp/sse.c
448 src/f32-igemm/1x8-sse-dup.c
449 src/f32-igemm/1x8-sse-load1.c
450 src/f32-igemm/1x8s4-sse.c
451 src/f32-igemm/4x2c4-sse.c
452 src/f32-igemm/4x8-sse-dup.c
453 src/f32-igemm/4x8-sse-load1.c
454 src/f32-igemm/4x8s4-sse.c
455 src/f32-dwconv/up4x25-sse.c
456 src/f32-dwconv/up4x4-sse.c
457 src/f32-dwconv/up4x9-sse.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800458 src/f32-dwconv/up4x25-sse-acc2.c
459 src/f32-dwconv/up4x4-sse-acc2.c
460 src/f32-dwconv/up4x9-sse-acc2.c
461 src/f32-dwconv/up8x25-sse.c
462 src/f32-dwconv/up8x4-sse.c
463 src/f32-dwconv/up8x9-sse.c
464 src/f32-dwconv/up8x25-sse-acc2.c
465 src/f32-dwconv/up8x4-sse-acc2.c
466 src/f32-dwconv/up8x9-sse-acc2.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700467 src/f32-gavgpool-spchw/sse-x4.c
468 src/f32-gavgpool/mp7p7q-sse.c
469 src/f32-gavgpool/up7-sse.c
470 src/f32-gemm/1x8-sse-dup.c
471 src/f32-gemm/1x8-sse-load1.c
472 src/f32-gemm/1x8s4-sse.c
473 src/f32-gemm/4x8-sse-dup.c
474 src/f32-gemm/4x8-sse-load1.c
475 src/f32-gemm/4x8s4-sse.c
476 src/f32-gemminc/1x8-sse-dup.c
477 src/f32-gemminc/1x8-sse-load1.c
478 src/f32-gemminc/1x8s4-sse.c
479 src/f32-gemminc/4x8-sse-dup.c
480 src/f32-gemminc/4x8-sse-load1.c
481 src/f32-gemminc/4x8s4-sse.c
482 src/f32-hswish/sse.c
483 src/f32-maxpool/9p8q-sse.c
484 src/f32-pavgpool/mp9p8q-sse.c
485 src/f32-pavgpool/up9-sse.c
486 src/f32-dwconv-spchw/3x3p1-sse.c
487 src/f32-dwconv-spchw/3x3s2p1-sse.c
488 src/f32-ppmm/4x8-sse.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700489 src/f32-rmax/sse.c
490 src/f32-spmm/4x1-sse.c
491 src/f32-spmm/8x1-sse.c
492 src/f32-vadd/sse.c
493 src/f32-vmul/sse.c
Marat Dukhan49e6ee92019-11-06 15:55:29 -0800494 src/f32-vmulcaddc/c4-sse-2x.c
495 src/f32-vmulcaddc/c8-sse-2x.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700496 src/f32-vsub/sse.c
497 src/x32-packx/x4-sse.c)
498
499SET(XNNPACK_SSE2_MICROKERNEL_SRCS
500 src/f32-argmaxpool/mp9p8q-sse2.c
501 src/f32-argmaxpool/up4-sse2.c
502 src/f32-argmaxpool/up9-sse2.c
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800503 src/f32-prelu/sse2-2x4.c
504 src/f32-prelu/sse2-2x8.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700505 src/q8-avgpool/mp9p8q-sse2.c
506 src/q8-avgpool/up9-sse2.c
507 src/q8-igemm/4x4c2-sse2.c
508 src/q8-dwconv/up8x9-sse2.c
509 src/q8-gavgpool/mp7p7q-sse2.c
510 src/q8-gavgpool/up7-sse2.c
511 src/q8-gemm/2x4c8-sse2.c
512 src/q8-gemm/4x4c2-sse2.c
513 src/q8-vadd/sse2.c
514 src/u8-clamp/sse2.c
515 src/u8-maxpool/9p8q-sse2.c
516 src/u8-rmax/sse2.c
517 src/x32-pad/x2-sse2.c
518 src/x32-zip/x2-sse2.c
519 src/x32-zip/x3-sse2.c
520 src/x32-zip/x4-sse2.c
521 src/x32-zip/xm-sse2.c
522 src/x8-zip/x2-sse2.c
523 src/x8-zip/x3-sse2.c
524 src/x8-zip/x4-sse2.c
525 src/x8-zip/xm-sse2.c)
526
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800527SET(XNNPACK_SSE41_MICROKERNEL_SRCS
528 src/f32-prelu/sse41-2x4.c
529 src/f32-prelu/sse41-2x8.c)
530
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700531SET(XNNPACK_AVX_MICROKERNEL_SRCS
Marat Dukhana5977df2019-10-30 22:02:15 -0700532 src/f32-rmax/avx.c
533 src/f32-vscale/avx-unroll32.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700534
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700535SET(XNNPACK_AVX2_MICROKERNEL_SRCS
Marat Dukhan97579532019-10-18 16:40:39 -0700536 src/f32-raddexpminusmax/avx2-p5-unroll64.c
Marat Dukhan6f8d4d32019-10-25 17:07:09 -0700537 src/f32-raddextexp/avx2-p5-unroll64.c
Marat Dukhan97579532019-10-18 16:40:39 -0700538 src/f32-raddstoreexpminusmax/avx2-p5-unroll64.c
539 src/f32-vscaleexpminusmax/avx2-p5-unroll64.c
Marat Dukhan6f8d4d32019-10-25 17:07:09 -0700540 src/f32-vscaleextexp/avx2-p5-unroll64.c
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700541 src/math/exp-avx2-p5.c
542 src/math/exp-avx2-perm-p3.c
Marat Dukhan515c9772019-10-17 18:07:57 -0700543 src/math/exp-avx2-perm-p4.c
544 src/math/expminus-avx2-p5.c)
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700545
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700546SET(XNNPACK_AVX512F_MICROKERNEL_SRCS
Marat Dukhan97579532019-10-18 16:40:39 -0700547 src/f32-raddexpminusmax/avx512f-p5-scalef-unroll128.c
Marat Dukhan6f8d4d32019-10-25 17:07:09 -0700548 src/f32-raddextexp/avx512f-p5-scalef-unroll128.c
Marat Dukhan97579532019-10-18 16:40:39 -0700549 src/f32-raddstoreexpminusmax/avx512f-p5-scalef-unroll128.c
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700550 src/f32-rmax/avx512f.c
Marat Dukhana5977df2019-10-30 22:02:15 -0700551 src/f32-vscale/avx512f-unroll64.c
Marat Dukhan97579532019-10-18 16:40:39 -0700552 src/f32-vscaleexpminusmax/avx512f-p5-scalef-unroll128.c
Marat Dukhan6f8d4d32019-10-25 17:07:09 -0700553 src/f32-vscaleextexp/avx512f-p5-scalef-unroll128.c
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700554 src/math/exp-avx512f-p5-scalef.c
555 src/math/exp-avx512f-p5.c
Marat Dukhanfeb49232019-10-28 11:03:31 -0700556 src/math/exp-avx512f-perm-p3.c
557 src/math/exp-avx512f-perm2-p2.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700558
559SET(XNNPACK_AARCH32_ASM_MICROKERNEL_SRCS
560 src/q8-dwconv/up8x9-aarch32-neon.S)
561
562SET(XNNPACK_AARCH64_ASM_MICROKERNEL_SRCS
563 src/f32-dwconv/up4x9-aarch64-neonfma-cortex-a55.S
564 src/f32-dwconv/up4x9-aarch64-neonfma.S
565 src/f32-gemm/1x12-aarch64-neonfma-cortex-a53.S
Frank Barchard21be34f2019-10-09 19:32:19 -0700566 src/f32-gemm/1x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700567 src/f32-gemm/1x8-aarch64-neonfma-cortex-a57.S
568 src/f32-gemm/1x8-aarch64-neonfma-cortex-a75.S
569 src/f32-gemm/4x12-aarch64-neonfma-cortex-a53.S
Marat Dukhan8e6e9972019-10-31 14:15:31 -0700570 src/f32-gemm/4x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700571 src/f32-gemm/4x8-aarch64-neonfma-cortex-a57.S
572 src/f32-gemm/4x8-aarch64-neonfma-cortex-a75.S
573 src/f32-gemm/4x8-aarch64-neonfma-ld128.S
574 src/f32-gemm/4x8-aarch64-neonfma-ld64.S
575 src/f32-gemm/5x8-aarch64-neonfma-cortex-a75.S
Marat Dukhan8e6e9972019-10-31 14:15:31 -0700576 src/f32-gemm/6x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700577 src/f32-gemm/6x8-aarch64-neonfma-cortex-a57.S
578 src/f32-gemm/6x8-aarch64-neonfma-cortex-a73.S
579 src/f32-gemm/6x8-aarch64-neonfma-cortex-a75.S
580 src/f32-gemm/6x8-aarch64-neonfma-ld128.S
581 src/f32-gemm/6x8-aarch64-neonfma-ld64.S
582 src/f32-gemminc/1x12-aarch64-neonfma-cortex-a53.S
Frank Barchard21be34f2019-10-09 19:32:19 -0700583 src/f32-gemminc/1x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700584 src/f32-gemminc/1x8-aarch64-neonfma-cortex-a57.S
585 src/f32-gemminc/1x8-aarch64-neonfma-cortex-a75.S
586 src/f32-gemminc/4x12-aarch64-neonfma-cortex-a53.S
Marat Dukhan8e6e9972019-10-31 14:15:31 -0700587 src/f32-gemminc/4x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700588 src/f32-gemminc/4x8-aarch64-neonfma-cortex-a57.S
589 src/f32-gemminc/4x8-aarch64-neonfma-cortex-a75.S
590 src/f32-gemminc/4x8-aarch64-neonfma-ld128.S
591 src/f32-gemminc/4x8-aarch64-neonfma-ld64.S
592 src/f32-gemminc/5x8-aarch64-neonfma-cortex-a75.S
Marat Dukhan8e6e9972019-10-31 14:15:31 -0700593 src/f32-gemminc/6x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700594 src/f32-gemminc/6x8-aarch64-neonfma-cortex-a57.S
595 src/f32-gemminc/6x8-aarch64-neonfma-cortex-a73.S
596 src/f32-gemminc/6x8-aarch64-neonfma-cortex-a75.S
597 src/f32-gemminc/6x8-aarch64-neonfma-ld128.S
598 src/f32-gemminc/6x8-aarch64-neonfma-ld64.S
599 src/f32-igemm/1x12-aarch64-neonfma-cortex-a53.S
Frank Barchard21be34f2019-10-09 19:32:19 -0700600 src/f32-igemm/1x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700601 src/f32-igemm/1x8-aarch64-neonfma-cortex-a57.S
602 src/f32-igemm/1x8-aarch64-neonfma-cortex-a75.S
603 src/f32-igemm/4x12-aarch64-neonfma-cortex-a53.S
604 src/f32-igemm/4x8-aarch64-neonfma-cortex-a75.S
605 src/f32-igemm/5x8-aarch64-neonfma-cortex-a75.S
Marat Dukhan8e6e9972019-10-31 14:15:31 -0700606 src/f32-igemm/6x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700607 src/f32-igemm/6x8-aarch64-neonfma-cortex-a57.S
608 src/f32-igemm/6x8-aarch64-neonfma-cortex-a73.S
609 src/f32-igemm/6x8-aarch64-neonfma-cortex-a75.S)
610
611SET(XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SCALAR_MICROKERNEL_SRCS})
612IF(NOT CMAKE_SYSTEM_NAME STREQUAL "Emscripten")
613 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_PSIMD_MICROKERNEL_SRCS})
614ENDIF()
615IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^armv[5-8]" OR IOS_ARCH MATCHES "^armv7")
616 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEON_MICROKERNEL_SRCS})
617 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEONFMA_MICROKERNEL_SRCS})
618 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AARCH32_ASM_MICROKERNEL_SRCS})
619ENDIF()
620IF(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64" OR IOS_ARCH MATCHES "^arm64.*")
621 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEON_MICROKERNEL_SRCS})
622 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEONFMA_MICROKERNEL_SRCS})
623 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${AARCH64_XNNPACK_NEONFMA_MICROKERNEL_SRCS})
624 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AARCH64_NEONFP16ARITH_MICROKERNEL_SRCS})
625 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AARCH64_ASM_MICROKERNEL_SRCS})
626ENDIF()
627IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^(i[3-6]86|x86_64)$" OR IOS_ARCH MATCHES "^(i386|x86_64)$")
628 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SSE_MICROKERNEL_SRCS})
629 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SSE2_MICROKERNEL_SRCS})
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800630 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SSE41_MICROKERNEL_SRCS})
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700631 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AVX_MICROKERNEL_SRCS})
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700632 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AVX2_MICROKERNEL_SRCS})
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700633 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AVX512F_MICROKERNEL_SRCS})
634ENDIF()
635
636IF(XNNPACK_LIBRARY_TYPE STREQUAL "default")
637 ADD_LIBRARY(XNNPACK ${XNNPACK_COLD_SRCS} ${XNNPACK_HOT_SRCS} ${XNNPACK_MICROKERNEL_SRCS})
638ELSEIF(XNNPACK_LIBRARY_TYPE STREQUAL "shared")
639 ADD_LIBRARY(XNNPACK SHARED ${XNNPACK_COLD_SRCS} ${XNNPACK_HOT_SRCS} ${XNNPACK_MICROKERNEL_SRCS})
640ELSEIF(XNNPACK_LIBRARY_TYPE STREQUAL "static")
641 ADD_LIBRARY(XNNPACK STATIC ${XNNPACK_COLD_SRCS} ${XNNPACK_HOT_SRCS} ${XNNPACK_MICROKERNEL_SRCS})
642ELSE()
643 MESSAGE(FATAL_ERROR "Unsupported XNNPACK library type \"${XNNPACK_LIBRARY_TYPE}\". Must be \"static\", \"shared\", or \"default\"")
644ENDIF()
645SET_TARGET_PROPERTIES(XNNPACK PROPERTIES
646 C_STANDARD 99
647 C_EXTENSIONS YES)
648IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^armv[5-8]" OR IOS_ARCH MATCHES "^armv7")
649 SET_PROPERTY(SOURCE ${XNNPACK_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -marm ")
650 SET_PROPERTY(SOURCE ${XNNPACK_PSIMD_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfpu=neon ")
651 SET_PROPERTY(SOURCE ${XNNPACK_NEON_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfpu=neon ")
652 SET_PROPERTY(SOURCE ${XNNPACK_NEONFMA_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfpu=neon-vfpv4 ")
653 IF(IOS)
654 SET_PROPERTY(SOURCE ${XNNPACK_AARCH32_ASM_UKERNELS} APPEND_STRING PROPERTY COMPILE_FLAGS " -arch ${IOS_ARCH} ")
655 ENDIF()
656ENDIF()
657IF(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64" OR IOS_ARCH MATCHES "^arm64.*")
658 SET_PROPERTY(SOURCE ${XNNPACK_AARCH64_NEONFP16ARITH_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -march=armv8.2-a+fp16 ")
659 IF(IOS)
660 SET_PROPERTY(SOURCE ${XNNPACK_AARCH64_ASM_UKERNELS} APPEND_STRING PROPERTY COMPILE_FLAGS " -arch ${IOS_ARCH} ")
661 ENDIF()
662ENDIF()
663IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^(i[3-6]86|x86_64)$" OR IOS_ARCH MATCHES "^(i386|x86_64)$")
664 SET_PROPERTY(SOURCE ${XNNPACK_SSE_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -msse ")
665 SET_PROPERTY(SOURCE ${XNNPACK_SSE2_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -msse2 ")
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800666 SET_PROPERTY(SOURCE ${XNNPACK_SSE41_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -msse4.1 ")
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700667 SET_PROPERTY(SOURCE ${XNNPACK_AVX_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mavx ")
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700668 SET_PROPERTY(SOURCE ${XNNPACK_AVX2_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfma -mavx2 ")
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700669 SET_PROPERTY(SOURCE ${XNNPACK_AVX512F_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mavx512f ")
670ENDIF()
671IF(CMAKE_BUILD_TYPE STREQUAL "Debug")
672 TARGET_COMPILE_DEFINITIONS(XNNPACK PRIVATE XNN_LOG_LEVEL=5)
673ELSE()
674 TARGET_COMPILE_DEFINITIONS(XNNPACK PRIVATE XNN_LOG_LEVEL=0)
675 SET_PROPERTY(SOURCE ${XNNPACK_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -O2 ")
676 SET_PROPERTY(SOURCE ${XNNPACK_HOT_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -O2 ")
677 SET_PROPERTY(SOURCE ${XNNPACK_COLD_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -Os ")
678ENDIF()
679IF(NOT CMAKE_SYSTEM_NAME STREQUAL "Emscripten")
680 SET_PROPERTY(SOURCE ${XNNPACK_PSIMD_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -O3 -ffast-math ")
681ENDIF()
682
683TARGET_INCLUDE_DIRECTORIES(XNNPACK PUBLIC include)
684TARGET_INCLUDE_DIRECTORIES(XNNPACK PRIVATE src)
685SET_TARGET_PROPERTIES(XNNPACK PROPERTIES PUBLIC_HEADER include/XNNPACK.h)
686
687# ---[ Configure clog
688IF(NOT TARGET clog)
689 SET(CLOG_BUILD_TESTS OFF CACHE BOOL "")
690 SET(CLOG_RUNTIME_TYPE "${CPUINFO_RUNTIME_TYPE}" CACHE STRING "")
691 ADD_SUBDIRECTORY(
692 "${CLOG_SOURCE_DIR}/deps/clog"
693 "${CMAKE_BINARY_DIR}/clog")
694 # We build static version of clog but a dynamic library may indirectly depend on it
695 SET_PROPERTY(TARGET clog PROPERTY POSITION_INDEPENDENT_CODE ON)
696ENDIF()
697TARGET_LINK_LIBRARIES(XNNPACK PRIVATE clog)
698
699# ---[ Configure cpuinfo
700IF(NOT TARGET cpuinfo)
701 SET(CPUINFO_BUILD_TOOLS OFF CACHE BOOL "")
702 SET(CPUINFO_BUILD_UNIT_TESTS OFF CACHE BOOL "")
703 SET(CPUINFO_BUILD_MOCK_TESTS OFF CACHE BOOL "")
704 SET(CPUINFO_BUILD_BENCHMARKS OFF CACHE BOOL "")
705 ADD_SUBDIRECTORY(
706 "${CPUINFO_SOURCE_DIR}"
707 "${CMAKE_BINARY_DIR}/cpuinfo")
708ENDIF()
709TARGET_LINK_LIBRARIES(XNNPACK PRIVATE cpuinfo)
710
711# ---[ Configure pthreadpool
712IF(NOT TARGET pthreadpool)
713 SET(PTHREADPOOL_BUILD_TESTS OFF CACHE BOOL "")
714 SET(PTHREADPOOL_BUILD_BENCHMARKS OFF CACHE BOOL "")
Marat Dukhand70028a2019-10-29 09:53:16 -0700715 SET(PTHREADPOOL_ALLOW_DEPRECATED_API OFF CACHE BOOL "")
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700716 ADD_SUBDIRECTORY(
717 "${PTHREADPOOL_SOURCE_DIR}"
718 "${CMAKE_BINARY_DIR}/pthreadpool")
719ENDIF()
720TARGET_LINK_LIBRARIES(XNNPACK PUBLIC pthreadpool)
721
722# ---[ Configure FXdiv
723IF(NOT TARGET fxdiv)
724 SET(FXDIV_BUILD_TESTS OFF CACHE BOOL "")
725 SET(FXDIV_BUILD_BENCHMARKS OFF CACHE BOOL "")
726 ADD_SUBDIRECTORY(
727 "${FXDIV_SOURCE_DIR}"
728 "${CMAKE_BINARY_DIR}/FXdiv")
729ENDIF()
730TARGET_LINK_LIBRARIES(XNNPACK PRIVATE fxdiv)
731
732# ---[ Configure psimd
733IF(NOT TARGET psimd)
734 ADD_SUBDIRECTORY(
735 "${PSIMD_SOURCE_DIR}"
736 "${CMAKE_BINARY_DIR}/psimd")
737ENDIF()
738TARGET_LINK_LIBRARIES(XNNPACK PRIVATE psimd)
739
740# ---[ Configure FP16
741IF(NOT TARGET fp16)
742 SET(FP16_BUILD_TESTS OFF CACHE BOOL "")
743 SET(FP16_BUILD_BENCHMARKS OFF CACHE BOOL "")
744 ADD_SUBDIRECTORY(
745 "${FP16_SOURCE_DIR}"
746 "${CMAKE_BINARY_DIR}/FP16")
747ENDIF()
748TARGET_LINK_LIBRARIES(XNNPACK PRIVATE fp16)
749
750INSTALL(TARGETS XNNPACK
751 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
752 ARCHIVE DESTINATION ${CMAKE_INSTALL_LIBDIR}
753 PUBLIC_HEADER DESTINATION ${CMAKE_INSTALL_INCLUDEDIR})
754
755# ---[ XNNPACK unit tests
756IF(XNNPACK_BUILD_TESTS)
757 # ---[ Build google test
758 IF(NOT TARGET gtest)
759 SET(gtest_force_shared_crt ON CACHE BOOL "" FORCE)
760 ADD_SUBDIRECTORY(
761 "${GOOGLETEST_SOURCE_DIR}"
762 "${CMAKE_BINARY_DIR}/googletest")
763 ENDIF()
764
765 # ---[ Build operator-level unit tests
766 ADD_EXECUTABLE(add-test test/add.cc)
767 SET_TARGET_PROPERTIES(add-test PROPERTIES
768 CXX_STANDARD 11
769 CXX_STANDARD_REQUIRED YES
770 CXX_EXTENSIONS NO)
771 TARGET_INCLUDE_DIRECTORIES(add-test PRIVATE src test)
772 TARGET_LINK_LIBRARIES(add-test PRIVATE XNNPACK gtest gtest_main)
773 ADD_TEST(add-test add-test)
774
775 ADD_EXECUTABLE(argmax-pooling-test test/argmax-pooling.cc)
776 SET_TARGET_PROPERTIES(argmax-pooling-test PROPERTIES
777 CXX_STANDARD 11
778 CXX_STANDARD_REQUIRED YES
779 CXX_EXTENSIONS NO)
780 TARGET_INCLUDE_DIRECTORIES(argmax-pooling-test PRIVATE src test)
781 TARGET_LINK_LIBRARIES(argmax-pooling-test PRIVATE XNNPACK gtest gtest_main)
782 ADD_TEST(argmax-pooling-test argmax-pooling-test)
783
784 ADD_EXECUTABLE(average-pooling-test test/average-pooling.cc)
785 SET_TARGET_PROPERTIES(average-pooling-test PROPERTIES
786 CXX_STANDARD 11
787 CXX_STANDARD_REQUIRED YES
788 CXX_EXTENSIONS NO)
789 TARGET_INCLUDE_DIRECTORIES(average-pooling-test PRIVATE src test)
790 TARGET_LINK_LIBRARIES(average-pooling-test PRIVATE XNNPACK gtest gtest_main)
791 ADD_TEST(average-pooling-test average-pooling-test)
792
793 ADD_EXECUTABLE(channel-pad-test test/channel-pad.cc)
794 SET_TARGET_PROPERTIES(channel-pad-test PROPERTIES
795 CXX_STANDARD 11
796 CXX_STANDARD_REQUIRED YES
797 CXX_EXTENSIONS NO)
798 TARGET_INCLUDE_DIRECTORIES(channel-pad-test PRIVATE src test)
799 TARGET_LINK_LIBRARIES(channel-pad-test PRIVATE XNNPACK gtest gtest_main)
800 ADD_TEST(channel-pad-test channel-pad-test)
801
802 ADD_EXECUTABLE(channel-shuffle-test test/channel-shuffle.cc)
803 SET_TARGET_PROPERTIES(channel-shuffle-test PROPERTIES
804 CXX_STANDARD 11
805 CXX_STANDARD_REQUIRED YES
806 CXX_EXTENSIONS NO)
807 TARGET_INCLUDE_DIRECTORIES(channel-shuffle-test PRIVATE src test)
808 TARGET_LINK_LIBRARIES(channel-shuffle-test PRIVATE XNNPACK gtest gtest_main)
809 ADD_TEST(channel-shuffle-test channel-shuffle-test)
810
811 ADD_EXECUTABLE(clamp-test test/clamp.cc)
812 SET_TARGET_PROPERTIES(clamp-test PROPERTIES
813 CXX_STANDARD 11
814 CXX_STANDARD_REQUIRED YES
815 CXX_EXTENSIONS NO)
816 TARGET_INCLUDE_DIRECTORIES(clamp-test PRIVATE src test)
817 TARGET_LINK_LIBRARIES(clamp-test PRIVATE XNNPACK gtest gtest_main)
818 ADD_TEST(clamp-test clamp-test)
819
820 ADD_EXECUTABLE(convolution-test test/convolution.cc)
821 SET_TARGET_PROPERTIES(convolution-test PROPERTIES
822 CXX_STANDARD 11
823 CXX_STANDARD_REQUIRED YES
824 CXX_EXTENSIONS NO)
825 TARGET_INCLUDE_DIRECTORIES(convolution-test PRIVATE src test)
826 TARGET_LINK_LIBRARIES(convolution-test PRIVATE XNNPACK gtest gtest_main)
827 ADD_TEST(convolution-test convolution-test)
828
829 ADD_EXECUTABLE(convolution-spnchw-test test/convolution-spnchw.cc)
830 SET_TARGET_PROPERTIES(convolution-spnchw-test PROPERTIES
831 CXX_STANDARD 11
832 CXX_STANDARD_REQUIRED YES
833 CXX_EXTENSIONS NO)
834 TARGET_INCLUDE_DIRECTORIES(convolution-spnchw-test PRIVATE src test)
835 TARGET_LINK_LIBRARIES(convolution-spnchw-test PRIVATE XNNPACK gtest gtest_main)
836 ADD_TEST(convolution-spnchw-test convolution-spnchw-test)
837
838 ADD_EXECUTABLE(deconvolution-test test/deconvolution.cc)
839 SET_TARGET_PROPERTIES(deconvolution-test PROPERTIES
840 CXX_STANDARD 11
841 CXX_STANDARD_REQUIRED YES
842 CXX_EXTENSIONS NO)
843 TARGET_INCLUDE_DIRECTORIES(deconvolution-test PRIVATE src test)
844 TARGET_LINK_LIBRARIES(deconvolution-test PRIVATE XNNPACK gtest gtest_main)
845 ADD_TEST(deconvolution-test deconvolution-test)
846
847 ADD_EXECUTABLE(fully-connected-test test/fully-connected.cc)
848 SET_TARGET_PROPERTIES(fully-connected-test PROPERTIES
849 CXX_STANDARD 11
850 CXX_STANDARD_REQUIRED YES
851 CXX_EXTENSIONS NO)
852 TARGET_INCLUDE_DIRECTORIES(fully-connected-test PRIVATE src test)
853 TARGET_LINK_LIBRARIES(fully-connected-test PRIVATE XNNPACK gtest gtest_main)
854 ADD_TEST(fully-connected-test fully-connected-test)
855
856 ADD_EXECUTABLE(global-average-pooling-test test/global-average-pooling.cc)
857 SET_TARGET_PROPERTIES(global-average-pooling-test PROPERTIES
858 CXX_STANDARD 11
859 CXX_STANDARD_REQUIRED YES
860 CXX_EXTENSIONS NO)
861 TARGET_INCLUDE_DIRECTORIES(global-average-pooling-test PRIVATE src test)
862 TARGET_LINK_LIBRARIES(global-average-pooling-test PRIVATE XNNPACK gtest gtest_main)
863 ADD_TEST(global-average-pooling-test global-average-pooling-test)
864
865 ADD_EXECUTABLE(global-average-pooling-spnchw-test test/global-average-pooling-spnchw.cc)
866 SET_TARGET_PROPERTIES(global-average-pooling-spnchw-test PROPERTIES
867 CXX_STANDARD 11
868 CXX_STANDARD_REQUIRED YES
869 CXX_EXTENSIONS NO)
870 TARGET_INCLUDE_DIRECTORIES(global-average-pooling-spnchw-test PRIVATE src test)
871 TARGET_LINK_LIBRARIES(global-average-pooling-spnchw-test PRIVATE XNNPACK gtest gtest_main)
872 ADD_TEST(global-average-pooling-spnchw-test global-average-pooling-spnchw-test)
873
874 ADD_EXECUTABLE(hardswish-test test/hardswish.cc)
875 SET_TARGET_PROPERTIES(hardswish-test PROPERTIES
876 CXX_STANDARD 11
877 CXX_STANDARD_REQUIRED YES
878 CXX_EXTENSIONS NO)
879 TARGET_INCLUDE_DIRECTORIES(hardswish-test PRIVATE src test)
880 TARGET_LINK_LIBRARIES(hardswish-test PRIVATE XNNPACK gtest gtest_main)
881 ADD_TEST(hardswish-test hardswish-test)
882
883 ADD_EXECUTABLE(leaky-relu-test test/leaky-relu.cc)
884 SET_TARGET_PROPERTIES(leaky-relu-test PROPERTIES
885 CXX_STANDARD 11
886 CXX_STANDARD_REQUIRED YES
887 CXX_EXTENSIONS NO)
888 TARGET_INCLUDE_DIRECTORIES(leaky-relu-test PRIVATE src test)
889 TARGET_LINK_LIBRARIES(leaky-relu-test PRIVATE XNNPACK gtest gtest_main)
890 ADD_TEST(leaky-relu-test leaky-relu-test)
891
892 ADD_EXECUTABLE(max-pooling-test test/max-pooling.cc)
893 SET_TARGET_PROPERTIES(max-pooling-test PROPERTIES
894 CXX_STANDARD 11
895 CXX_STANDARD_REQUIRED YES
896 CXX_EXTENSIONS NO)
897 TARGET_INCLUDE_DIRECTORIES(max-pooling-test PRIVATE src test)
898 TARGET_LINK_LIBRARIES(max-pooling-test PRIVATE XNNPACK gtest gtest_main)
899 ADD_TEST(max-pooling-test max-pooling-test)
900
901 ADD_EXECUTABLE(prelu-test test/prelu.cc)
902 SET_TARGET_PROPERTIES(prelu-test PROPERTIES
903 CXX_STANDARD 11
904 CXX_STANDARD_REQUIRED YES
905 CXX_EXTENSIONS NO)
906 TARGET_INCLUDE_DIRECTORIES(prelu-test PRIVATE src test)
907 TARGET_LINK_LIBRARIES(prelu-test PRIVATE XNNPACK gtest gtest_main)
908 ADD_TEST(prelu-test prelu-test)
909
910 ADD_EXECUTABLE(sigmoid-test test/sigmoid.cc)
911 SET_TARGET_PROPERTIES(sigmoid-test PROPERTIES
912 CXX_STANDARD 11
913 CXX_STANDARD_REQUIRED YES
914 CXX_EXTENSIONS NO)
915 TARGET_INCLUDE_DIRECTORIES(sigmoid-test PRIVATE src test)
916 TARGET_LINK_LIBRARIES(sigmoid-test PRIVATE XNNPACK gtest gtest_main)
917 ADD_TEST(sigmoid-test sigmoid-test)
918
919 ADD_EXECUTABLE(softargmax-test test/softargmax.cc)
920 SET_TARGET_PROPERTIES(softargmax-test PROPERTIES
921 CXX_STANDARD 11
922 CXX_STANDARD_REQUIRED YES
923 CXX_EXTENSIONS NO)
924 TARGET_INCLUDE_DIRECTORIES(softargmax-test PRIVATE src test)
925 TARGET_LINK_LIBRARIES(softargmax-test PRIVATE XNNPACK gtest gtest_main)
926 ADD_TEST(softargmax-test softargmax-test)
927
928 ADD_EXECUTABLE(unpooling-test test/unpooling.cc)
929 SET_TARGET_PROPERTIES(unpooling-test PROPERTIES
930 CXX_STANDARD 11
931 CXX_STANDARD_REQUIRED YES
932 CXX_EXTENSIONS NO)
933 TARGET_INCLUDE_DIRECTORIES(unpooling-test PRIVATE src test)
934 TARGET_LINK_LIBRARIES(unpooling-test PRIVATE XNNPACK gtest gtest_main)
935 ADD_TEST(unpooling-test unpooling-test)
936
937 # ---[ Build microkernel-level unit tests
938 ADD_EXECUTABLE(f16-gemm-test test/f16-gemm.cc)
939 SET_TARGET_PROPERTIES(f16-gemm-test PROPERTIES
940 CXX_STANDARD 11
941 CXX_STANDARD_REQUIRED YES
942 CXX_EXTENSIONS YES)
943 TARGET_INCLUDE_DIRECTORIES(f16-gemm-test PRIVATE src test)
944 TARGET_LINK_LIBRARIES(f16-gemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
945 ADD_TEST(f16-gemm-test f16-gemm-test)
946
947 ADD_EXECUTABLE(f32-argmaxpool-test test/f32-argmaxpool.cc)
948 SET_TARGET_PROPERTIES(f32-argmaxpool-test PROPERTIES
949 CXX_STANDARD 11
950 CXX_STANDARD_REQUIRED YES
951 CXX_EXTENSIONS YES)
952 TARGET_INCLUDE_DIRECTORIES(f32-argmaxpool-test PRIVATE src test)
953 TARGET_LINK_LIBRARIES(f32-argmaxpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
954 ADD_TEST(f32-argmaxpool-test f32-argmaxpool-test)
955
956 ADD_EXECUTABLE(f32-avgpool-test test/f32-avgpool.cc)
957 SET_TARGET_PROPERTIES(f32-avgpool-test PROPERTIES
958 CXX_STANDARD 11
959 CXX_STANDARD_REQUIRED YES
960 CXX_EXTENSIONS YES)
961 TARGET_INCLUDE_DIRECTORIES(f32-avgpool-test PRIVATE src test)
962 TARGET_LINK_LIBRARIES(f32-avgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
963 ADD_TEST(f32-avgpool-test f32-avgpool-test)
964
965 ADD_EXECUTABLE(f32-clamp-test test/f32-clamp.cc)
966 SET_TARGET_PROPERTIES(f32-clamp-test PROPERTIES
967 CXX_STANDARD 11
968 CXX_STANDARD_REQUIRED YES
969 CXX_EXTENSIONS YES)
970 TARGET_INCLUDE_DIRECTORIES(f32-clamp-test PRIVATE src test)
971 TARGET_LINK_LIBRARIES(f32-clamp-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
972 ADD_TEST(f32-clamp-test f32-clamp-test)
973
974 ADD_EXECUTABLE(f32-conv-hwc-test test/f32-conv-hwc.cc)
975 SET_TARGET_PROPERTIES(f32-conv-hwc-test PROPERTIES
976 CXX_STANDARD 11
977 CXX_STANDARD_REQUIRED YES
978 CXX_EXTENSIONS YES)
979 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc-test PRIVATE src test)
980 TARGET_LINK_LIBRARIES(f32-conv-hwc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
981 ADD_TEST(f32-conv-hwc-test f32-conv-hwc-test)
982
983 ADD_EXECUTABLE(f32-conv-hwc2spchw-test test/f32-conv-hwc2spchw.cc)
984 SET_TARGET_PROPERTIES(f32-conv-hwc2spchw-test PROPERTIES
985 CXX_STANDARD 11
986 CXX_STANDARD_REQUIRED YES
987 CXX_EXTENSIONS YES)
988 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc2spchw-test PRIVATE src test)
989 TARGET_LINK_LIBRARIES(f32-conv-hwc2spchw-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
990 ADD_TEST(f32-conv-hwc2spchw-test f32-conv-hwc2spchw-test)
991
992 ADD_EXECUTABLE(f32-dwconv-spchw-test test/f32-dwconv-spchw.cc)
993 SET_TARGET_PROPERTIES(f32-dwconv-spchw-test PROPERTIES
994 CXX_STANDARD 11
995 CXX_STANDARD_REQUIRED YES
996 CXX_EXTENSIONS YES)
997 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-spchw-test PRIVATE src test)
998 TARGET_LINK_LIBRARIES(f32-dwconv-spchw-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
999 ADD_TEST(f32-dwconv-spchw-test f32-dwconv-spchw-test)
1000
1001 ADD_EXECUTABLE(f32-dwconv-test test/f32-dwconv.cc)
1002 SET_TARGET_PROPERTIES(f32-dwconv-test PROPERTIES
1003 CXX_STANDARD 11
1004 CXX_STANDARD_REQUIRED YES
1005 CXX_EXTENSIONS YES)
1006 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-test PRIVATE src test)
1007 TARGET_LINK_LIBRARIES(f32-dwconv-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1008 ADD_TEST(f32-dwconv-test f32-dwconv-test)
1009
1010 ADD_EXECUTABLE(f32-gavgpool-spchw-test test/f32-gavgpool-spchw.cc)
1011 SET_TARGET_PROPERTIES(f32-gavgpool-spchw-test PROPERTIES
1012 CXX_STANDARD 11
1013 CXX_STANDARD_REQUIRED YES
1014 CXX_EXTENSIONS YES)
1015 TARGET_INCLUDE_DIRECTORIES(f32-gavgpool-spchw-test PRIVATE src test)
1016 TARGET_LINK_LIBRARIES(f32-gavgpool-spchw-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1017 ADD_TEST(f32-gavgpool-spchw-test f32-gavgpool-spchw-test)
1018
1019 ADD_EXECUTABLE(f32-gavgpool-test test/f32-gavgpool.cc)
1020 SET_TARGET_PROPERTIES(f32-gavgpool-test PROPERTIES
1021 CXX_STANDARD 11
1022 CXX_STANDARD_REQUIRED YES
1023 CXX_EXTENSIONS YES)
1024 TARGET_INCLUDE_DIRECTORIES(f32-gavgpool-test PRIVATE src test)
1025 TARGET_LINK_LIBRARIES(f32-gavgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1026 ADD_TEST(f32-gavgpool-test f32-gavgpool-test)
1027
1028 ADD_EXECUTABLE(f32-gemm-test test/f32-gemm.cc)
1029 SET_TARGET_PROPERTIES(f32-gemm-test PROPERTIES
1030 CXX_STANDARD 11
1031 CXX_STANDARD_REQUIRED YES
1032 CXX_EXTENSIONS YES)
1033 TARGET_INCLUDE_DIRECTORIES(f32-gemm-test PRIVATE src test)
1034 TARGET_LINK_LIBRARIES(f32-gemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1035 ADD_TEST(f32-gemm-test f32-gemm-test)
1036
1037 ADD_EXECUTABLE(f32-gemminc-test test/f32-gemminc.cc)
1038 SET_TARGET_PROPERTIES(f32-gemminc-test PROPERTIES
1039 CXX_STANDARD 11
1040 CXX_STANDARD_REQUIRED YES
1041 CXX_EXTENSIONS YES)
1042 TARGET_INCLUDE_DIRECTORIES(f32-gemminc-test PRIVATE src test)
1043 TARGET_LINK_LIBRARIES(f32-gemminc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1044 ADD_TEST(f32-gemminc-test f32-gemminc-test)
1045
1046 ADD_EXECUTABLE(f32-hswish-test test/f32-hswish.cc)
1047 SET_TARGET_PROPERTIES(f32-hswish-test PROPERTIES
1048 CXX_STANDARD 11
1049 CXX_STANDARD_REQUIRED YES
1050 CXX_EXTENSIONS YES)
1051 TARGET_INCLUDE_DIRECTORIES(f32-hswish-test PRIVATE src test)
1052 TARGET_LINK_LIBRARIES(f32-hswish-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1053 ADD_TEST(f32-hswish-test f32-hswish-test)
1054
1055 ADD_EXECUTABLE(f32-igemm-test test/f32-igemm.cc)
1056 SET_TARGET_PROPERTIES(f32-igemm-test PROPERTIES
1057 CXX_STANDARD 11
1058 CXX_STANDARD_REQUIRED YES
1059 CXX_EXTENSIONS YES)
1060 TARGET_INCLUDE_DIRECTORIES(f32-igemm-test PRIVATE src test)
1061 TARGET_LINK_LIBRARIES(f32-igemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1062 ADD_TEST(f32-igemm-test f32-igemm-test)
1063
1064 ADD_EXECUTABLE(f32-maxpool-test test/f32-maxpool.cc)
1065 SET_TARGET_PROPERTIES(f32-maxpool-test PROPERTIES
1066 CXX_STANDARD 11
1067 CXX_STANDARD_REQUIRED YES
1068 CXX_EXTENSIONS YES)
1069 TARGET_INCLUDE_DIRECTORIES(f32-maxpool-test PRIVATE src test)
1070 TARGET_LINK_LIBRARIES(f32-maxpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1071 ADD_TEST(f32-maxpool-test f32-maxpool-test)
1072
1073 ADD_EXECUTABLE(f32-pavgpool-test test/f32-pavgpool.cc)
1074 SET_TARGET_PROPERTIES(f32-pavgpool-test PROPERTIES
1075 CXX_STANDARD 11
1076 CXX_STANDARD_REQUIRED YES
1077 CXX_EXTENSIONS YES)
1078 TARGET_INCLUDE_DIRECTORIES(f32-pavgpool-test PRIVATE src test)
1079 TARGET_LINK_LIBRARIES(f32-pavgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1080 ADD_TEST(f32-pavgpool-test f32-pavgpool-test)
1081
1082 ADD_EXECUTABLE(f32-ppmm-test test/f32-ppmm.cc)
1083 SET_TARGET_PROPERTIES(f32-ppmm-test PROPERTIES
1084 CXX_STANDARD 11
1085 CXX_STANDARD_REQUIRED YES
1086 CXX_EXTENSIONS YES)
1087 TARGET_INCLUDE_DIRECTORIES(f32-ppmm-test PRIVATE src test)
1088 TARGET_LINK_LIBRARIES(f32-ppmm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1089 ADD_TEST(f32-ppmm-test f32-ppmm-test)
1090
1091 ADD_EXECUTABLE(f32-prelu-test test/f32-prelu.cc)
1092 SET_TARGET_PROPERTIES(f32-prelu-test PROPERTIES
1093 CXX_STANDARD 11
1094 CXX_STANDARD_REQUIRED YES
1095 CXX_EXTENSIONS YES)
1096 TARGET_INCLUDE_DIRECTORIES(f32-prelu-test PRIVATE src test)
1097 TARGET_LINK_LIBRARIES(f32-prelu-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1098 ADD_TEST(f32-prelu-test f32-prelu-test)
1099
Marat Dukhan97579532019-10-18 16:40:39 -07001100 ADD_EXECUTABLE(f32-raddexpminusmax-test test/f32-raddexpminusmax.cc)
1101 SET_TARGET_PROPERTIES(f32-raddexpminusmax-test PROPERTIES
1102 CXX_STANDARD 11
1103 CXX_STANDARD_REQUIRED YES
1104 CXX_EXTENSIONS YES)
1105 TARGET_INCLUDE_DIRECTORIES(f32-raddexpminusmax-test PRIVATE src test)
1106 TARGET_LINK_LIBRARIES(f32-raddexpminusmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1107 ADD_TEST(f32-raddexpminusmax-test f32-raddexpminusmax-test)
1108
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001109 ADD_EXECUTABLE(f32-raddextexp-test test/f32-raddextexp.cc)
1110 SET_TARGET_PROPERTIES(f32-raddextexp-test PROPERTIES
1111 CXX_STANDARD 11
1112 CXX_STANDARD_REQUIRED YES
1113 CXX_EXTENSIONS YES)
1114 TARGET_INCLUDE_DIRECTORIES(f32-raddextexp-test PRIVATE src test)
Marat Dukhana5977df2019-10-30 22:02:15 -07001115 TARGET_LINK_LIBRARIES(f32-raddextexp-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001116 ADD_TEST(f32-raddextexp-test f32-raddextexp-test)
1117
Marat Dukhan97579532019-10-18 16:40:39 -07001118 ADD_EXECUTABLE(f32-raddstoreexpminusmax-test test/f32-raddstoreexpminusmax.cc)
1119 SET_TARGET_PROPERTIES(f32-raddstoreexpminusmax-test PROPERTIES
1120 CXX_STANDARD 11
1121 CXX_STANDARD_REQUIRED YES
1122 CXX_EXTENSIONS YES)
1123 TARGET_INCLUDE_DIRECTORIES(f32-raddstoreexpminusmax-test PRIVATE src test)
1124 TARGET_LINK_LIBRARIES(f32-raddstoreexpminusmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1125 ADD_TEST(f32-raddstoreexpminusmax-test f32-raddstoreexpminusmax-test)
1126
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001127 ADD_EXECUTABLE(f32-rmax-test test/f32-rmax.cc)
1128 SET_TARGET_PROPERTIES(f32-rmax-test PROPERTIES
1129 CXX_STANDARD 11
1130 CXX_STANDARD_REQUIRED YES
1131 CXX_EXTENSIONS YES)
1132 TARGET_INCLUDE_DIRECTORIES(f32-rmax-test PRIVATE src test)
1133 TARGET_LINK_LIBRARIES(f32-rmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1134 ADD_TEST(f32-rmax-test f32-rmax-test)
1135
1136 ADD_EXECUTABLE(f32-spmm-test test/f32-spmm.cc)
1137 SET_TARGET_PROPERTIES(f32-spmm-test PROPERTIES
1138 CXX_STANDARD 11
1139 CXX_STANDARD_REQUIRED YES
1140 CXX_EXTENSIONS YES)
1141 TARGET_INCLUDE_DIRECTORIES(f32-spmm-test PRIVATE src test)
1142 TARGET_LINK_LIBRARIES(f32-spmm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1143 ADD_TEST(f32-spmm-test f32-spmm-test)
1144
1145 ADD_EXECUTABLE(f32-vadd-test test/f32-vadd.cc)
1146 SET_TARGET_PROPERTIES(f32-vadd-test PROPERTIES
1147 CXX_STANDARD 11
1148 CXX_STANDARD_REQUIRED YES
1149 CXX_EXTENSIONS YES)
1150 TARGET_INCLUDE_DIRECTORIES(f32-vadd-test PRIVATE src test)
1151 TARGET_LINK_LIBRARIES(f32-vadd-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1152 ADD_TEST(f32-vadd-test f32-vadd-test)
1153
1154 ADD_EXECUTABLE(f32-vmul-test test/f32-vmul.cc)
1155 SET_TARGET_PROPERTIES(f32-vmul-test PROPERTIES
1156 CXX_STANDARD 11
1157 CXX_STANDARD_REQUIRED YES
1158 CXX_EXTENSIONS YES)
1159 TARGET_INCLUDE_DIRECTORIES(f32-vmul-test PRIVATE src test)
1160 TARGET_LINK_LIBRARIES(f32-vmul-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1161 ADD_TEST(f32-vmul-test f32-vmul-test)
1162
1163 ADD_EXECUTABLE(f32-vmulcaddc-test test/f32-vmulcaddc.cc)
1164 SET_TARGET_PROPERTIES(f32-vmulcaddc-test PROPERTIES
1165 CXX_STANDARD 11
1166 CXX_STANDARD_REQUIRED YES
1167 CXX_EXTENSIONS YES)
1168 TARGET_INCLUDE_DIRECTORIES(f32-vmulcaddc-test PRIVATE src test)
1169 TARGET_LINK_LIBRARIES(f32-vmulcaddc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1170 ADD_TEST(f32-vmulcaddc-test f32-vmulcaddc-test)
1171
Marat Dukhan97579532019-10-18 16:40:39 -07001172 ADD_EXECUTABLE(f32-vscaleexpminusmax-test test/f32-vscaleexpminusmax.cc)
1173 SET_TARGET_PROPERTIES(f32-vscaleexpminusmax-test PROPERTIES
1174 CXX_STANDARD 11
1175 CXX_STANDARD_REQUIRED YES
1176 CXX_EXTENSIONS YES)
1177 TARGET_INCLUDE_DIRECTORIES(f32-vscaleexpminusmax-test PRIVATE src test)
1178 TARGET_LINK_LIBRARIES(f32-vscaleexpminusmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1179 ADD_TEST(f32-vscaleexpminusmax-test f32-vscaleexpminusmax-test)
1180
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001181 ADD_EXECUTABLE(f32-vscaleextexp-test test/f32-vscaleextexp.cc)
1182 SET_TARGET_PROPERTIES(f32-vscaleexpminusmax-test PROPERTIES
1183 CXX_STANDARD 11
1184 CXX_STANDARD_REQUIRED YES
1185 CXX_EXTENSIONS YES)
1186 TARGET_INCLUDE_DIRECTORIES(f32-vscaleextexp-test PRIVATE src test)
Marat Dukhana5977df2019-10-30 22:02:15 -07001187 TARGET_LINK_LIBRARIES(f32-vscaleextexp-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001188 ADD_TEST(f32-vscaleextexp-test f32-vscaleextexp-test)
1189
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001190 ADD_EXECUTABLE(f32-vsub-test test/f32-vsub.cc)
1191 SET_TARGET_PROPERTIES(f32-vsub-test PROPERTIES
1192 CXX_STANDARD 11
1193 CXX_STANDARD_REQUIRED YES
1194 CXX_EXTENSIONS YES)
1195 TARGET_INCLUDE_DIRECTORIES(f32-vsub-test PRIVATE src test)
1196 TARGET_LINK_LIBRARIES(f32-vsub-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1197 ADD_TEST(f32-vsub-test f32-vsub-test)
1198
1199 ADD_EXECUTABLE(q8-avgpool-test test/q8-avgpool.cc)
1200 SET_TARGET_PROPERTIES(q8-avgpool-test PROPERTIES
1201 CXX_STANDARD 11
1202 CXX_STANDARD_REQUIRED YES
1203 CXX_EXTENSIONS YES)
1204 TARGET_INCLUDE_DIRECTORIES(q8-avgpool-test PRIVATE src test)
1205 TARGET_LINK_LIBRARIES(q8-avgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1206 ADD_TEST(q8-avgpool-test q8-avgpool-test)
1207
1208 ADD_EXECUTABLE(q8-dwconv-test test/q8-dwconv.cc)
1209 SET_TARGET_PROPERTIES(q8-dwconv-test PROPERTIES
1210 CXX_STANDARD 11
1211 CXX_STANDARD_REQUIRED YES
1212 CXX_EXTENSIONS YES)
1213 TARGET_INCLUDE_DIRECTORIES(q8-dwconv-test PRIVATE src test)
1214 TARGET_LINK_LIBRARIES(q8-dwconv-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1215 ADD_TEST(q8-dwconv-test q8-dwconv-test)
1216
1217 ADD_EXECUTABLE(q8-gavgpool-test test/q8-gavgpool.cc)
1218 SET_TARGET_PROPERTIES(q8-gavgpool-test PROPERTIES
1219 CXX_STANDARD 11
1220 CXX_STANDARD_REQUIRED YES
1221 CXX_EXTENSIONS YES)
1222 TARGET_INCLUDE_DIRECTORIES(q8-gavgpool-test PRIVATE src test)
1223 TARGET_LINK_LIBRARIES(q8-gavgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1224 ADD_TEST(q8-gavgpool-test q8-gavgpool-test)
1225
1226 ADD_EXECUTABLE(q8-gemm-test test/q8-gemm.cc)
1227 SET_TARGET_PROPERTIES(q8-gemm-test PROPERTIES
1228 CXX_STANDARD 11
1229 CXX_STANDARD_REQUIRED YES
1230 CXX_EXTENSIONS YES)
1231 TARGET_INCLUDE_DIRECTORIES(q8-gemm-test PRIVATE src test)
1232 TARGET_LINK_LIBRARIES(q8-gemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1233 ADD_TEST(q8-gemm-test q8-gemm-test)
1234
1235 ADD_EXECUTABLE(q8-igemm-test test/q8-igemm.cc)
1236 SET_TARGET_PROPERTIES(q8-igemm-test PROPERTIES
1237 CXX_STANDARD 11
1238 CXX_STANDARD_REQUIRED YES
1239 CXX_EXTENSIONS YES)
1240 TARGET_INCLUDE_DIRECTORIES(q8-igemm-test PRIVATE src test)
1241 TARGET_LINK_LIBRARIES(q8-igemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1242 ADD_TEST(q8-igemm-test q8-igemm-test)
1243
1244 ADD_EXECUTABLE(q8-vadd-test test/q8-vadd.cc)
1245 SET_TARGET_PROPERTIES(q8-vadd-test PROPERTIES
1246 CXX_STANDARD 11
1247 CXX_STANDARD_REQUIRED YES
1248 CXX_EXTENSIONS YES)
1249 TARGET_INCLUDE_DIRECTORIES(q8-vadd-test PRIVATE src test)
1250 TARGET_LINK_LIBRARIES(q8-vadd-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1251 ADD_TEST(q8-vadd-test q8-vadd-test)
1252
1253 ADD_EXECUTABLE(u8-clamp-test test/u8-clamp.cc)
1254 SET_TARGET_PROPERTIES(u8-clamp-test PROPERTIES
1255 CXX_STANDARD 11
1256 CXX_STANDARD_REQUIRED YES
1257 CXX_EXTENSIONS YES)
1258 TARGET_INCLUDE_DIRECTORIES(u8-clamp-test PRIVATE src test)
1259 TARGET_LINK_LIBRARIES(u8-clamp-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1260 ADD_TEST(u8-clamp-test u8-clamp-test)
1261
1262 ADD_EXECUTABLE(u8-lut32norm-test test/u8-lut32norm.cc)
1263 SET_TARGET_PROPERTIES(u8-lut32norm-test PROPERTIES
1264 CXX_STANDARD 11
1265 CXX_STANDARD_REQUIRED YES
1266 CXX_EXTENSIONS YES)
1267 TARGET_INCLUDE_DIRECTORIES(u8-lut32norm-test PRIVATE src test)
1268 TARGET_LINK_LIBRARIES(u8-lut32norm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1269 ADD_TEST(u8-lut32norm-test u8-lut32norm-test)
1270
1271 ADD_EXECUTABLE(u8-maxpool-test test/u8-maxpool.cc)
1272 SET_TARGET_PROPERTIES(u8-maxpool-test PROPERTIES
1273 CXX_STANDARD 11
1274 CXX_STANDARD_REQUIRED YES
1275 CXX_EXTENSIONS YES)
1276 TARGET_INCLUDE_DIRECTORIES(u8-maxpool-test PRIVATE src test)
1277 TARGET_LINK_LIBRARIES(u8-maxpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1278 ADD_TEST(u8-maxpool-test u8-maxpool-test)
1279
1280 ADD_EXECUTABLE(u8-rmax-test test/u8-rmax.cc)
1281 SET_TARGET_PROPERTIES(u8-rmax-test PROPERTIES
1282 CXX_STANDARD 11
1283 CXX_STANDARD_REQUIRED YES
1284 CXX_EXTENSIONS YES)
1285 TARGET_INCLUDE_DIRECTORIES(u8-rmax-test PRIVATE src test)
1286 TARGET_LINK_LIBRARIES(u8-rmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1287 ADD_TEST(u8-rmax-test u8-rmax-test)
1288
1289 ADD_EXECUTABLE(x32-packx-test test/x32-packx.cc)
1290 SET_TARGET_PROPERTIES(x32-packx-test PROPERTIES
1291 CXX_STANDARD 11
1292 CXX_STANDARD_REQUIRED YES
1293 CXX_EXTENSIONS YES)
1294 TARGET_INCLUDE_DIRECTORIES(x32-packx-test PRIVATE src test)
1295 TARGET_LINK_LIBRARIES(x32-packx-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1296 ADD_TEST(x32-packx-test x32-packx-test)
1297
1298 ADD_EXECUTABLE(x32-pad-test test/x32-pad.cc)
1299 SET_TARGET_PROPERTIES(x32-pad-test PROPERTIES
1300 CXX_STANDARD 11
1301 CXX_STANDARD_REQUIRED YES
1302 CXX_EXTENSIONS YES)
1303 TARGET_INCLUDE_DIRECTORIES(x32-pad-test PRIVATE src test)
1304 TARGET_LINK_LIBRARIES(x32-pad-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1305 ADD_TEST(x32-pad-test x32-pad-test)
1306
1307 ADD_EXECUTABLE(x32-unpool-test test/x32-unpool.cc)
1308 SET_TARGET_PROPERTIES(x32-unpool-test PROPERTIES
1309 CXX_STANDARD 11
1310 CXX_STANDARD_REQUIRED YES
1311 CXX_EXTENSIONS YES)
1312 TARGET_INCLUDE_DIRECTORIES(x32-unpool-test PRIVATE src test)
1313 TARGET_LINK_LIBRARIES(x32-unpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1314 ADD_TEST(x32-unpool-test x32-unpool-test)
1315
1316 ADD_EXECUTABLE(x32-zip-test test/x32-zip.cc)
1317 SET_TARGET_PROPERTIES(x32-zip-test PROPERTIES
1318 CXX_STANDARD 11
1319 CXX_STANDARD_REQUIRED YES
1320 CXX_EXTENSIONS YES)
1321 TARGET_INCLUDE_DIRECTORIES(x32-zip-test PRIVATE src test)
1322 TARGET_LINK_LIBRARIES(x32-zip-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1323 ADD_TEST(x32-zip-test x32-zip-test)
1324
1325 ADD_EXECUTABLE(x8-lut-test test/x8-lut.cc)
1326 SET_TARGET_PROPERTIES(x8-lut-test PROPERTIES
1327 CXX_STANDARD 11
1328 CXX_STANDARD_REQUIRED YES
1329 CXX_EXTENSIONS YES)
1330 TARGET_INCLUDE_DIRECTORIES(x8-lut-test PRIVATE src test)
1331 TARGET_LINK_LIBRARIES(x8-lut-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1332 ADD_TEST(x8-lut-test x8-lut-test)
1333
1334 ADD_EXECUTABLE(x8-zip-test test/x8-zip.cc)
1335 SET_TARGET_PROPERTIES(x8-zip-test PROPERTIES
1336 CXX_STANDARD 11
1337 CXX_STANDARD_REQUIRED YES
1338 CXX_EXTENSIONS YES)
1339 TARGET_INCLUDE_DIRECTORIES(x8-zip-test PRIVATE src test)
1340 TARGET_LINK_LIBRARIES(x8-zip-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1341 ADD_TEST(x8-zip-test x8-zip-test)
1342ENDIF()
1343
1344# ---[ XNNPACK microbenchmarks
1345IF(XNNPACK_BUILD_BENCHMARKS)
1346 # ---[ Build google benchmark
1347 IF(NOT TARGET benchmark)
1348 SET(BENCHMARK_ENABLE_TESTING OFF CACHE BOOL "")
1349 ADD_SUBDIRECTORY(
1350 "${GOOGLEBENCHMARK_SOURCE_DIR}"
1351 "${CONFU_DEPENDENCIES_BINARY_DIR}/googlebenchmark")
1352 ENDIF()
1353
1354 ADD_LIBRARY(bench-utils bench/utils.cc)
1355 SET_TARGET_PROPERTIES(bench-utils PROPERTIES
1356 CXX_STANDARD 11
1357 CXX_STANDARD_REQUIRED YES
1358 CXX_EXTENSIONS NO)
1359 TARGET_INCLUDE_DIRECTORIES(bench-utils PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
Marat Dukhanbad48fe2019-11-04 10:35:22 -08001360 TARGET_LINK_LIBRARIES(bench-utils PRIVATE benchmark cpuinfo)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001361
Marat Dukhan5f18d262019-10-31 10:24:14 -07001362 # ---[ Build end-to-end microbenchmarks
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001363 ADD_EXECUTABLE(end2end-bench bench/end2end.cc models/mobilenet-v1.cc models/mobilenet-v2.cc)
1364 SET_TARGET_PROPERTIES(end2end-bench PROPERTIES
1365 CXX_STANDARD 11
1366 CXX_STANDARD_REQUIRED YES
1367 CXX_EXTENSIONS NO)
1368 TARGET_INCLUDE_DIRECTORIES(end2end-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
Frank Barcharde72e2872019-10-31 11:12:15 -07001369 TARGET_LINK_LIBRARIES(end2end-bench PRIVATE XNNPACK benchmark bench-utils)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001370
Marat Dukhanef4416e2019-10-31 13:44:40 -07001371 ADD_EXECUTABLE(f32-dwconv-e2e-bench bench/f32-dwconv-e2e.cc models/mobilenet-v1.cc models/mobilenet-v2.cc)
1372 SET_TARGET_PROPERTIES(f32-dwconv-e2e-bench PROPERTIES
1373 CXX_STANDARD 11
1374 CXX_STANDARD_REQUIRED YES
1375 CXX_EXTENSIONS NO)
Marat Dukhanc9d2f3f2019-10-31 14:02:07 -07001376 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-e2e-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}" src)
Marat Dukhanef4416e2019-10-31 13:44:40 -07001377 TARGET_LINK_LIBRARIES(f32-dwconv-e2e-bench PRIVATE XNNPACK benchmark bench-utils)
1378
Marat Dukhan0a5a53f2019-10-31 10:58:31 -07001379 ADD_EXECUTABLE(f32-gemm-e2e-bench bench/f32-gemm-e2e.cc models/mobilenet-v1.cc models/mobilenet-v2.cc)
1380 SET_TARGET_PROPERTIES(f32-gemm-e2e-bench PROPERTIES
Marat Dukhan5f18d262019-10-31 10:24:14 -07001381 CXX_STANDARD 11
1382 CXX_STANDARD_REQUIRED YES
1383 CXX_EXTENSIONS NO)
Marat Dukhanc9d2f3f2019-10-31 14:02:07 -07001384 TARGET_INCLUDE_DIRECTORIES(f32-gemm-e2e-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}" src)
Frank Barchardc712fa42019-10-31 14:00:21 -07001385 TARGET_LINK_LIBRARIES(f32-gemm-e2e-bench PRIVATE XNNPACK benchmark bench-utils)
Marat Dukhan5f18d262019-10-31 10:24:14 -07001386
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001387 # ---[ Build operator-level microbenchmarks
1388 ADD_EXECUTABLE(add-bench bench/add.cc)
1389 SET_TARGET_PROPERTIES(add-bench PROPERTIES
1390 CXX_STANDARD 11
1391 CXX_STANDARD_REQUIRED YES
1392 CXX_EXTENSIONS NO)
1393 TARGET_INCLUDE_DIRECTORIES(add-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1394 TARGET_LINK_LIBRARIES(add-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1395
1396 ADD_EXECUTABLE(average-pooling-bench bench/average-pooling.cc)
1397 SET_TARGET_PROPERTIES(average-pooling-bench PROPERTIES
1398 CXX_STANDARD 11
1399 CXX_STANDARD_REQUIRED YES
1400 CXX_EXTENSIONS NO)
1401 TARGET_INCLUDE_DIRECTORIES(average-pooling-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1402 TARGET_LINK_LIBRARIES(average-pooling-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1403
1404 ADD_EXECUTABLE(channel-shuffle-bench bench/channel-shuffle.cc)
1405 SET_TARGET_PROPERTIES(channel-shuffle-bench PROPERTIES
1406 CXX_STANDARD 11
1407 CXX_STANDARD_REQUIRED YES
1408 CXX_EXTENSIONS NO)
1409 TARGET_INCLUDE_DIRECTORIES(channel-shuffle-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1410 TARGET_LINK_LIBRARIES(channel-shuffle-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1411
1412 ADD_EXECUTABLE(convolution-bench bench/convolution.cc)
1413 SET_TARGET_PROPERTIES(convolution-bench PROPERTIES
1414 CXX_STANDARD 11
1415 CXX_STANDARD_REQUIRED YES
1416 CXX_EXTENSIONS NO)
1417 TARGET_INCLUDE_DIRECTORIES(convolution-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1418 TARGET_LINK_LIBRARIES(convolution-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1419
1420 ADD_EXECUTABLE(deconvolution-bench bench/deconvolution.cc)
1421 SET_TARGET_PROPERTIES(deconvolution-bench PROPERTIES
1422 CXX_STANDARD 11
1423 CXX_STANDARD_REQUIRED YES
1424 CXX_EXTENSIONS NO)
1425 TARGET_INCLUDE_DIRECTORIES(deconvolution-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1426 TARGET_LINK_LIBRARIES(deconvolution-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1427
1428 ADD_EXECUTABLE(global-average-pooling-bench bench/global-average-pooling.cc)
1429 SET_TARGET_PROPERTIES(global-average-pooling-bench PROPERTIES
1430 CXX_STANDARD 11
1431 CXX_STANDARD_REQUIRED YES
1432 CXX_EXTENSIONS NO)
1433 TARGET_INCLUDE_DIRECTORIES(global-average-pooling-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1434 TARGET_LINK_LIBRARIES(global-average-pooling-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1435
1436 ADD_EXECUTABLE(max-pooling-bench bench/max-pooling.cc)
1437 SET_TARGET_PROPERTIES(max-pooling-bench PROPERTIES
1438 CXX_STANDARD 11
1439 CXX_STANDARD_REQUIRED YES
1440 CXX_EXTENSIONS NO)
1441 TARGET_INCLUDE_DIRECTORIES(max-pooling-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1442 TARGET_LINK_LIBRARIES(max-pooling-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1443
Marat Dukhan95b22432019-10-30 16:30:14 -07001444 ADD_EXECUTABLE(prelu-bench bench/prelu.cc)
1445 SET_TARGET_PROPERTIES(prelu-bench PROPERTIES
1446 CXX_STANDARD 11
1447 CXX_STANDARD_REQUIRED YES
1448 CXX_EXTENSIONS NO)
1449 TARGET_INCLUDE_DIRECTORIES(prelu-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1450 TARGET_LINK_LIBRARIES(prelu-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1451
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001452 ADD_EXECUTABLE(sigmoid-bench bench/sigmoid.cc)
1453 SET_TARGET_PROPERTIES(sigmoid-bench PROPERTIES
1454 CXX_STANDARD 11
1455 CXX_STANDARD_REQUIRED YES
1456 CXX_EXTENSIONS NO)
1457 TARGET_INCLUDE_DIRECTORIES(sigmoid-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1458 TARGET_LINK_LIBRARIES(sigmoid-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1459
1460 ADD_EXECUTABLE(softargmax-bench bench/softargmax.cc)
1461 SET_TARGET_PROPERTIES(softargmax-bench PROPERTIES
1462 CXX_STANDARD 11
1463 CXX_STANDARD_REQUIRED YES
1464 CXX_EXTENSIONS NO)
1465 TARGET_INCLUDE_DIRECTORIES(softargmax-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1466 TARGET_LINK_LIBRARIES(softargmax-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1467
1468 # ---[ Build microkernel-level microbenchmarks
1469 ADD_EXECUTABLE(f16-gemm-bench bench/f16-gemm.cc)
1470 SET_TARGET_PROPERTIES(f16-gemm-bench PROPERTIES
1471 CXX_STANDARD 11
1472 CXX_STANDARD_REQUIRED YES
1473 CXX_EXTENSIONS YES)
1474 TARGET_INCLUDE_DIRECTORIES(f16-gemm-bench PRIVATE src)
1475 TARGET_INCLUDE_DIRECTORIES(f16-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1476 TARGET_LINK_LIBRARIES(f16-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1477
1478 ADD_EXECUTABLE(f32-conv-hwc-bench bench/f32-conv-hwc.cc)
1479 SET_TARGET_PROPERTIES(f32-conv-hwc-bench PROPERTIES
1480 CXX_STANDARD 11
1481 CXX_STANDARD_REQUIRED YES
1482 CXX_EXTENSIONS YES)
1483 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc-bench PRIVATE src)
1484 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1485 TARGET_LINK_LIBRARIES(f32-conv-hwc-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1486
1487 ADD_EXECUTABLE(f32-dwconv-spchw-bench bench/f32-dwconv-spchw.cc)
1488 SET_TARGET_PROPERTIES(f32-dwconv-spchw-bench PROPERTIES
1489 CXX_STANDARD 11
1490 CXX_STANDARD_REQUIRED YES
1491 CXX_EXTENSIONS YES)
1492 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-spchw-bench PRIVATE src)
1493 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-spchw-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1494 TARGET_LINK_LIBRARIES(f32-dwconv-spchw-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1495
1496 ADD_EXECUTABLE(f32-dwconv-bench bench/f32-dwconv.cc)
1497 SET_TARGET_PROPERTIES(f32-dwconv-bench PROPERTIES
1498 CXX_STANDARD 11
1499 CXX_STANDARD_REQUIRED YES
1500 CXX_EXTENSIONS YES)
1501 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-bench PRIVATE src)
1502 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1503 TARGET_LINK_LIBRARIES(f32-dwconv-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1504
1505 ADD_EXECUTABLE(f32-gemm-bench bench/f32-gemm.cc)
1506 SET_TARGET_PROPERTIES(f32-gemm-bench PROPERTIES
1507 CXX_STANDARD 11
1508 CXX_STANDARD_REQUIRED YES
1509 CXX_EXTENSIONS YES)
1510 TARGET_INCLUDE_DIRECTORIES(f32-gemm-bench PRIVATE src)
1511 TARGET_INCLUDE_DIRECTORIES(f32-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1512 TARGET_LINK_LIBRARIES(f32-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1513
1514 ADD_EXECUTABLE(f32-igemm-bench bench/f32-igemm.cc)
1515 SET_TARGET_PROPERTIES(f32-igemm-bench PROPERTIES
1516 CXX_STANDARD 11
1517 CXX_STANDARD_REQUIRED YES
1518 CXX_EXTENSIONS YES)
1519 TARGET_INCLUDE_DIRECTORIES(f32-igemm-bench PRIVATE src)
1520 TARGET_INCLUDE_DIRECTORIES(f32-igemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1521 TARGET_LINK_LIBRARIES(f32-igemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1522
1523 ADD_EXECUTABLE(f32-im2col-gemm-bench bench/f32-im2col-gemm.cc src/im2col.c)
1524 SET_TARGET_PROPERTIES(f32-im2col-gemm-bench PROPERTIES
1525 C_STANDARD 99
1526 C_STANDARD_REQUIRED YES
1527 C_EXTENSIONS NO
1528 CXX_STANDARD 11
1529 CXX_STANDARD_REQUIRED YES
1530 CXX_EXTENSIONS YES)
1531 TARGET_INCLUDE_DIRECTORIES(f32-im2col-gemm-bench PRIVATE src)
1532 TARGET_INCLUDE_DIRECTORIES(f32-im2col-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1533 TARGET_LINK_LIBRARIES(f32-im2col-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1534
1535 ADD_EXECUTABLE(f32-rmax-bench bench/f32-rmax.cc)
1536 SET_TARGET_PROPERTIES(f32-rmax-bench PROPERTIES
1537 CXX_STANDARD 11
1538 CXX_STANDARD_REQUIRED YES
1539 CXX_EXTENSIONS YES)
1540 TARGET_INCLUDE_DIRECTORIES(f32-rmax-bench PRIVATE src)
1541 TARGET_INCLUDE_DIRECTORIES(f32-rmax-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1542 TARGET_LINK_LIBRARIES(f32-rmax-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1543
1544 ADD_EXECUTABLE(f32-spmm-bench bench/f32-spmm.cc)
1545 SET_TARGET_PROPERTIES(f32-spmm-bench PROPERTIES
1546 CXX_STANDARD 11
1547 CXX_STANDARD_REQUIRED YES
1548 CXX_EXTENSIONS YES)
1549 TARGET_INCLUDE_DIRECTORIES(f32-spmm-bench PRIVATE src)
1550 TARGET_INCLUDE_DIRECTORIES(f32-spmm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1551 TARGET_LINK_LIBRARIES(f32-spmm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1552
Ashkan Aliabadi7892d972019-10-24 16:00:52 -07001553 ADD_EXECUTABLE(f32-softargmax-bench bench/f32-softargmax.cc)
1554 SET_TARGET_PROPERTIES(f32-softargmax-bench PROPERTIES
Marat Dukhan4a4a7fa2019-10-21 13:46:14 -07001555 CXX_STANDARD 11
1556 CXX_STANDARD_REQUIRED YES
1557 CXX_EXTENSIONS YES)
Ashkan Aliabadi7892d972019-10-24 16:00:52 -07001558 TARGET_INCLUDE_DIRECTORIES(f32-softargmax-bench PRIVATE src)
1559 TARGET_INCLUDE_DIRECTORIES(f32-softargmax-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1560 TARGET_LINK_LIBRARIES(f32-softargmax-bench PRIVATE XNNPACK fp16 benchmark bench-utils)
Marat Dukhan4a4a7fa2019-10-21 13:46:14 -07001561
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001562 ADD_EXECUTABLE(q8-gemm-bench bench/q8-gemm.cc)
1563 SET_TARGET_PROPERTIES(q8-gemm-bench PROPERTIES
1564 CXX_STANDARD 11
1565 CXX_STANDARD_REQUIRED YES
1566 CXX_EXTENSIONS YES)
1567 TARGET_INCLUDE_DIRECTORIES(q8-gemm-bench PRIVATE src)
1568 TARGET_INCLUDE_DIRECTORIES(q8-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1569 TARGET_LINK_LIBRARIES(q8-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1570ENDIF()