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