blob: 219fe7746f529eee59c6fced188f8b336703983c [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
Marat Dukhanc76065f2019-11-14 18:13:33 -0800150 src/resize-bilinear.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700151 src/sigmoid.c
152 src/softargmax.c
153 src/unpooling.c)
154
155SET(XNNPACK_COLD_SRCS ${XNNPACK_OPERATOR_SRCS})
156LIST(APPEND XNNPACK_COLD_SRCS
157 src/init.c
158 src/operator-delete.c)
159
160SET(XNNPACK_HOT_SRCS
161 src/indirection.c
162 src/operator-run.c)
163
164SET(XNNPACK_SCALAR_MICROKERNEL_SRCS
165 src/f32-argmaxpool/mp9p8q-scalar.c
166 src/f32-argmaxpool/up4-scalar.c
167 src/f32-argmaxpool/up9-scalar.c
168 src/f32-avgpool/mp9p8q-scalar.c
169 src/f32-avgpool/up9-scalar.c
Marat Dukhan35dacfb2019-11-07 19:18:16 -0800170 src/f32-bilinear/scalar-c1.c
171 src/f32-bilinear/scalar-c2.c
172 src/f32-bilinear/scalar-c4.c
Marat Dukhanc76065f2019-11-14 18:13:33 -0800173 src/f32-binop/vadd-scalar-x1.c
174 src/f32-binop/vadd-scalar-x2.c
175 src/f32-binop/vadd-scalar-x4.c
176 src/f32-binop/vaddc-scalar-x1.c
177 src/f32-binop/vaddc-scalar-x2.c
178 src/f32-binop/vaddc-scalar-x4.c
179 src/f32-binop/vmul-scalar-x1.c
180 src/f32-binop/vmul-scalar-x2.c
181 src/f32-binop/vmul-scalar-x4.c
182 src/f32-binop/vmulc-scalar-x1.c
183 src/f32-binop/vmulc-scalar-x2.c
184 src/f32-binop/vmulc-scalar-x4.c
185 src/f32-binop/vsub-scalar-x1.c
186 src/f32-binop/vsub-scalar-x2.c
187 src/f32-binop/vsub-scalar-x4.c
188 src/f32-binop/vsubc-scalar-x1.c
189 src/f32-binop/vsubc-scalar-x2.c
190 src/f32-binop/vsubc-scalar-x4.c
191 src/f32-binop/vrsubc-scalar-x1.c
192 src/f32-binop/vrsubc-scalar-x2.c
193 src/f32-binop/vrsubc-scalar-x4.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700194 src/f32-clamp/scalar.c
Erich Elsen563df5f2019-10-23 08:02:21 -0700195 src/f32-conv-hwc2spchw/3x3s2p1c3x4-scalar-1x1.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700196 src/f32-igemm/1x4-scalar.c
197 src/f32-igemm/2x4-scalar.c
198 src/f32-igemm/4x2-scalar.c
199 src/f32-igemm/4x4-scalar.c
200 src/f32-dwconv/up1x25-scalar.c
201 src/f32-dwconv/up1x4-scalar.c
202 src/f32-dwconv/up1x9-scalar.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800203 src/f32-dwconv/up1x25-scalar-acc2.c
204 src/f32-dwconv/up1x4-scalar-acc2.c
205 src/f32-dwconv/up1x9-scalar-acc2.c
206 src/f32-dwconv/up2x25-scalar.c
207 src/f32-dwconv/up2x4-scalar.c
208 src/f32-dwconv/up2x9-scalar.c
209 src/f32-dwconv/up2x25-scalar-acc2.c
210 src/f32-dwconv/up2x4-scalar-acc2.c
211 src/f32-dwconv/up2x9-scalar-acc2.c
Erich Elsen0cc2c532019-10-15 04:44:18 -0700212 src/f32-dwconv-spchw/3x3p1-scalar.c
Erich Elsen38709a62019-11-08 11:58:45 -0800213 src/f32-dwconv-spchw/5x5p2-scalar.c
Erich Elsenac4de802019-10-16 04:35:30 -0700214 src/f32-dwconv-spchw/3x3s2p1-scalar.c
Erich Elsen38709a62019-11-08 11:58:45 -0800215 src/f32-dwconv-spchw/5x5s2p2-scalar.c
Erich Elsen34dc2c02019-10-16 05:11:41 -0700216 src/f32-gavgpool-spchw/scalar-x1.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700217 src/f32-gavgpool/mp7p7q-scalar.c
218 src/f32-gavgpool/up7-scalar.c
219 src/f32-gemm/1x4-scalar.c
220 src/f32-gemm/2x4-scalar.c
221 src/f32-gemm/4x2-scalar.c
222 src/f32-gemm/4x4-scalar.c
223 src/f32-gemminc/1x4-scalar.c
224 src/f32-gemminc/2x4-scalar.c
225 src/f32-gemminc/4x4-scalar.c
226 src/f32-hswish/scalar.c
227 src/f32-maxpool/9p8q-scalar.c
228 src/f32-pavgpool/mp9p8q-scalar.c
229 src/f32-pavgpool/up9-scalar.c
230 src/f32-ppmm/2x4-scalar.c
231 src/f32-ppmm/3x3-scalar.c
232 src/f32-ppmm/4x2-scalar.c
233 src/f32-ppmm/4x4-scalar.c
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800234 src/f32-prelu/scalar-2x1.c
235 src/f32-prelu/scalar-2x4.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700236 src/f32-rmax/scalar.c
237 src/f32-spmm/1x1-scalar-pipelined.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700238 src/f32-spmm/1x1-scalar.c
239 src/f32-spmm/2x1-scalar-pipelined.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700240 src/f32-spmm/2x1-scalar.c
241 src/f32-spmm/4x1-scalar-pipelined.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700242 src/f32-spmm/4x1-scalar.c
243 src/f32-spmm/8x1-scalar-pipelined.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700244 src/f32-spmm/8x1-scalar.c
Erich Elsenc6afd9b2019-10-24 16:10:53 -0700245 src/f32-spmm/8x2-scalar.c
246 src/f32-spmm/8x4-scalar.c
Marat Dukhan49e6ee92019-11-06 15:55:29 -0800247 src/f32-vmulcaddc/c1-scalar-2x.c
248 src/f32-vmulcaddc/c2-scalar-2x.c
249 src/f32-vmulcaddc/c4-scalar-2x.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700250 src/q8-avgpool/mp9p8q-scalar.c
251 src/q8-avgpool/up9-scalar.c
252 src/q8-igemm/2x2-scalar.c
253 src/q8-dwconv/up1x9-scalar.c
254 src/q8-gavgpool/mp7p7q-scalar.c
255 src/q8-gavgpool/up7-scalar.c
256 src/q8-gemm/2x2-scalar.c
257 src/q8-vadd/scalar.c
258 src/u8-clamp/scalar.c
259 src/u8-lut32norm/scalar.c
260 src/u8-maxpool/9p8q-scalar.c
261 src/u8-rmax/scalar.c
262 src/x32-packx/x2-scalar.c
263 src/x32-packx/x3-scalar.c
264 src/x32-packx/x4-scalar.c
265 src/x32-pad/x2-scalar.c
266 src/x32-unpool/scalar.c
267 src/x32-zip/x2-scalar.c
268 src/x32-zip/x3-scalar.c
269 src/x32-zip/x4-scalar.c
270 src/x32-zip/xm-scalar.c
271 src/x8-lut/scalar.c
272 src/x8-zip/x2-scalar.c
273 src/x8-zip/x3-scalar.c
274 src/x8-zip/x4-scalar.c
275 src/x8-zip/xm-scalar.c)
276
277SET(XNNPACK_PSIMD_MICROKERNEL_SRCS
278 src/f32-argmaxpool/mp9p8q-psimd.c
279 src/f32-argmaxpool/up4-psimd.c
280 src/f32-argmaxpool/up9-psimd.c
281 src/f32-avgpool/mp9p8q-psimd.c
282 src/f32-avgpool/up9-psimd.c
Marat Dukhan35dacfb2019-11-07 19:18:16 -0800283 src/f32-bilinear/psimd-c4.c
284 src/f32-bilinear/psimd-c8.c
Marat Dukhanc76065f2019-11-14 18:13:33 -0800285 src/f32-binop/vadd-psimd-x4.c
286 src/f32-binop/vadd-psimd-x8.c
287 src/f32-binop/vaddc-psimd-x4.c
288 src/f32-binop/vaddc-psimd-x8.c
289 src/f32-binop/vmul-psimd-x4.c
290 src/f32-binop/vmul-psimd-x8.c
291 src/f32-binop/vmulc-psimd-x4.c
292 src/f32-binop/vmulc-psimd-x8.c
293 src/f32-binop/vrsubc-psimd-x4.c
294 src/f32-binop/vrsubc-psimd-x8.c
295 src/f32-binop/vsub-psimd-x4.c
296 src/f32-binop/vsub-psimd-x8.c
297 src/f32-binop/vsubc-psimd-x4.c
298 src/f32-binop/vsubc-psimd-x8.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700299 src/f32-clamp/psimd.c
300 src/f32-igemm/1x8-psimd-loadsplat.c
301 src/f32-igemm/1x8-psimd-splat.c
302 src/f32-igemm/1x8s4-psimd.c
303 src/f32-igemm/4x2c4-psimd.c
304 src/f32-igemm/4x8-psimd-loadsplat.c
305 src/f32-igemm/4x8-psimd-splat.c
306 src/f32-igemm/4x8s4-psimd.c
307 src/f32-igemm/6x8-psimd-loadsplat.c
308 src/f32-igemm/6x8-psimd-splat.c
309 src/f32-igemm/6x8s4-psimd.c
310 src/f32-dwconv/up4x25-psimd.c
311 src/f32-dwconv/up4x4-psimd.c
312 src/f32-dwconv/up4x9-psimd.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800313 src/f32-dwconv/up4x25-psimd-acc2.c
314 src/f32-dwconv/up4x4-psimd-acc2.c
315 src/f32-dwconv/up4x9-psimd-acc2.c
316 src/f32-dwconv/up8x25-psimd.c
317 src/f32-dwconv/up8x4-psimd.c
318 src/f32-dwconv/up8x9-psimd.c
319 src/f32-dwconv/up8x25-psimd-acc2.c
320 src/f32-dwconv/up8x4-psimd-acc2.c
321 src/f32-dwconv/up8x9-psimd-acc2.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700322 src/f32-gavgpool/mp7p7q-psimd.c
323 src/f32-gavgpool/up7-psimd.c
324 src/f32-gemm/1x8-psimd-loadsplat.c
325 src/f32-gemm/1x8-psimd-splat.c
326 src/f32-gemm/1x8s4-psimd.c
327 src/f32-gemm/4x8-psimd-loadsplat.c
328 src/f32-gemm/4x8-psimd-splat.c
329 src/f32-gemm/4x8s4-psimd.c
330 src/f32-gemm/6x8-psimd-loadsplat.c
331 src/f32-gemm/6x8-psimd-splat.c
332 src/f32-gemm/6x8s4-psimd.c
333 src/f32-gemminc/1x8-psimd-loadsplat.c
334 src/f32-gemminc/1x8-psimd-splat.c
335 src/f32-gemminc/1x8s4-psimd.c
336 src/f32-gemminc/4x8-psimd-loadsplat.c
337 src/f32-gemminc/4x8-psimd-splat.c
338 src/f32-gemminc/4x8s4-psimd.c
339 src/f32-gemminc/6x8-psimd-loadsplat.c
340 src/f32-gemminc/6x8-psimd-splat.c
341 src/f32-gemminc/6x8s4-psimd.c
342 src/f32-hswish/psimd.c
343 src/f32-maxpool/9p8q-psimd.c
344 src/f32-pavgpool/mp9p8q-psimd.c
345 src/f32-pavgpool/up9-psimd.c
346 src/f32-ppmm/4x8-psimd.c
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800347 src/f32-prelu/psimd-2x4.c
348 src/f32-prelu/psimd-2x8.c
Marat Dukhan49e6ee92019-11-06 15:55:29 -0800349 src/f32-vmulcaddc/c4-psimd-2x.c
350 src/f32-vmulcaddc/c8-psimd-2x.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700351 src/x32-packx/x4-psimd.c
352 src/x32-pad/x2-psimd.c
353 src/x32-unpool/psimd.c
354 src/x32-zip/x2-psimd.c
355 src/x32-zip/x3-psimd.c
356 src/x32-zip/x4-psimd.c
357 src/x32-zip/xm-psimd.c)
358
359SET(XNNPACK_NEON_MICROKERNEL_SRCS
360 src/f32-avgpool/mp9p8q-neon.c
361 src/f32-avgpool/up9-neon.c
Marat Dukhan35dacfb2019-11-07 19:18:16 -0800362 src/f32-bilinear/neon-c4.c
363 src/f32-bilinear/neon-c8.c
Marat Dukhanc76065f2019-11-14 18:13:33 -0800364 src/f32-binop/vadd-neon-x4.c
365 src/f32-binop/vadd-neon-x8.c
366 src/f32-binop/vaddc-neon-x4.c
367 src/f32-binop/vaddc-neon-x8.c
368 src/f32-binop/vmul-neon-x4.c
369 src/f32-binop/vmul-neon-x8.c
370 src/f32-binop/vmulc-neon-x4.c
371 src/f32-binop/vmulc-neon-x8.c
372 src/f32-binop/vrsubc-neon-x4.c
373 src/f32-binop/vrsubc-neon-x8.c
374 src/f32-binop/vsub-neon-x4.c
375 src/f32-binop/vsub-neon-x8.c
376 src/f32-binop/vsubc-neon-x4.c
377 src/f32-binop/vsubc-neon-x8.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700378 src/f32-clamp/neon.c
379 src/f32-igemm/1x8-neon-ld64.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700380 src/f32-igemm/4x2-neon-ld64.c
381 src/f32-igemm/4x4-neon-ld64.c
382 src/f32-igemm/4x8-neon-ld128.c
383 src/f32-igemm/4x8-neon-ld64.c
384 src/f32-igemm/6x8-neon-ld64.c
385 src/f32-dwconv/up4x9-neon.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800386 src/f32-dwconv/up4x9-neon-acc2.c
387 src/f32-dwconv/up8x9-neon.c
388 src/f32-dwconv/up8x9-neon-acc2.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700389 src/f32-gavgpool-spchw/neon-x4.c
390 src/f32-gavgpool/mp7p7q-neon.c
391 src/f32-gavgpool/up7-neon.c
392 src/f32-gemm/1x8-neon-ld64.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700393 src/f32-gemm/4x2-neon-ld64.c
394 src/f32-gemm/4x8-neon-ld128.c
395 src/f32-gemm/4x8-neon-ld64.c
396 src/f32-gemm/5x8-neon-ld64.c
397 src/f32-gemm/6x8-neon-ld64.c
398 src/f32-gemminc/1x8-neon-ld64.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700399 src/f32-gemminc/4x8-neon-ld128.c
400 src/f32-gemminc/4x8-neon-ld64.c
401 src/f32-gemminc/5x8-neon-ld64.c
402 src/f32-gemminc/6x8-neon-ld64.c
403 src/f32-hswish/neon.c
404 src/f32-pavgpool/mp9p8q-neon.c
405 src/f32-pavgpool/up9-neon.c
406 src/f32-ppmm/4x8-neon.c
407 src/f32-ppmm/8x8-neon.c
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800408 src/f32-prelu/neon-2x4.c
409 src/f32-prelu/neon-2x8.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700410 src/f32-rmax/neon.c
Marat Dukhan49e6ee92019-11-06 15:55:29 -0800411 src/f32-vmulcaddc/c4-neon-2x.c
412 src/f32-vmulcaddc/c8-neon-2x.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700413 src/q8-avgpool/mp9p8q-neon.c
414 src/q8-avgpool/up9-neon.c
415 src/q8-igemm/4x8-neon.c
416 src/q8-igemm/8x8-neon.c
417 src/q8-dwconv/up8x9-neon.c
418 src/q8-gavgpool/mp7p7q-neon.c
419 src/q8-gavgpool/up7-neon.c
420 src/q8-gemm/4x8-neon.c
421 src/q8-gemm/8x8-neon.c
422 src/q8-vadd/neon.c
423 src/u8-clamp/neon.c
424 src/u8-maxpool/9p8q-neon.c
425 src/u8-rmax/neon.c
426 src/x32-packx/x4-neon-st4.c
427 src/x32-pad/x2-neon.c
428 src/x32-zip/x2-neon.c
429 src/x32-zip/x3-neon.c
430 src/x32-zip/x4-neon.c
431 src/x32-zip/xm-neon.c
432 src/x8-zip/x2-neon.c
433 src/x8-zip/x3-neon.c
434 src/x8-zip/x4-neon.c
435 src/x8-zip/xm-neon.c)
436
437SET(XNNPACK_NEONFMA_MICROKERNEL_SRCS
Marat Dukhan35dacfb2019-11-07 19:18:16 -0800438 src/f32-bilinear/neonfma-c4.c
439 src/f32-bilinear/neonfma-c8.c
Frank Barcharddb45b6a2019-10-09 16:42:45 -0700440 src/f32-igemm/1x8-neonfma-ld64.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700441 src/f32-igemm/4x2-neonfma-ld64.c
442 src/f32-igemm/4x4-neonfma-ld64.c
443 src/f32-igemm/4x8-neonfma-ld128.c
444 src/f32-igemm/4x8-neonfma-ld64.c
445 src/f32-igemm/6x8-neonfma-ld64.c
446 src/f32-dwconv/up4x9-neonfma.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800447 src/f32-dwconv/up4x9-neonfma-acc2.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700448 src/f32-dwconv/up8x9-neonfma.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800449 src/f32-dwconv/up8x9-neonfma-acc2.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700450 src/f32-gemm/1x8-neonfma-ld64.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700451 src/f32-gemm/4x2-neonfma-ld64.c
452 src/f32-gemm/4x8-neonfma-ld128.c
453 src/f32-gemm/4x8-neonfma-ld64.c
454 src/f32-gemm/5x8-neonfma-ld64.c
455 src/f32-gemm/6x8-neonfma-ld64.c
456 src/f32-gemminc/1x8-neonfma-ld64.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700457 src/f32-gemminc/4x8-neonfma-ld128.c
458 src/f32-gemminc/4x8-neonfma-ld64.c
459 src/f32-gemminc/5x8-neonfma-ld64.c
460 src/f32-gemminc/6x8-neonfma-ld64.c
461 src/f32-hswish/neonfma.c
462 src/f32-ppmm/4x8-neonfma.c
463 src/f32-ppmm/8x8-neonfma.c
Marat Dukhan49e6ee92019-11-06 15:55:29 -0800464 src/f32-vmulcaddc/c4-neonfma-2x.c
465 src/f32-vmulcaddc/c8-neonfma-2x.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700466
467SET(AARCH64_XNNPACK_NEONFMA_MICROKERNEL_SRCS
468 src/f32-conv-hwc/3x3s2p1c3x4-neonfma-2x2.c
469 src/f32-conv-hwc/3x3s2p1c3x8-neonfma-2x2.c
470 src/f32-conv-hwc2spchw/3x3s2p1c3x4-neonfma-2x2.c
471 src/f32-dwconv-spchw/3x3p1-neonfma.c
472 src/f32-dwconv-spchw/5x5p2-neonfma.c
473 src/f32-dwconv-spchw/3x3s2p1-neonfma.c
474 src/f32-dwconv-spchw/5x5s2p2-neonfma.c
475 src/f32-spmm/12x1-neonfma.c
476 src/f32-spmm/12x2-neonfma.c
477 src/f32-spmm/12x4-neonfma.c
478 src/f32-spmm/16x1-neonfma-pipelined.c
479 src/f32-spmm/16x1-neonfma-unroll2.c
480 src/f32-spmm/16x1-neonfma.c
481 src/f32-spmm/16x2-neonfma.c
482 src/f32-spmm/16x4-neonfma.c
483 src/f32-spmm/4x1-neonfma-pipelined.c
484 src/f32-spmm/4x1-neonfma-unroll2.c
485 src/f32-spmm/4x1-neonfma.c
486 src/f32-spmm/4x2-neonfma.c
487 src/f32-spmm/4x4-neonfma.c
488 src/f32-spmm/8x1-neonfma-pipelined.c
489 src/f32-spmm/8x1-neonfma-unroll2.c
490 src/f32-spmm/8x1-neonfma.c
491 src/f32-spmm/8x2-neonfma.c
492 src/f32-spmm/8x4-neonfma.c)
493
494SET(XNNPACK_AARCH64_NEONFP16ARITH_MICROKERNEL_SRCS
495 src/f16-gemm/4x8-neonfp16arith-ld64.c
496 src/f16-gemm/6x8-neonfp16arith-ld64.c
497 src/f16-gemm/8x8-neonfp16arith-ld64.c)
498
499SET(XNNPACK_SSE_MICROKERNEL_SRCS
500 src/f32-avgpool/mp9p8q-sse.c
501 src/f32-avgpool/up9-sse.c
Marat Dukhan35dacfb2019-11-07 19:18:16 -0800502 src/f32-bilinear/sse-c4.c
503 src/f32-bilinear/sse-c8.c
Marat Dukhanc76065f2019-11-14 18:13:33 -0800504 src/f32-binop/vadd-sse-x4.c
505 src/f32-binop/vadd-sse-x8.c
506 src/f32-binop/vaddc-sse-x4.c
507 src/f32-binop/vaddc-sse-x8.c
508 src/f32-binop/vmul-sse-x4.c
509 src/f32-binop/vmul-sse-x8.c
510 src/f32-binop/vmulc-sse-x4.c
511 src/f32-binop/vmulc-sse-x8.c
512 src/f32-binop/vrsubc-sse-x4.c
513 src/f32-binop/vrsubc-sse-x8.c
514 src/f32-binop/vsub-sse-x4.c
515 src/f32-binop/vsub-sse-x8.c
516 src/f32-binop/vsubc-sse-x4.c
517 src/f32-binop/vsubc-sse-x8.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700518 src/f32-clamp/sse.c
519 src/f32-igemm/1x8-sse-dup.c
520 src/f32-igemm/1x8-sse-load1.c
521 src/f32-igemm/1x8s4-sse.c
522 src/f32-igemm/4x2c4-sse.c
523 src/f32-igemm/4x8-sse-dup.c
524 src/f32-igemm/4x8-sse-load1.c
525 src/f32-igemm/4x8s4-sse.c
526 src/f32-dwconv/up4x25-sse.c
527 src/f32-dwconv/up4x4-sse.c
528 src/f32-dwconv/up4x9-sse.c
Marat Dukhan5098c3e2019-11-07 12:01:19 -0800529 src/f32-dwconv/up4x25-sse-acc2.c
530 src/f32-dwconv/up4x4-sse-acc2.c
531 src/f32-dwconv/up4x9-sse-acc2.c
532 src/f32-dwconv/up8x25-sse.c
533 src/f32-dwconv/up8x4-sse.c
534 src/f32-dwconv/up8x9-sse.c
535 src/f32-dwconv/up8x25-sse-acc2.c
536 src/f32-dwconv/up8x4-sse-acc2.c
537 src/f32-dwconv/up8x9-sse-acc2.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700538 src/f32-gavgpool-spchw/sse-x4.c
539 src/f32-gavgpool/mp7p7q-sse.c
540 src/f32-gavgpool/up7-sse.c
541 src/f32-gemm/1x8-sse-dup.c
542 src/f32-gemm/1x8-sse-load1.c
543 src/f32-gemm/1x8s4-sse.c
544 src/f32-gemm/4x8-sse-dup.c
545 src/f32-gemm/4x8-sse-load1.c
546 src/f32-gemm/4x8s4-sse.c
547 src/f32-gemminc/1x8-sse-dup.c
548 src/f32-gemminc/1x8-sse-load1.c
549 src/f32-gemminc/1x8s4-sse.c
550 src/f32-gemminc/4x8-sse-dup.c
551 src/f32-gemminc/4x8-sse-load1.c
552 src/f32-gemminc/4x8s4-sse.c
553 src/f32-hswish/sse.c
554 src/f32-maxpool/9p8q-sse.c
555 src/f32-pavgpool/mp9p8q-sse.c
556 src/f32-pavgpool/up9-sse.c
557 src/f32-dwconv-spchw/3x3p1-sse.c
558 src/f32-dwconv-spchw/3x3s2p1-sse.c
559 src/f32-ppmm/4x8-sse.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700560 src/f32-rmax/sse.c
561 src/f32-spmm/4x1-sse.c
562 src/f32-spmm/8x1-sse.c
Marat Dukhan49e6ee92019-11-06 15:55:29 -0800563 src/f32-vmulcaddc/c4-sse-2x.c
564 src/f32-vmulcaddc/c8-sse-2x.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700565 src/x32-packx/x4-sse.c)
566
567SET(XNNPACK_SSE2_MICROKERNEL_SRCS
568 src/f32-argmaxpool/mp9p8q-sse2.c
569 src/f32-argmaxpool/up4-sse2.c
570 src/f32-argmaxpool/up9-sse2.c
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800571 src/f32-prelu/sse2-2x4.c
572 src/f32-prelu/sse2-2x8.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700573 src/q8-avgpool/mp9p8q-sse2.c
574 src/q8-avgpool/up9-sse2.c
575 src/q8-igemm/4x4c2-sse2.c
576 src/q8-dwconv/up8x9-sse2.c
577 src/q8-gavgpool/mp7p7q-sse2.c
578 src/q8-gavgpool/up7-sse2.c
579 src/q8-gemm/2x4c8-sse2.c
580 src/q8-gemm/4x4c2-sse2.c
581 src/q8-vadd/sse2.c
582 src/u8-clamp/sse2.c
583 src/u8-maxpool/9p8q-sse2.c
584 src/u8-rmax/sse2.c
585 src/x32-pad/x2-sse2.c
586 src/x32-zip/x2-sse2.c
587 src/x32-zip/x3-sse2.c
588 src/x32-zip/x4-sse2.c
589 src/x32-zip/xm-sse2.c
590 src/x8-zip/x2-sse2.c
591 src/x8-zip/x3-sse2.c
592 src/x8-zip/x4-sse2.c
593 src/x8-zip/xm-sse2.c)
594
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800595SET(XNNPACK_SSE41_MICROKERNEL_SRCS
596 src/f32-prelu/sse41-2x4.c
597 src/f32-prelu/sse41-2x8.c)
598
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700599SET(XNNPACK_AVX_MICROKERNEL_SRCS
Marat Dukhana5977df2019-10-30 22:02:15 -0700600 src/f32-rmax/avx.c
601 src/f32-vscale/avx-unroll32.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700602
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700603SET(XNNPACK_AVX2_MICROKERNEL_SRCS
Marat Dukhan97579532019-10-18 16:40:39 -0700604 src/f32-raddexpminusmax/avx2-p5-unroll64.c
Marat Dukhan6f8d4d32019-10-25 17:07:09 -0700605 src/f32-raddextexp/avx2-p5-unroll64.c
Marat Dukhan97579532019-10-18 16:40:39 -0700606 src/f32-raddstoreexpminusmax/avx2-p5-unroll64.c
607 src/f32-vscaleexpminusmax/avx2-p5-unroll64.c
Marat Dukhan6f8d4d32019-10-25 17:07:09 -0700608 src/f32-vscaleextexp/avx2-p5-unroll64.c
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700609 src/math/exp-avx2-p5.c
610 src/math/exp-avx2-perm-p3.c
Marat Dukhan515c9772019-10-17 18:07:57 -0700611 src/math/exp-avx2-perm-p4.c
612 src/math/expminus-avx2-p5.c)
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700613
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700614SET(XNNPACK_AVX512F_MICROKERNEL_SRCS
Marat Dukhan97579532019-10-18 16:40:39 -0700615 src/f32-raddexpminusmax/avx512f-p5-scalef-unroll128.c
Marat Dukhan6f8d4d32019-10-25 17:07:09 -0700616 src/f32-raddextexp/avx512f-p5-scalef-unroll128.c
Marat Dukhan97579532019-10-18 16:40:39 -0700617 src/f32-raddstoreexpminusmax/avx512f-p5-scalef-unroll128.c
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700618 src/f32-rmax/avx512f.c
Marat Dukhana5977df2019-10-30 22:02:15 -0700619 src/f32-vscale/avx512f-unroll64.c
Marat Dukhan97579532019-10-18 16:40:39 -0700620 src/f32-vscaleexpminusmax/avx512f-p5-scalef-unroll128.c
Marat Dukhan6f8d4d32019-10-25 17:07:09 -0700621 src/f32-vscaleextexp/avx512f-p5-scalef-unroll128.c
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700622 src/math/exp-avx512f-p5-scalef.c
623 src/math/exp-avx512f-p5.c
Marat Dukhanfeb49232019-10-28 11:03:31 -0700624 src/math/exp-avx512f-perm-p3.c
625 src/math/exp-avx512f-perm2-p2.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700626
627SET(XNNPACK_AARCH32_ASM_MICROKERNEL_SRCS
628 src/q8-dwconv/up8x9-aarch32-neon.S)
629
630SET(XNNPACK_AARCH64_ASM_MICROKERNEL_SRCS
631 src/f32-dwconv/up4x9-aarch64-neonfma-cortex-a55.S
632 src/f32-dwconv/up4x9-aarch64-neonfma.S
633 src/f32-gemm/1x12-aarch64-neonfma-cortex-a53.S
Frank Barchard21be34f2019-10-09 19:32:19 -0700634 src/f32-gemm/1x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700635 src/f32-gemm/1x8-aarch64-neonfma-cortex-a57.S
636 src/f32-gemm/1x8-aarch64-neonfma-cortex-a75.S
637 src/f32-gemm/4x12-aarch64-neonfma-cortex-a53.S
Marat Dukhan8e6e9972019-10-31 14:15:31 -0700638 src/f32-gemm/4x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700639 src/f32-gemm/4x8-aarch64-neonfma-cortex-a57.S
640 src/f32-gemm/4x8-aarch64-neonfma-cortex-a75.S
641 src/f32-gemm/4x8-aarch64-neonfma-ld128.S
642 src/f32-gemm/4x8-aarch64-neonfma-ld64.S
643 src/f32-gemm/5x8-aarch64-neonfma-cortex-a75.S
Marat Dukhan8e6e9972019-10-31 14:15:31 -0700644 src/f32-gemm/6x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700645 src/f32-gemm/6x8-aarch64-neonfma-cortex-a57.S
646 src/f32-gemm/6x8-aarch64-neonfma-cortex-a73.S
647 src/f32-gemm/6x8-aarch64-neonfma-cortex-a75.S
648 src/f32-gemm/6x8-aarch64-neonfma-ld128.S
649 src/f32-gemm/6x8-aarch64-neonfma-ld64.S
650 src/f32-gemminc/1x12-aarch64-neonfma-cortex-a53.S
Frank Barchard21be34f2019-10-09 19:32:19 -0700651 src/f32-gemminc/1x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700652 src/f32-gemminc/1x8-aarch64-neonfma-cortex-a57.S
653 src/f32-gemminc/1x8-aarch64-neonfma-cortex-a75.S
654 src/f32-gemminc/4x12-aarch64-neonfma-cortex-a53.S
Marat Dukhan8e6e9972019-10-31 14:15:31 -0700655 src/f32-gemminc/4x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700656 src/f32-gemminc/4x8-aarch64-neonfma-cortex-a57.S
657 src/f32-gemminc/4x8-aarch64-neonfma-cortex-a75.S
658 src/f32-gemminc/4x8-aarch64-neonfma-ld128.S
659 src/f32-gemminc/4x8-aarch64-neonfma-ld64.S
660 src/f32-gemminc/5x8-aarch64-neonfma-cortex-a75.S
Marat Dukhan8e6e9972019-10-31 14:15:31 -0700661 src/f32-gemminc/6x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700662 src/f32-gemminc/6x8-aarch64-neonfma-cortex-a57.S
663 src/f32-gemminc/6x8-aarch64-neonfma-cortex-a73.S
664 src/f32-gemminc/6x8-aarch64-neonfma-cortex-a75.S
665 src/f32-gemminc/6x8-aarch64-neonfma-ld128.S
666 src/f32-gemminc/6x8-aarch64-neonfma-ld64.S
667 src/f32-igemm/1x12-aarch64-neonfma-cortex-a53.S
Frank Barchard21be34f2019-10-09 19:32:19 -0700668 src/f32-igemm/1x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700669 src/f32-igemm/1x8-aarch64-neonfma-cortex-a57.S
670 src/f32-igemm/1x8-aarch64-neonfma-cortex-a75.S
671 src/f32-igemm/4x12-aarch64-neonfma-cortex-a53.S
672 src/f32-igemm/4x8-aarch64-neonfma-cortex-a75.S
673 src/f32-igemm/5x8-aarch64-neonfma-cortex-a75.S
Marat Dukhan8e6e9972019-10-31 14:15:31 -0700674 src/f32-igemm/6x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700675 src/f32-igemm/6x8-aarch64-neonfma-cortex-a57.S
676 src/f32-igemm/6x8-aarch64-neonfma-cortex-a73.S
677 src/f32-igemm/6x8-aarch64-neonfma-cortex-a75.S)
678
679SET(XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SCALAR_MICROKERNEL_SRCS})
680IF(NOT CMAKE_SYSTEM_NAME STREQUAL "Emscripten")
681 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_PSIMD_MICROKERNEL_SRCS})
682ENDIF()
683IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^armv[5-8]" OR IOS_ARCH MATCHES "^armv7")
684 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEON_MICROKERNEL_SRCS})
685 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEONFMA_MICROKERNEL_SRCS})
686 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AARCH32_ASM_MICROKERNEL_SRCS})
687ENDIF()
688IF(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64" OR IOS_ARCH MATCHES "^arm64.*")
689 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEON_MICROKERNEL_SRCS})
690 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEONFMA_MICROKERNEL_SRCS})
691 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${AARCH64_XNNPACK_NEONFMA_MICROKERNEL_SRCS})
692 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AARCH64_NEONFP16ARITH_MICROKERNEL_SRCS})
693 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AARCH64_ASM_MICROKERNEL_SRCS})
694ENDIF()
695IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^(i[3-6]86|x86_64)$" OR IOS_ARCH MATCHES "^(i386|x86_64)$")
696 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SSE_MICROKERNEL_SRCS})
697 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SSE2_MICROKERNEL_SRCS})
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800698 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SSE41_MICROKERNEL_SRCS})
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700699 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AVX_MICROKERNEL_SRCS})
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700700 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AVX2_MICROKERNEL_SRCS})
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700701 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AVX512F_MICROKERNEL_SRCS})
702ENDIF()
703
704IF(XNNPACK_LIBRARY_TYPE STREQUAL "default")
705 ADD_LIBRARY(XNNPACK ${XNNPACK_COLD_SRCS} ${XNNPACK_HOT_SRCS} ${XNNPACK_MICROKERNEL_SRCS})
706ELSEIF(XNNPACK_LIBRARY_TYPE STREQUAL "shared")
707 ADD_LIBRARY(XNNPACK SHARED ${XNNPACK_COLD_SRCS} ${XNNPACK_HOT_SRCS} ${XNNPACK_MICROKERNEL_SRCS})
708ELSEIF(XNNPACK_LIBRARY_TYPE STREQUAL "static")
709 ADD_LIBRARY(XNNPACK STATIC ${XNNPACK_COLD_SRCS} ${XNNPACK_HOT_SRCS} ${XNNPACK_MICROKERNEL_SRCS})
710ELSE()
711 MESSAGE(FATAL_ERROR "Unsupported XNNPACK library type \"${XNNPACK_LIBRARY_TYPE}\". Must be \"static\", \"shared\", or \"default\"")
712ENDIF()
713SET_TARGET_PROPERTIES(XNNPACK PROPERTIES
714 C_STANDARD 99
715 C_EXTENSIONS YES)
716IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^armv[5-8]" OR IOS_ARCH MATCHES "^armv7")
717 SET_PROPERTY(SOURCE ${XNNPACK_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -marm ")
718 SET_PROPERTY(SOURCE ${XNNPACK_PSIMD_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfpu=neon ")
719 SET_PROPERTY(SOURCE ${XNNPACK_NEON_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfpu=neon ")
720 SET_PROPERTY(SOURCE ${XNNPACK_NEONFMA_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfpu=neon-vfpv4 ")
721 IF(IOS)
722 SET_PROPERTY(SOURCE ${XNNPACK_AARCH32_ASM_UKERNELS} APPEND_STRING PROPERTY COMPILE_FLAGS " -arch ${IOS_ARCH} ")
723 ENDIF()
724ENDIF()
725IF(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64" OR IOS_ARCH MATCHES "^arm64.*")
726 SET_PROPERTY(SOURCE ${XNNPACK_AARCH64_NEONFP16ARITH_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -march=armv8.2-a+fp16 ")
727 IF(IOS)
728 SET_PROPERTY(SOURCE ${XNNPACK_AARCH64_ASM_UKERNELS} APPEND_STRING PROPERTY COMPILE_FLAGS " -arch ${IOS_ARCH} ")
729 ENDIF()
730ENDIF()
731IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^(i[3-6]86|x86_64)$" OR IOS_ARCH MATCHES "^(i386|x86_64)$")
732 SET_PROPERTY(SOURCE ${XNNPACK_SSE_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -msse ")
733 SET_PROPERTY(SOURCE ${XNNPACK_SSE2_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -msse2 ")
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800734 SET_PROPERTY(SOURCE ${XNNPACK_SSE41_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -msse4.1 ")
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700735 SET_PROPERTY(SOURCE ${XNNPACK_AVX_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mavx ")
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700736 SET_PROPERTY(SOURCE ${XNNPACK_AVX2_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfma -mavx2 ")
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700737 SET_PROPERTY(SOURCE ${XNNPACK_AVX512F_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mavx512f ")
738ENDIF()
739IF(CMAKE_BUILD_TYPE STREQUAL "Debug")
740 TARGET_COMPILE_DEFINITIONS(XNNPACK PRIVATE XNN_LOG_LEVEL=5)
741ELSE()
742 TARGET_COMPILE_DEFINITIONS(XNNPACK PRIVATE XNN_LOG_LEVEL=0)
743 SET_PROPERTY(SOURCE ${XNNPACK_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -O2 ")
744 SET_PROPERTY(SOURCE ${XNNPACK_HOT_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -O2 ")
745 SET_PROPERTY(SOURCE ${XNNPACK_COLD_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -Os ")
746ENDIF()
747IF(NOT CMAKE_SYSTEM_NAME STREQUAL "Emscripten")
748 SET_PROPERTY(SOURCE ${XNNPACK_PSIMD_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -O3 -ffast-math ")
749ENDIF()
750
751TARGET_INCLUDE_DIRECTORIES(XNNPACK PUBLIC include)
752TARGET_INCLUDE_DIRECTORIES(XNNPACK PRIVATE src)
753SET_TARGET_PROPERTIES(XNNPACK PROPERTIES PUBLIC_HEADER include/XNNPACK.h)
754
755# ---[ Configure clog
756IF(NOT TARGET clog)
757 SET(CLOG_BUILD_TESTS OFF CACHE BOOL "")
758 SET(CLOG_RUNTIME_TYPE "${CPUINFO_RUNTIME_TYPE}" CACHE STRING "")
759 ADD_SUBDIRECTORY(
760 "${CLOG_SOURCE_DIR}/deps/clog"
761 "${CMAKE_BINARY_DIR}/clog")
762 # We build static version of clog but a dynamic library may indirectly depend on it
763 SET_PROPERTY(TARGET clog PROPERTY POSITION_INDEPENDENT_CODE ON)
764ENDIF()
765TARGET_LINK_LIBRARIES(XNNPACK PRIVATE clog)
766
767# ---[ Configure cpuinfo
768IF(NOT TARGET cpuinfo)
769 SET(CPUINFO_BUILD_TOOLS OFF CACHE BOOL "")
770 SET(CPUINFO_BUILD_UNIT_TESTS OFF CACHE BOOL "")
771 SET(CPUINFO_BUILD_MOCK_TESTS OFF CACHE BOOL "")
772 SET(CPUINFO_BUILD_BENCHMARKS OFF CACHE BOOL "")
773 ADD_SUBDIRECTORY(
774 "${CPUINFO_SOURCE_DIR}"
775 "${CMAKE_BINARY_DIR}/cpuinfo")
776ENDIF()
777TARGET_LINK_LIBRARIES(XNNPACK PRIVATE cpuinfo)
778
779# ---[ Configure pthreadpool
780IF(NOT TARGET pthreadpool)
781 SET(PTHREADPOOL_BUILD_TESTS OFF CACHE BOOL "")
782 SET(PTHREADPOOL_BUILD_BENCHMARKS OFF CACHE BOOL "")
Marat Dukhand70028a2019-10-29 09:53:16 -0700783 SET(PTHREADPOOL_ALLOW_DEPRECATED_API OFF CACHE BOOL "")
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700784 ADD_SUBDIRECTORY(
785 "${PTHREADPOOL_SOURCE_DIR}"
786 "${CMAKE_BINARY_DIR}/pthreadpool")
787ENDIF()
788TARGET_LINK_LIBRARIES(XNNPACK PUBLIC pthreadpool)
789
790# ---[ Configure FXdiv
791IF(NOT TARGET fxdiv)
792 SET(FXDIV_BUILD_TESTS OFF CACHE BOOL "")
793 SET(FXDIV_BUILD_BENCHMARKS OFF CACHE BOOL "")
794 ADD_SUBDIRECTORY(
795 "${FXDIV_SOURCE_DIR}"
796 "${CMAKE_BINARY_DIR}/FXdiv")
797ENDIF()
798TARGET_LINK_LIBRARIES(XNNPACK PRIVATE fxdiv)
799
800# ---[ Configure psimd
801IF(NOT TARGET psimd)
802 ADD_SUBDIRECTORY(
803 "${PSIMD_SOURCE_DIR}"
804 "${CMAKE_BINARY_DIR}/psimd")
805ENDIF()
806TARGET_LINK_LIBRARIES(XNNPACK PRIVATE psimd)
807
808# ---[ Configure FP16
809IF(NOT TARGET fp16)
810 SET(FP16_BUILD_TESTS OFF CACHE BOOL "")
811 SET(FP16_BUILD_BENCHMARKS OFF CACHE BOOL "")
812 ADD_SUBDIRECTORY(
813 "${FP16_SOURCE_DIR}"
814 "${CMAKE_BINARY_DIR}/FP16")
815ENDIF()
816TARGET_LINK_LIBRARIES(XNNPACK PRIVATE fp16)
817
818INSTALL(TARGETS XNNPACK
819 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
820 ARCHIVE DESTINATION ${CMAKE_INSTALL_LIBDIR}
821 PUBLIC_HEADER DESTINATION ${CMAKE_INSTALL_INCLUDEDIR})
822
823# ---[ XNNPACK unit tests
824IF(XNNPACK_BUILD_TESTS)
825 # ---[ Build google test
826 IF(NOT TARGET gtest)
827 SET(gtest_force_shared_crt ON CACHE BOOL "" FORCE)
828 ADD_SUBDIRECTORY(
829 "${GOOGLETEST_SOURCE_DIR}"
830 "${CMAKE_BINARY_DIR}/googletest")
831 ENDIF()
832
833 # ---[ Build operator-level unit tests
834 ADD_EXECUTABLE(add-test test/add.cc)
835 SET_TARGET_PROPERTIES(add-test PROPERTIES
836 CXX_STANDARD 11
837 CXX_STANDARD_REQUIRED YES
838 CXX_EXTENSIONS NO)
839 TARGET_INCLUDE_DIRECTORIES(add-test PRIVATE src test)
840 TARGET_LINK_LIBRARIES(add-test PRIVATE XNNPACK gtest gtest_main)
841 ADD_TEST(add-test add-test)
842
843 ADD_EXECUTABLE(argmax-pooling-test test/argmax-pooling.cc)
844 SET_TARGET_PROPERTIES(argmax-pooling-test PROPERTIES
845 CXX_STANDARD 11
846 CXX_STANDARD_REQUIRED YES
847 CXX_EXTENSIONS NO)
848 TARGET_INCLUDE_DIRECTORIES(argmax-pooling-test PRIVATE src test)
849 TARGET_LINK_LIBRARIES(argmax-pooling-test PRIVATE XNNPACK gtest gtest_main)
850 ADD_TEST(argmax-pooling-test argmax-pooling-test)
851
852 ADD_EXECUTABLE(average-pooling-test test/average-pooling.cc)
853 SET_TARGET_PROPERTIES(average-pooling-test PROPERTIES
854 CXX_STANDARD 11
855 CXX_STANDARD_REQUIRED YES
856 CXX_EXTENSIONS NO)
857 TARGET_INCLUDE_DIRECTORIES(average-pooling-test PRIVATE src test)
858 TARGET_LINK_LIBRARIES(average-pooling-test PRIVATE XNNPACK gtest gtest_main)
859 ADD_TEST(average-pooling-test average-pooling-test)
860
861 ADD_EXECUTABLE(channel-pad-test test/channel-pad.cc)
862 SET_TARGET_PROPERTIES(channel-pad-test PROPERTIES
863 CXX_STANDARD 11
864 CXX_STANDARD_REQUIRED YES
865 CXX_EXTENSIONS NO)
866 TARGET_INCLUDE_DIRECTORIES(channel-pad-test PRIVATE src test)
867 TARGET_LINK_LIBRARIES(channel-pad-test PRIVATE XNNPACK gtest gtest_main)
868 ADD_TEST(channel-pad-test channel-pad-test)
869
870 ADD_EXECUTABLE(channel-shuffle-test test/channel-shuffle.cc)
871 SET_TARGET_PROPERTIES(channel-shuffle-test PROPERTIES
872 CXX_STANDARD 11
873 CXX_STANDARD_REQUIRED YES
874 CXX_EXTENSIONS NO)
875 TARGET_INCLUDE_DIRECTORIES(channel-shuffle-test PRIVATE src test)
876 TARGET_LINK_LIBRARIES(channel-shuffle-test PRIVATE XNNPACK gtest gtest_main)
877 ADD_TEST(channel-shuffle-test channel-shuffle-test)
878
879 ADD_EXECUTABLE(clamp-test test/clamp.cc)
880 SET_TARGET_PROPERTIES(clamp-test PROPERTIES
881 CXX_STANDARD 11
882 CXX_STANDARD_REQUIRED YES
883 CXX_EXTENSIONS NO)
884 TARGET_INCLUDE_DIRECTORIES(clamp-test PRIVATE src test)
885 TARGET_LINK_LIBRARIES(clamp-test PRIVATE XNNPACK gtest gtest_main)
886 ADD_TEST(clamp-test clamp-test)
887
888 ADD_EXECUTABLE(convolution-test test/convolution.cc)
889 SET_TARGET_PROPERTIES(convolution-test PROPERTIES
890 CXX_STANDARD 11
891 CXX_STANDARD_REQUIRED YES
892 CXX_EXTENSIONS NO)
893 TARGET_INCLUDE_DIRECTORIES(convolution-test PRIVATE src test)
894 TARGET_LINK_LIBRARIES(convolution-test PRIVATE XNNPACK gtest gtest_main)
895 ADD_TEST(convolution-test convolution-test)
896
897 ADD_EXECUTABLE(convolution-spnchw-test test/convolution-spnchw.cc)
898 SET_TARGET_PROPERTIES(convolution-spnchw-test PROPERTIES
899 CXX_STANDARD 11
900 CXX_STANDARD_REQUIRED YES
901 CXX_EXTENSIONS NO)
902 TARGET_INCLUDE_DIRECTORIES(convolution-spnchw-test PRIVATE src test)
903 TARGET_LINK_LIBRARIES(convolution-spnchw-test PRIVATE XNNPACK gtest gtest_main)
904 ADD_TEST(convolution-spnchw-test convolution-spnchw-test)
905
906 ADD_EXECUTABLE(deconvolution-test test/deconvolution.cc)
907 SET_TARGET_PROPERTIES(deconvolution-test PROPERTIES
908 CXX_STANDARD 11
909 CXX_STANDARD_REQUIRED YES
910 CXX_EXTENSIONS NO)
911 TARGET_INCLUDE_DIRECTORIES(deconvolution-test PRIVATE src test)
912 TARGET_LINK_LIBRARIES(deconvolution-test PRIVATE XNNPACK gtest gtest_main)
913 ADD_TEST(deconvolution-test deconvolution-test)
914
915 ADD_EXECUTABLE(fully-connected-test test/fully-connected.cc)
916 SET_TARGET_PROPERTIES(fully-connected-test PROPERTIES
917 CXX_STANDARD 11
918 CXX_STANDARD_REQUIRED YES
919 CXX_EXTENSIONS NO)
920 TARGET_INCLUDE_DIRECTORIES(fully-connected-test PRIVATE src test)
921 TARGET_LINK_LIBRARIES(fully-connected-test PRIVATE XNNPACK gtest gtest_main)
922 ADD_TEST(fully-connected-test fully-connected-test)
923
924 ADD_EXECUTABLE(global-average-pooling-test test/global-average-pooling.cc)
925 SET_TARGET_PROPERTIES(global-average-pooling-test PROPERTIES
926 CXX_STANDARD 11
927 CXX_STANDARD_REQUIRED YES
928 CXX_EXTENSIONS NO)
929 TARGET_INCLUDE_DIRECTORIES(global-average-pooling-test PRIVATE src test)
930 TARGET_LINK_LIBRARIES(global-average-pooling-test PRIVATE XNNPACK gtest gtest_main)
931 ADD_TEST(global-average-pooling-test global-average-pooling-test)
932
933 ADD_EXECUTABLE(global-average-pooling-spnchw-test test/global-average-pooling-spnchw.cc)
934 SET_TARGET_PROPERTIES(global-average-pooling-spnchw-test PROPERTIES
935 CXX_STANDARD 11
936 CXX_STANDARD_REQUIRED YES
937 CXX_EXTENSIONS NO)
938 TARGET_INCLUDE_DIRECTORIES(global-average-pooling-spnchw-test PRIVATE src test)
939 TARGET_LINK_LIBRARIES(global-average-pooling-spnchw-test PRIVATE XNNPACK gtest gtest_main)
940 ADD_TEST(global-average-pooling-spnchw-test global-average-pooling-spnchw-test)
941
942 ADD_EXECUTABLE(hardswish-test test/hardswish.cc)
943 SET_TARGET_PROPERTIES(hardswish-test PROPERTIES
944 CXX_STANDARD 11
945 CXX_STANDARD_REQUIRED YES
946 CXX_EXTENSIONS NO)
947 TARGET_INCLUDE_DIRECTORIES(hardswish-test PRIVATE src test)
948 TARGET_LINK_LIBRARIES(hardswish-test PRIVATE XNNPACK gtest gtest_main)
949 ADD_TEST(hardswish-test hardswish-test)
950
951 ADD_EXECUTABLE(leaky-relu-test test/leaky-relu.cc)
952 SET_TARGET_PROPERTIES(leaky-relu-test PROPERTIES
953 CXX_STANDARD 11
954 CXX_STANDARD_REQUIRED YES
955 CXX_EXTENSIONS NO)
956 TARGET_INCLUDE_DIRECTORIES(leaky-relu-test PRIVATE src test)
957 TARGET_LINK_LIBRARIES(leaky-relu-test PRIVATE XNNPACK gtest gtest_main)
958 ADD_TEST(leaky-relu-test leaky-relu-test)
959
960 ADD_EXECUTABLE(max-pooling-test test/max-pooling.cc)
961 SET_TARGET_PROPERTIES(max-pooling-test PROPERTIES
962 CXX_STANDARD 11
963 CXX_STANDARD_REQUIRED YES
964 CXX_EXTENSIONS NO)
965 TARGET_INCLUDE_DIRECTORIES(max-pooling-test PRIVATE src test)
966 TARGET_LINK_LIBRARIES(max-pooling-test PRIVATE XNNPACK gtest gtest_main)
967 ADD_TEST(max-pooling-test max-pooling-test)
968
969 ADD_EXECUTABLE(prelu-test test/prelu.cc)
970 SET_TARGET_PROPERTIES(prelu-test PROPERTIES
971 CXX_STANDARD 11
972 CXX_STANDARD_REQUIRED YES
973 CXX_EXTENSIONS NO)
974 TARGET_INCLUDE_DIRECTORIES(prelu-test PRIVATE src test)
975 TARGET_LINK_LIBRARIES(prelu-test PRIVATE XNNPACK gtest gtest_main)
976 ADD_TEST(prelu-test prelu-test)
977
Marat Dukhan69722492019-11-11 19:55:50 -0800978 ADD_EXECUTABLE(resize-bilinear-test test/resize-bilinear.cc)
979 SET_TARGET_PROPERTIES(resize-bilinear-test PROPERTIES
980 CXX_STANDARD 11
981 CXX_STANDARD_REQUIRED YES
982 CXX_EXTENSIONS NO)
983 TARGET_INCLUDE_DIRECTORIES(resize-bilinear-test PRIVATE src test)
984 TARGET_LINK_LIBRARIES(resize-bilinear-test PRIVATE XNNPACK gtest gtest_main)
985 ADD_TEST(resize-bilinear-test resize-bilinear-test)
986
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700987 ADD_EXECUTABLE(sigmoid-test test/sigmoid.cc)
988 SET_TARGET_PROPERTIES(sigmoid-test PROPERTIES
989 CXX_STANDARD 11
990 CXX_STANDARD_REQUIRED YES
991 CXX_EXTENSIONS NO)
992 TARGET_INCLUDE_DIRECTORIES(sigmoid-test PRIVATE src test)
993 TARGET_LINK_LIBRARIES(sigmoid-test PRIVATE XNNPACK gtest gtest_main)
994 ADD_TEST(sigmoid-test sigmoid-test)
995
996 ADD_EXECUTABLE(softargmax-test test/softargmax.cc)
997 SET_TARGET_PROPERTIES(softargmax-test PROPERTIES
998 CXX_STANDARD 11
999 CXX_STANDARD_REQUIRED YES
1000 CXX_EXTENSIONS NO)
1001 TARGET_INCLUDE_DIRECTORIES(softargmax-test PRIVATE src test)
1002 TARGET_LINK_LIBRARIES(softargmax-test PRIVATE XNNPACK gtest gtest_main)
1003 ADD_TEST(softargmax-test softargmax-test)
1004
1005 ADD_EXECUTABLE(unpooling-test test/unpooling.cc)
1006 SET_TARGET_PROPERTIES(unpooling-test PROPERTIES
1007 CXX_STANDARD 11
1008 CXX_STANDARD_REQUIRED YES
1009 CXX_EXTENSIONS NO)
1010 TARGET_INCLUDE_DIRECTORIES(unpooling-test PRIVATE src test)
1011 TARGET_LINK_LIBRARIES(unpooling-test PRIVATE XNNPACK gtest gtest_main)
1012 ADD_TEST(unpooling-test unpooling-test)
1013
1014 # ---[ Build microkernel-level unit tests
1015 ADD_EXECUTABLE(f16-gemm-test test/f16-gemm.cc)
1016 SET_TARGET_PROPERTIES(f16-gemm-test PROPERTIES
1017 CXX_STANDARD 11
1018 CXX_STANDARD_REQUIRED YES
1019 CXX_EXTENSIONS YES)
1020 TARGET_INCLUDE_DIRECTORIES(f16-gemm-test PRIVATE src test)
1021 TARGET_LINK_LIBRARIES(f16-gemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1022 ADD_TEST(f16-gemm-test f16-gemm-test)
1023
1024 ADD_EXECUTABLE(f32-argmaxpool-test test/f32-argmaxpool.cc)
1025 SET_TARGET_PROPERTIES(f32-argmaxpool-test PROPERTIES
1026 CXX_STANDARD 11
1027 CXX_STANDARD_REQUIRED YES
1028 CXX_EXTENSIONS YES)
1029 TARGET_INCLUDE_DIRECTORIES(f32-argmaxpool-test PRIVATE src test)
1030 TARGET_LINK_LIBRARIES(f32-argmaxpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1031 ADD_TEST(f32-argmaxpool-test f32-argmaxpool-test)
1032
1033 ADD_EXECUTABLE(f32-avgpool-test test/f32-avgpool.cc)
1034 SET_TARGET_PROPERTIES(f32-avgpool-test PROPERTIES
1035 CXX_STANDARD 11
1036 CXX_STANDARD_REQUIRED YES
1037 CXX_EXTENSIONS YES)
1038 TARGET_INCLUDE_DIRECTORIES(f32-avgpool-test PRIVATE src test)
1039 TARGET_LINK_LIBRARIES(f32-avgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1040 ADD_TEST(f32-avgpool-test f32-avgpool-test)
1041
Marat Dukhanc465fc22019-11-07 19:20:28 -08001042 ADD_EXECUTABLE(f32-bilinear-test test/f32-bilinear.cc)
1043 SET_TARGET_PROPERTIES(f32-bilinear-test PROPERTIES
1044 CXX_STANDARD 11
1045 CXX_STANDARD_REQUIRED YES
1046 CXX_EXTENSIONS YES)
1047 TARGET_INCLUDE_DIRECTORIES(f32-bilinear-test PRIVATE src test)
1048 TARGET_LINK_LIBRARIES(f32-bilinear-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1049 ADD_TEST(f32-bilinear-test f32-bilinear-test)
1050
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001051 ADD_EXECUTABLE(f32-clamp-test test/f32-clamp.cc)
1052 SET_TARGET_PROPERTIES(f32-clamp-test PROPERTIES
1053 CXX_STANDARD 11
1054 CXX_STANDARD_REQUIRED YES
1055 CXX_EXTENSIONS YES)
1056 TARGET_INCLUDE_DIRECTORIES(f32-clamp-test PRIVATE src test)
1057 TARGET_LINK_LIBRARIES(f32-clamp-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1058 ADD_TEST(f32-clamp-test f32-clamp-test)
1059
1060 ADD_EXECUTABLE(f32-conv-hwc-test test/f32-conv-hwc.cc)
1061 SET_TARGET_PROPERTIES(f32-conv-hwc-test PROPERTIES
1062 CXX_STANDARD 11
1063 CXX_STANDARD_REQUIRED YES
1064 CXX_EXTENSIONS YES)
1065 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc-test PRIVATE src test)
1066 TARGET_LINK_LIBRARIES(f32-conv-hwc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1067 ADD_TEST(f32-conv-hwc-test f32-conv-hwc-test)
1068
1069 ADD_EXECUTABLE(f32-conv-hwc2spchw-test test/f32-conv-hwc2spchw.cc)
1070 SET_TARGET_PROPERTIES(f32-conv-hwc2spchw-test PROPERTIES
1071 CXX_STANDARD 11
1072 CXX_STANDARD_REQUIRED YES
1073 CXX_EXTENSIONS YES)
1074 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc2spchw-test PRIVATE src test)
1075 TARGET_LINK_LIBRARIES(f32-conv-hwc2spchw-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1076 ADD_TEST(f32-conv-hwc2spchw-test f32-conv-hwc2spchw-test)
1077
1078 ADD_EXECUTABLE(f32-dwconv-spchw-test test/f32-dwconv-spchw.cc)
1079 SET_TARGET_PROPERTIES(f32-dwconv-spchw-test PROPERTIES
1080 CXX_STANDARD 11
1081 CXX_STANDARD_REQUIRED YES
1082 CXX_EXTENSIONS YES)
1083 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-spchw-test PRIVATE src test)
1084 TARGET_LINK_LIBRARIES(f32-dwconv-spchw-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1085 ADD_TEST(f32-dwconv-spchw-test f32-dwconv-spchw-test)
1086
1087 ADD_EXECUTABLE(f32-dwconv-test test/f32-dwconv.cc)
1088 SET_TARGET_PROPERTIES(f32-dwconv-test PROPERTIES
1089 CXX_STANDARD 11
1090 CXX_STANDARD_REQUIRED YES
1091 CXX_EXTENSIONS YES)
1092 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-test PRIVATE src test)
1093 TARGET_LINK_LIBRARIES(f32-dwconv-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1094 ADD_TEST(f32-dwconv-test f32-dwconv-test)
1095
1096 ADD_EXECUTABLE(f32-gavgpool-spchw-test test/f32-gavgpool-spchw.cc)
1097 SET_TARGET_PROPERTIES(f32-gavgpool-spchw-test PROPERTIES
1098 CXX_STANDARD 11
1099 CXX_STANDARD_REQUIRED YES
1100 CXX_EXTENSIONS YES)
1101 TARGET_INCLUDE_DIRECTORIES(f32-gavgpool-spchw-test PRIVATE src test)
1102 TARGET_LINK_LIBRARIES(f32-gavgpool-spchw-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1103 ADD_TEST(f32-gavgpool-spchw-test f32-gavgpool-spchw-test)
1104
1105 ADD_EXECUTABLE(f32-gavgpool-test test/f32-gavgpool.cc)
1106 SET_TARGET_PROPERTIES(f32-gavgpool-test PROPERTIES
1107 CXX_STANDARD 11
1108 CXX_STANDARD_REQUIRED YES
1109 CXX_EXTENSIONS YES)
1110 TARGET_INCLUDE_DIRECTORIES(f32-gavgpool-test PRIVATE src test)
1111 TARGET_LINK_LIBRARIES(f32-gavgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1112 ADD_TEST(f32-gavgpool-test f32-gavgpool-test)
1113
1114 ADD_EXECUTABLE(f32-gemm-test test/f32-gemm.cc)
1115 SET_TARGET_PROPERTIES(f32-gemm-test PROPERTIES
1116 CXX_STANDARD 11
1117 CXX_STANDARD_REQUIRED YES
1118 CXX_EXTENSIONS YES)
1119 TARGET_INCLUDE_DIRECTORIES(f32-gemm-test PRIVATE src test)
1120 TARGET_LINK_LIBRARIES(f32-gemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1121 ADD_TEST(f32-gemm-test f32-gemm-test)
1122
1123 ADD_EXECUTABLE(f32-gemminc-test test/f32-gemminc.cc)
1124 SET_TARGET_PROPERTIES(f32-gemminc-test PROPERTIES
1125 CXX_STANDARD 11
1126 CXX_STANDARD_REQUIRED YES
1127 CXX_EXTENSIONS YES)
1128 TARGET_INCLUDE_DIRECTORIES(f32-gemminc-test PRIVATE src test)
1129 TARGET_LINK_LIBRARIES(f32-gemminc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1130 ADD_TEST(f32-gemminc-test f32-gemminc-test)
1131
1132 ADD_EXECUTABLE(f32-hswish-test test/f32-hswish.cc)
1133 SET_TARGET_PROPERTIES(f32-hswish-test PROPERTIES
1134 CXX_STANDARD 11
1135 CXX_STANDARD_REQUIRED YES
1136 CXX_EXTENSIONS YES)
1137 TARGET_INCLUDE_DIRECTORIES(f32-hswish-test PRIVATE src test)
1138 TARGET_LINK_LIBRARIES(f32-hswish-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1139 ADD_TEST(f32-hswish-test f32-hswish-test)
1140
1141 ADD_EXECUTABLE(f32-igemm-test test/f32-igemm.cc)
1142 SET_TARGET_PROPERTIES(f32-igemm-test PROPERTIES
1143 CXX_STANDARD 11
1144 CXX_STANDARD_REQUIRED YES
1145 CXX_EXTENSIONS YES)
1146 TARGET_INCLUDE_DIRECTORIES(f32-igemm-test PRIVATE src test)
1147 TARGET_LINK_LIBRARIES(f32-igemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1148 ADD_TEST(f32-igemm-test f32-igemm-test)
1149
1150 ADD_EXECUTABLE(f32-maxpool-test test/f32-maxpool.cc)
1151 SET_TARGET_PROPERTIES(f32-maxpool-test PROPERTIES
1152 CXX_STANDARD 11
1153 CXX_STANDARD_REQUIRED YES
1154 CXX_EXTENSIONS YES)
1155 TARGET_INCLUDE_DIRECTORIES(f32-maxpool-test PRIVATE src test)
1156 TARGET_LINK_LIBRARIES(f32-maxpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1157 ADD_TEST(f32-maxpool-test f32-maxpool-test)
1158
1159 ADD_EXECUTABLE(f32-pavgpool-test test/f32-pavgpool.cc)
1160 SET_TARGET_PROPERTIES(f32-pavgpool-test PROPERTIES
1161 CXX_STANDARD 11
1162 CXX_STANDARD_REQUIRED YES
1163 CXX_EXTENSIONS YES)
1164 TARGET_INCLUDE_DIRECTORIES(f32-pavgpool-test PRIVATE src test)
1165 TARGET_LINK_LIBRARIES(f32-pavgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1166 ADD_TEST(f32-pavgpool-test f32-pavgpool-test)
1167
1168 ADD_EXECUTABLE(f32-ppmm-test test/f32-ppmm.cc)
1169 SET_TARGET_PROPERTIES(f32-ppmm-test PROPERTIES
1170 CXX_STANDARD 11
1171 CXX_STANDARD_REQUIRED YES
1172 CXX_EXTENSIONS YES)
1173 TARGET_INCLUDE_DIRECTORIES(f32-ppmm-test PRIVATE src test)
1174 TARGET_LINK_LIBRARIES(f32-ppmm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1175 ADD_TEST(f32-ppmm-test f32-ppmm-test)
1176
1177 ADD_EXECUTABLE(f32-prelu-test test/f32-prelu.cc)
1178 SET_TARGET_PROPERTIES(f32-prelu-test PROPERTIES
1179 CXX_STANDARD 11
1180 CXX_STANDARD_REQUIRED YES
1181 CXX_EXTENSIONS YES)
1182 TARGET_INCLUDE_DIRECTORIES(f32-prelu-test PRIVATE src test)
1183 TARGET_LINK_LIBRARIES(f32-prelu-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1184 ADD_TEST(f32-prelu-test f32-prelu-test)
1185
Marat Dukhan97579532019-10-18 16:40:39 -07001186 ADD_EXECUTABLE(f32-raddexpminusmax-test test/f32-raddexpminusmax.cc)
1187 SET_TARGET_PROPERTIES(f32-raddexpminusmax-test PROPERTIES
1188 CXX_STANDARD 11
1189 CXX_STANDARD_REQUIRED YES
1190 CXX_EXTENSIONS YES)
1191 TARGET_INCLUDE_DIRECTORIES(f32-raddexpminusmax-test PRIVATE src test)
1192 TARGET_LINK_LIBRARIES(f32-raddexpminusmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1193 ADD_TEST(f32-raddexpminusmax-test f32-raddexpminusmax-test)
1194
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001195 ADD_EXECUTABLE(f32-raddextexp-test test/f32-raddextexp.cc)
1196 SET_TARGET_PROPERTIES(f32-raddextexp-test PROPERTIES
1197 CXX_STANDARD 11
1198 CXX_STANDARD_REQUIRED YES
1199 CXX_EXTENSIONS YES)
1200 TARGET_INCLUDE_DIRECTORIES(f32-raddextexp-test PRIVATE src test)
Marat Dukhana5977df2019-10-30 22:02:15 -07001201 TARGET_LINK_LIBRARIES(f32-raddextexp-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001202 ADD_TEST(f32-raddextexp-test f32-raddextexp-test)
1203
Marat Dukhan97579532019-10-18 16:40:39 -07001204 ADD_EXECUTABLE(f32-raddstoreexpminusmax-test test/f32-raddstoreexpminusmax.cc)
1205 SET_TARGET_PROPERTIES(f32-raddstoreexpminusmax-test PROPERTIES
1206 CXX_STANDARD 11
1207 CXX_STANDARD_REQUIRED YES
1208 CXX_EXTENSIONS YES)
1209 TARGET_INCLUDE_DIRECTORIES(f32-raddstoreexpminusmax-test PRIVATE src test)
1210 TARGET_LINK_LIBRARIES(f32-raddstoreexpminusmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1211 ADD_TEST(f32-raddstoreexpminusmax-test f32-raddstoreexpminusmax-test)
1212
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001213 ADD_EXECUTABLE(f32-rmax-test test/f32-rmax.cc)
1214 SET_TARGET_PROPERTIES(f32-rmax-test PROPERTIES
1215 CXX_STANDARD 11
1216 CXX_STANDARD_REQUIRED YES
1217 CXX_EXTENSIONS YES)
1218 TARGET_INCLUDE_DIRECTORIES(f32-rmax-test PRIVATE src test)
1219 TARGET_LINK_LIBRARIES(f32-rmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1220 ADD_TEST(f32-rmax-test f32-rmax-test)
1221
1222 ADD_EXECUTABLE(f32-spmm-test test/f32-spmm.cc)
1223 SET_TARGET_PROPERTIES(f32-spmm-test PROPERTIES
1224 CXX_STANDARD 11
1225 CXX_STANDARD_REQUIRED YES
1226 CXX_EXTENSIONS YES)
1227 TARGET_INCLUDE_DIRECTORIES(f32-spmm-test PRIVATE src test)
1228 TARGET_LINK_LIBRARIES(f32-spmm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1229 ADD_TEST(f32-spmm-test f32-spmm-test)
1230
1231 ADD_EXECUTABLE(f32-vadd-test test/f32-vadd.cc)
1232 SET_TARGET_PROPERTIES(f32-vadd-test PROPERTIES
1233 CXX_STANDARD 11
1234 CXX_STANDARD_REQUIRED YES
1235 CXX_EXTENSIONS YES)
1236 TARGET_INCLUDE_DIRECTORIES(f32-vadd-test PRIVATE src test)
1237 TARGET_LINK_LIBRARIES(f32-vadd-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1238 ADD_TEST(f32-vadd-test f32-vadd-test)
1239
Marat Dukhanc76065f2019-11-14 18:13:33 -08001240 ADD_EXECUTABLE(f32-vaddc-test test/f32-vaddc.cc)
1241 SET_TARGET_PROPERTIES(f32-vaddc-test PROPERTIES
1242 CXX_STANDARD 11
1243 CXX_STANDARD_REQUIRED YES
1244 CXX_EXTENSIONS YES)
1245 TARGET_INCLUDE_DIRECTORIES(f32-vaddc-test PRIVATE src test)
1246 TARGET_LINK_LIBRARIES(f32-vaddc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1247 ADD_TEST(f32-vaddc-test f32-vaddc-test)
1248
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001249 ADD_EXECUTABLE(f32-vmul-test test/f32-vmul.cc)
1250 SET_TARGET_PROPERTIES(f32-vmul-test PROPERTIES
1251 CXX_STANDARD 11
1252 CXX_STANDARD_REQUIRED YES
1253 CXX_EXTENSIONS YES)
1254 TARGET_INCLUDE_DIRECTORIES(f32-vmul-test PRIVATE src test)
1255 TARGET_LINK_LIBRARIES(f32-vmul-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1256 ADD_TEST(f32-vmul-test f32-vmul-test)
1257
Marat Dukhanc76065f2019-11-14 18:13:33 -08001258 ADD_EXECUTABLE(f32-vmulc-test test/f32-vmulc.cc)
1259 SET_TARGET_PROPERTIES(f32-vmulc-test PROPERTIES
1260 CXX_STANDARD 11
1261 CXX_STANDARD_REQUIRED YES
1262 CXX_EXTENSIONS YES)
1263 TARGET_INCLUDE_DIRECTORIES(f32-vmulc-test PRIVATE src test)
1264 TARGET_LINK_LIBRARIES(f32-vmulc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1265 ADD_TEST(f32-vmulc-test f32-vmulc-test)
1266
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001267 ADD_EXECUTABLE(f32-vmulcaddc-test test/f32-vmulcaddc.cc)
1268 SET_TARGET_PROPERTIES(f32-vmulcaddc-test PROPERTIES
1269 CXX_STANDARD 11
1270 CXX_STANDARD_REQUIRED YES
1271 CXX_EXTENSIONS YES)
1272 TARGET_INCLUDE_DIRECTORIES(f32-vmulcaddc-test PRIVATE src test)
1273 TARGET_LINK_LIBRARIES(f32-vmulcaddc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1274 ADD_TEST(f32-vmulcaddc-test f32-vmulcaddc-test)
1275
Marat Dukhan97579532019-10-18 16:40:39 -07001276 ADD_EXECUTABLE(f32-vscaleexpminusmax-test test/f32-vscaleexpminusmax.cc)
1277 SET_TARGET_PROPERTIES(f32-vscaleexpminusmax-test PROPERTIES
1278 CXX_STANDARD 11
1279 CXX_STANDARD_REQUIRED YES
1280 CXX_EXTENSIONS YES)
1281 TARGET_INCLUDE_DIRECTORIES(f32-vscaleexpminusmax-test PRIVATE src test)
1282 TARGET_LINK_LIBRARIES(f32-vscaleexpminusmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1283 ADD_TEST(f32-vscaleexpminusmax-test f32-vscaleexpminusmax-test)
1284
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001285 ADD_EXECUTABLE(f32-vscaleextexp-test test/f32-vscaleextexp.cc)
1286 SET_TARGET_PROPERTIES(f32-vscaleexpminusmax-test PROPERTIES
1287 CXX_STANDARD 11
1288 CXX_STANDARD_REQUIRED YES
1289 CXX_EXTENSIONS YES)
1290 TARGET_INCLUDE_DIRECTORIES(f32-vscaleextexp-test PRIVATE src test)
Marat Dukhana5977df2019-10-30 22:02:15 -07001291 TARGET_LINK_LIBRARIES(f32-vscaleextexp-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001292 ADD_TEST(f32-vscaleextexp-test f32-vscaleextexp-test)
1293
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001294 ADD_EXECUTABLE(f32-vsub-test test/f32-vsub.cc)
1295 SET_TARGET_PROPERTIES(f32-vsub-test PROPERTIES
1296 CXX_STANDARD 11
1297 CXX_STANDARD_REQUIRED YES
1298 CXX_EXTENSIONS YES)
1299 TARGET_INCLUDE_DIRECTORIES(f32-vsub-test PRIVATE src test)
1300 TARGET_LINK_LIBRARIES(f32-vsub-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1301 ADD_TEST(f32-vsub-test f32-vsub-test)
1302
Marat Dukhanc76065f2019-11-14 18:13:33 -08001303 ADD_EXECUTABLE(f32-vsubc-test test/f32-vsubc.cc)
1304 SET_TARGET_PROPERTIES(f32-vsubc-test PROPERTIES
1305 CXX_STANDARD 11
1306 CXX_STANDARD_REQUIRED YES
1307 CXX_EXTENSIONS YES)
1308 TARGET_INCLUDE_DIRECTORIES(f32-vsubc-test PRIVATE src test)
1309 TARGET_LINK_LIBRARIES(f32-vsubc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1310 ADD_TEST(f32-vsubc-test f32-vsubc-test)
1311
1312 ADD_EXECUTABLE(f32-vrsubc-test test/f32-vrsubc.cc)
1313 SET_TARGET_PROPERTIES(f32-vrsubc-test PROPERTIES
1314 CXX_STANDARD 11
1315 CXX_STANDARD_REQUIRED YES
1316 CXX_EXTENSIONS YES)
1317 TARGET_INCLUDE_DIRECTORIES(f32-vrsubc-test PRIVATE src test)
1318 TARGET_LINK_LIBRARIES(f32-vrsubc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1319 ADD_TEST(f32-vrsubc-test f32-vrsubc-test)
1320
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001321 ADD_EXECUTABLE(q8-avgpool-test test/q8-avgpool.cc)
1322 SET_TARGET_PROPERTIES(q8-avgpool-test PROPERTIES
1323 CXX_STANDARD 11
1324 CXX_STANDARD_REQUIRED YES
1325 CXX_EXTENSIONS YES)
1326 TARGET_INCLUDE_DIRECTORIES(q8-avgpool-test PRIVATE src test)
1327 TARGET_LINK_LIBRARIES(q8-avgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1328 ADD_TEST(q8-avgpool-test q8-avgpool-test)
1329
1330 ADD_EXECUTABLE(q8-dwconv-test test/q8-dwconv.cc)
1331 SET_TARGET_PROPERTIES(q8-dwconv-test PROPERTIES
1332 CXX_STANDARD 11
1333 CXX_STANDARD_REQUIRED YES
1334 CXX_EXTENSIONS YES)
1335 TARGET_INCLUDE_DIRECTORIES(q8-dwconv-test PRIVATE src test)
1336 TARGET_LINK_LIBRARIES(q8-dwconv-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1337 ADD_TEST(q8-dwconv-test q8-dwconv-test)
1338
1339 ADD_EXECUTABLE(q8-gavgpool-test test/q8-gavgpool.cc)
1340 SET_TARGET_PROPERTIES(q8-gavgpool-test PROPERTIES
1341 CXX_STANDARD 11
1342 CXX_STANDARD_REQUIRED YES
1343 CXX_EXTENSIONS YES)
1344 TARGET_INCLUDE_DIRECTORIES(q8-gavgpool-test PRIVATE src test)
1345 TARGET_LINK_LIBRARIES(q8-gavgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1346 ADD_TEST(q8-gavgpool-test q8-gavgpool-test)
1347
1348 ADD_EXECUTABLE(q8-gemm-test test/q8-gemm.cc)
1349 SET_TARGET_PROPERTIES(q8-gemm-test PROPERTIES
1350 CXX_STANDARD 11
1351 CXX_STANDARD_REQUIRED YES
1352 CXX_EXTENSIONS YES)
1353 TARGET_INCLUDE_DIRECTORIES(q8-gemm-test PRIVATE src test)
1354 TARGET_LINK_LIBRARIES(q8-gemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1355 ADD_TEST(q8-gemm-test q8-gemm-test)
1356
1357 ADD_EXECUTABLE(q8-igemm-test test/q8-igemm.cc)
1358 SET_TARGET_PROPERTIES(q8-igemm-test PROPERTIES
1359 CXX_STANDARD 11
1360 CXX_STANDARD_REQUIRED YES
1361 CXX_EXTENSIONS YES)
1362 TARGET_INCLUDE_DIRECTORIES(q8-igemm-test PRIVATE src test)
1363 TARGET_LINK_LIBRARIES(q8-igemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1364 ADD_TEST(q8-igemm-test q8-igemm-test)
1365
1366 ADD_EXECUTABLE(q8-vadd-test test/q8-vadd.cc)
1367 SET_TARGET_PROPERTIES(q8-vadd-test PROPERTIES
1368 CXX_STANDARD 11
1369 CXX_STANDARD_REQUIRED YES
1370 CXX_EXTENSIONS YES)
1371 TARGET_INCLUDE_DIRECTORIES(q8-vadd-test PRIVATE src test)
1372 TARGET_LINK_LIBRARIES(q8-vadd-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1373 ADD_TEST(q8-vadd-test q8-vadd-test)
1374
1375 ADD_EXECUTABLE(u8-clamp-test test/u8-clamp.cc)
1376 SET_TARGET_PROPERTIES(u8-clamp-test PROPERTIES
1377 CXX_STANDARD 11
1378 CXX_STANDARD_REQUIRED YES
1379 CXX_EXTENSIONS YES)
1380 TARGET_INCLUDE_DIRECTORIES(u8-clamp-test PRIVATE src test)
1381 TARGET_LINK_LIBRARIES(u8-clamp-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1382 ADD_TEST(u8-clamp-test u8-clamp-test)
1383
1384 ADD_EXECUTABLE(u8-lut32norm-test test/u8-lut32norm.cc)
1385 SET_TARGET_PROPERTIES(u8-lut32norm-test PROPERTIES
1386 CXX_STANDARD 11
1387 CXX_STANDARD_REQUIRED YES
1388 CXX_EXTENSIONS YES)
1389 TARGET_INCLUDE_DIRECTORIES(u8-lut32norm-test PRIVATE src test)
1390 TARGET_LINK_LIBRARIES(u8-lut32norm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1391 ADD_TEST(u8-lut32norm-test u8-lut32norm-test)
1392
1393 ADD_EXECUTABLE(u8-maxpool-test test/u8-maxpool.cc)
1394 SET_TARGET_PROPERTIES(u8-maxpool-test PROPERTIES
1395 CXX_STANDARD 11
1396 CXX_STANDARD_REQUIRED YES
1397 CXX_EXTENSIONS YES)
1398 TARGET_INCLUDE_DIRECTORIES(u8-maxpool-test PRIVATE src test)
1399 TARGET_LINK_LIBRARIES(u8-maxpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1400 ADD_TEST(u8-maxpool-test u8-maxpool-test)
1401
1402 ADD_EXECUTABLE(u8-rmax-test test/u8-rmax.cc)
1403 SET_TARGET_PROPERTIES(u8-rmax-test PROPERTIES
1404 CXX_STANDARD 11
1405 CXX_STANDARD_REQUIRED YES
1406 CXX_EXTENSIONS YES)
1407 TARGET_INCLUDE_DIRECTORIES(u8-rmax-test PRIVATE src test)
1408 TARGET_LINK_LIBRARIES(u8-rmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1409 ADD_TEST(u8-rmax-test u8-rmax-test)
1410
1411 ADD_EXECUTABLE(x32-packx-test test/x32-packx.cc)
1412 SET_TARGET_PROPERTIES(x32-packx-test PROPERTIES
1413 CXX_STANDARD 11
1414 CXX_STANDARD_REQUIRED YES
1415 CXX_EXTENSIONS YES)
1416 TARGET_INCLUDE_DIRECTORIES(x32-packx-test PRIVATE src test)
1417 TARGET_LINK_LIBRARIES(x32-packx-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1418 ADD_TEST(x32-packx-test x32-packx-test)
1419
1420 ADD_EXECUTABLE(x32-pad-test test/x32-pad.cc)
1421 SET_TARGET_PROPERTIES(x32-pad-test PROPERTIES
1422 CXX_STANDARD 11
1423 CXX_STANDARD_REQUIRED YES
1424 CXX_EXTENSIONS YES)
1425 TARGET_INCLUDE_DIRECTORIES(x32-pad-test PRIVATE src test)
1426 TARGET_LINK_LIBRARIES(x32-pad-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1427 ADD_TEST(x32-pad-test x32-pad-test)
1428
1429 ADD_EXECUTABLE(x32-unpool-test test/x32-unpool.cc)
1430 SET_TARGET_PROPERTIES(x32-unpool-test PROPERTIES
1431 CXX_STANDARD 11
1432 CXX_STANDARD_REQUIRED YES
1433 CXX_EXTENSIONS YES)
1434 TARGET_INCLUDE_DIRECTORIES(x32-unpool-test PRIVATE src test)
1435 TARGET_LINK_LIBRARIES(x32-unpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1436 ADD_TEST(x32-unpool-test x32-unpool-test)
1437
1438 ADD_EXECUTABLE(x32-zip-test test/x32-zip.cc)
1439 SET_TARGET_PROPERTIES(x32-zip-test PROPERTIES
1440 CXX_STANDARD 11
1441 CXX_STANDARD_REQUIRED YES
1442 CXX_EXTENSIONS YES)
1443 TARGET_INCLUDE_DIRECTORIES(x32-zip-test PRIVATE src test)
1444 TARGET_LINK_LIBRARIES(x32-zip-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1445 ADD_TEST(x32-zip-test x32-zip-test)
1446
1447 ADD_EXECUTABLE(x8-lut-test test/x8-lut.cc)
1448 SET_TARGET_PROPERTIES(x8-lut-test PROPERTIES
1449 CXX_STANDARD 11
1450 CXX_STANDARD_REQUIRED YES
1451 CXX_EXTENSIONS YES)
1452 TARGET_INCLUDE_DIRECTORIES(x8-lut-test PRIVATE src test)
1453 TARGET_LINK_LIBRARIES(x8-lut-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1454 ADD_TEST(x8-lut-test x8-lut-test)
1455
1456 ADD_EXECUTABLE(x8-zip-test test/x8-zip.cc)
1457 SET_TARGET_PROPERTIES(x8-zip-test PROPERTIES
1458 CXX_STANDARD 11
1459 CXX_STANDARD_REQUIRED YES
1460 CXX_EXTENSIONS YES)
1461 TARGET_INCLUDE_DIRECTORIES(x8-zip-test PRIVATE src test)
1462 TARGET_LINK_LIBRARIES(x8-zip-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1463 ADD_TEST(x8-zip-test x8-zip-test)
1464ENDIF()
1465
1466# ---[ XNNPACK microbenchmarks
1467IF(XNNPACK_BUILD_BENCHMARKS)
1468 # ---[ Build google benchmark
1469 IF(NOT TARGET benchmark)
1470 SET(BENCHMARK_ENABLE_TESTING OFF CACHE BOOL "")
1471 ADD_SUBDIRECTORY(
1472 "${GOOGLEBENCHMARK_SOURCE_DIR}"
1473 "${CONFU_DEPENDENCIES_BINARY_DIR}/googlebenchmark")
1474 ENDIF()
1475
1476 ADD_LIBRARY(bench-utils bench/utils.cc)
1477 SET_TARGET_PROPERTIES(bench-utils PROPERTIES
1478 CXX_STANDARD 11
1479 CXX_STANDARD_REQUIRED YES
1480 CXX_EXTENSIONS NO)
1481 TARGET_INCLUDE_DIRECTORIES(bench-utils PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
Marat Dukhanbad48fe2019-11-04 10:35:22 -08001482 TARGET_LINK_LIBRARIES(bench-utils PRIVATE benchmark cpuinfo)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001483
Marat Dukhan5f18d262019-10-31 10:24:14 -07001484 # ---[ Build end-to-end microbenchmarks
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001485 ADD_EXECUTABLE(end2end-bench bench/end2end.cc models/mobilenet-v1.cc models/mobilenet-v2.cc)
1486 SET_TARGET_PROPERTIES(end2end-bench PROPERTIES
1487 CXX_STANDARD 11
1488 CXX_STANDARD_REQUIRED YES
1489 CXX_EXTENSIONS NO)
1490 TARGET_INCLUDE_DIRECTORIES(end2end-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
Frank Barcharde72e2872019-10-31 11:12:15 -07001491 TARGET_LINK_LIBRARIES(end2end-bench PRIVATE XNNPACK benchmark bench-utils)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001492
Marat Dukhanef4416e2019-10-31 13:44:40 -07001493 ADD_EXECUTABLE(f32-dwconv-e2e-bench bench/f32-dwconv-e2e.cc models/mobilenet-v1.cc models/mobilenet-v2.cc)
1494 SET_TARGET_PROPERTIES(f32-dwconv-e2e-bench PROPERTIES
1495 CXX_STANDARD 11
1496 CXX_STANDARD_REQUIRED YES
1497 CXX_EXTENSIONS NO)
Marat Dukhanc9d2f3f2019-10-31 14:02:07 -07001498 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-e2e-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}" src)
Marat Dukhanef4416e2019-10-31 13:44:40 -07001499 TARGET_LINK_LIBRARIES(f32-dwconv-e2e-bench PRIVATE XNNPACK benchmark bench-utils)
1500
Marat Dukhan0a5a53f2019-10-31 10:58:31 -07001501 ADD_EXECUTABLE(f32-gemm-e2e-bench bench/f32-gemm-e2e.cc models/mobilenet-v1.cc models/mobilenet-v2.cc)
1502 SET_TARGET_PROPERTIES(f32-gemm-e2e-bench PROPERTIES
Marat Dukhan5f18d262019-10-31 10:24:14 -07001503 CXX_STANDARD 11
1504 CXX_STANDARD_REQUIRED YES
1505 CXX_EXTENSIONS NO)
Marat Dukhanc9d2f3f2019-10-31 14:02:07 -07001506 TARGET_INCLUDE_DIRECTORIES(f32-gemm-e2e-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}" src)
Frank Barchardc712fa42019-10-31 14:00:21 -07001507 TARGET_LINK_LIBRARIES(f32-gemm-e2e-bench PRIVATE XNNPACK benchmark bench-utils)
Marat Dukhan5f18d262019-10-31 10:24:14 -07001508
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001509 # ---[ Build operator-level microbenchmarks
1510 ADD_EXECUTABLE(add-bench bench/add.cc)
1511 SET_TARGET_PROPERTIES(add-bench PROPERTIES
1512 CXX_STANDARD 11
1513 CXX_STANDARD_REQUIRED YES
1514 CXX_EXTENSIONS NO)
1515 TARGET_INCLUDE_DIRECTORIES(add-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1516 TARGET_LINK_LIBRARIES(add-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1517
1518 ADD_EXECUTABLE(average-pooling-bench bench/average-pooling.cc)
1519 SET_TARGET_PROPERTIES(average-pooling-bench PROPERTIES
1520 CXX_STANDARD 11
1521 CXX_STANDARD_REQUIRED YES
1522 CXX_EXTENSIONS NO)
1523 TARGET_INCLUDE_DIRECTORIES(average-pooling-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1524 TARGET_LINK_LIBRARIES(average-pooling-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1525
1526 ADD_EXECUTABLE(channel-shuffle-bench bench/channel-shuffle.cc)
1527 SET_TARGET_PROPERTIES(channel-shuffle-bench PROPERTIES
1528 CXX_STANDARD 11
1529 CXX_STANDARD_REQUIRED YES
1530 CXX_EXTENSIONS NO)
1531 TARGET_INCLUDE_DIRECTORIES(channel-shuffle-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1532 TARGET_LINK_LIBRARIES(channel-shuffle-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1533
1534 ADD_EXECUTABLE(convolution-bench bench/convolution.cc)
1535 SET_TARGET_PROPERTIES(convolution-bench PROPERTIES
1536 CXX_STANDARD 11
1537 CXX_STANDARD_REQUIRED YES
1538 CXX_EXTENSIONS NO)
1539 TARGET_INCLUDE_DIRECTORIES(convolution-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1540 TARGET_LINK_LIBRARIES(convolution-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1541
1542 ADD_EXECUTABLE(deconvolution-bench bench/deconvolution.cc)
1543 SET_TARGET_PROPERTIES(deconvolution-bench PROPERTIES
1544 CXX_STANDARD 11
1545 CXX_STANDARD_REQUIRED YES
1546 CXX_EXTENSIONS NO)
1547 TARGET_INCLUDE_DIRECTORIES(deconvolution-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1548 TARGET_LINK_LIBRARIES(deconvolution-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1549
1550 ADD_EXECUTABLE(global-average-pooling-bench bench/global-average-pooling.cc)
1551 SET_TARGET_PROPERTIES(global-average-pooling-bench PROPERTIES
1552 CXX_STANDARD 11
1553 CXX_STANDARD_REQUIRED YES
1554 CXX_EXTENSIONS NO)
1555 TARGET_INCLUDE_DIRECTORIES(global-average-pooling-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1556 TARGET_LINK_LIBRARIES(global-average-pooling-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1557
1558 ADD_EXECUTABLE(max-pooling-bench bench/max-pooling.cc)
1559 SET_TARGET_PROPERTIES(max-pooling-bench PROPERTIES
1560 CXX_STANDARD 11
1561 CXX_STANDARD_REQUIRED YES
1562 CXX_EXTENSIONS NO)
1563 TARGET_INCLUDE_DIRECTORIES(max-pooling-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1564 TARGET_LINK_LIBRARIES(max-pooling-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1565
Marat Dukhan95b22432019-10-30 16:30:14 -07001566 ADD_EXECUTABLE(prelu-bench bench/prelu.cc)
1567 SET_TARGET_PROPERTIES(prelu-bench PROPERTIES
1568 CXX_STANDARD 11
1569 CXX_STANDARD_REQUIRED YES
1570 CXX_EXTENSIONS NO)
1571 TARGET_INCLUDE_DIRECTORIES(prelu-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1572 TARGET_LINK_LIBRARIES(prelu-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1573
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001574 ADD_EXECUTABLE(sigmoid-bench bench/sigmoid.cc)
1575 SET_TARGET_PROPERTIES(sigmoid-bench PROPERTIES
1576 CXX_STANDARD 11
1577 CXX_STANDARD_REQUIRED YES
1578 CXX_EXTENSIONS NO)
1579 TARGET_INCLUDE_DIRECTORIES(sigmoid-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1580 TARGET_LINK_LIBRARIES(sigmoid-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1581
1582 ADD_EXECUTABLE(softargmax-bench bench/softargmax.cc)
1583 SET_TARGET_PROPERTIES(softargmax-bench PROPERTIES
1584 CXX_STANDARD 11
1585 CXX_STANDARD_REQUIRED YES
1586 CXX_EXTENSIONS NO)
1587 TARGET_INCLUDE_DIRECTORIES(softargmax-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1588 TARGET_LINK_LIBRARIES(softargmax-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1589
1590 # ---[ Build microkernel-level microbenchmarks
1591 ADD_EXECUTABLE(f16-gemm-bench bench/f16-gemm.cc)
1592 SET_TARGET_PROPERTIES(f16-gemm-bench PROPERTIES
1593 CXX_STANDARD 11
1594 CXX_STANDARD_REQUIRED YES
1595 CXX_EXTENSIONS YES)
1596 TARGET_INCLUDE_DIRECTORIES(f16-gemm-bench PRIVATE src)
1597 TARGET_INCLUDE_DIRECTORIES(f16-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1598 TARGET_LINK_LIBRARIES(f16-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1599
1600 ADD_EXECUTABLE(f32-conv-hwc-bench bench/f32-conv-hwc.cc)
1601 SET_TARGET_PROPERTIES(f32-conv-hwc-bench PROPERTIES
1602 CXX_STANDARD 11
1603 CXX_STANDARD_REQUIRED YES
1604 CXX_EXTENSIONS YES)
1605 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc-bench PRIVATE src)
1606 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1607 TARGET_LINK_LIBRARIES(f32-conv-hwc-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1608
1609 ADD_EXECUTABLE(f32-dwconv-spchw-bench bench/f32-dwconv-spchw.cc)
1610 SET_TARGET_PROPERTIES(f32-dwconv-spchw-bench PROPERTIES
1611 CXX_STANDARD 11
1612 CXX_STANDARD_REQUIRED YES
1613 CXX_EXTENSIONS YES)
1614 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-spchw-bench PRIVATE src)
1615 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-spchw-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1616 TARGET_LINK_LIBRARIES(f32-dwconv-spchw-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1617
1618 ADD_EXECUTABLE(f32-dwconv-bench bench/f32-dwconv.cc)
1619 SET_TARGET_PROPERTIES(f32-dwconv-bench PROPERTIES
1620 CXX_STANDARD 11
1621 CXX_STANDARD_REQUIRED YES
1622 CXX_EXTENSIONS YES)
1623 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-bench PRIVATE src)
1624 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1625 TARGET_LINK_LIBRARIES(f32-dwconv-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1626
1627 ADD_EXECUTABLE(f32-gemm-bench bench/f32-gemm.cc)
1628 SET_TARGET_PROPERTIES(f32-gemm-bench PROPERTIES
1629 CXX_STANDARD 11
1630 CXX_STANDARD_REQUIRED YES
1631 CXX_EXTENSIONS YES)
1632 TARGET_INCLUDE_DIRECTORIES(f32-gemm-bench PRIVATE src)
1633 TARGET_INCLUDE_DIRECTORIES(f32-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1634 TARGET_LINK_LIBRARIES(f32-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1635
1636 ADD_EXECUTABLE(f32-igemm-bench bench/f32-igemm.cc)
1637 SET_TARGET_PROPERTIES(f32-igemm-bench PROPERTIES
1638 CXX_STANDARD 11
1639 CXX_STANDARD_REQUIRED YES
1640 CXX_EXTENSIONS YES)
1641 TARGET_INCLUDE_DIRECTORIES(f32-igemm-bench PRIVATE src)
1642 TARGET_INCLUDE_DIRECTORIES(f32-igemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1643 TARGET_LINK_LIBRARIES(f32-igemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1644
1645 ADD_EXECUTABLE(f32-im2col-gemm-bench bench/f32-im2col-gemm.cc src/im2col.c)
1646 SET_TARGET_PROPERTIES(f32-im2col-gemm-bench PROPERTIES
1647 C_STANDARD 99
1648 C_STANDARD_REQUIRED YES
1649 C_EXTENSIONS NO
1650 CXX_STANDARD 11
1651 CXX_STANDARD_REQUIRED YES
1652 CXX_EXTENSIONS YES)
1653 TARGET_INCLUDE_DIRECTORIES(f32-im2col-gemm-bench PRIVATE src)
1654 TARGET_INCLUDE_DIRECTORIES(f32-im2col-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1655 TARGET_LINK_LIBRARIES(f32-im2col-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1656
1657 ADD_EXECUTABLE(f32-rmax-bench bench/f32-rmax.cc)
1658 SET_TARGET_PROPERTIES(f32-rmax-bench PROPERTIES
1659 CXX_STANDARD 11
1660 CXX_STANDARD_REQUIRED YES
1661 CXX_EXTENSIONS YES)
1662 TARGET_INCLUDE_DIRECTORIES(f32-rmax-bench PRIVATE src)
1663 TARGET_INCLUDE_DIRECTORIES(f32-rmax-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1664 TARGET_LINK_LIBRARIES(f32-rmax-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1665
1666 ADD_EXECUTABLE(f32-spmm-bench bench/f32-spmm.cc)
1667 SET_TARGET_PROPERTIES(f32-spmm-bench PROPERTIES
1668 CXX_STANDARD 11
1669 CXX_STANDARD_REQUIRED YES
1670 CXX_EXTENSIONS YES)
1671 TARGET_INCLUDE_DIRECTORIES(f32-spmm-bench PRIVATE src)
1672 TARGET_INCLUDE_DIRECTORIES(f32-spmm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1673 TARGET_LINK_LIBRARIES(f32-spmm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1674
Ashkan Aliabadi7892d972019-10-24 16:00:52 -07001675 ADD_EXECUTABLE(f32-softargmax-bench bench/f32-softargmax.cc)
1676 SET_TARGET_PROPERTIES(f32-softargmax-bench PROPERTIES
Marat Dukhan4a4a7fa2019-10-21 13:46:14 -07001677 CXX_STANDARD 11
1678 CXX_STANDARD_REQUIRED YES
1679 CXX_EXTENSIONS YES)
Ashkan Aliabadi7892d972019-10-24 16:00:52 -07001680 TARGET_INCLUDE_DIRECTORIES(f32-softargmax-bench PRIVATE src)
1681 TARGET_INCLUDE_DIRECTORIES(f32-softargmax-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1682 TARGET_LINK_LIBRARIES(f32-softargmax-bench PRIVATE XNNPACK fp16 benchmark bench-utils)
Marat Dukhan4a4a7fa2019-10-21 13:46:14 -07001683
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001684 ADD_EXECUTABLE(q8-gemm-bench bench/q8-gemm.cc)
1685 SET_TARGET_PROPERTIES(q8-gemm-bench PROPERTIES
1686 CXX_STANDARD 11
1687 CXX_STANDARD_REQUIRED YES
1688 CXX_EXTENSIONS YES)
1689 TARGET_INCLUDE_DIRECTORIES(q8-gemm-bench PRIVATE src)
1690 TARGET_INCLUDE_DIRECTORIES(q8-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1691 TARGET_LINK_LIBRARIES(q8-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1692ENDIF()