blob: 4f4ab580fcb5a8663a682467e6f3400a067762a4 [file] [log] [blame]
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001# Copyright (c) Facebook, Inc. and its affiliates.
2# All rights reserved.
3#
4# Copyright 2019 Google LLC
5#
6# This source code is licensed under the BSD-style license found in the
7# LICENSE file in the root directory of this source tree.
8
9CMAKE_MINIMUM_REQUIRED(VERSION 3.5 FATAL_ERROR)
10
11INCLUDE(GNUInstallDirs)
12
13# ---[ Project and semantic versioning.
14PROJECT(XNNPACK C CXX ASM)
15
16# ---[ Options.
17SET(XNNPACK_LIBRARY_TYPE "default" CACHE STRING "Type of library (shared, static, or default) to build")
18SET_PROPERTY(CACHE XNNPACK_LIBRARY_TYPE PROPERTY STRINGS default static shared)
19OPTION(XNNPACK_BUILD_TESTS "Build XNNPACK unit tests" ON)
20OPTION(XNNPACK_BUILD_BENCHMARKS "Build XNNPACK benchmarks" ON)
21
22# ---[ CMake options
23IF(XNNPACK_BUILD_TESTS)
24 ENABLE_TESTING()
25ENDIF()
26
27# ---[ Build flags
28IF(NOT CMAKE_SYSTEM_PROCESSOR)
29 IF(IOS)
30 LIST(LENGTH IOS_ARCH IOS_ARCH_COUNT)
31 IF(IOS_ARCH_COUNT GREATER 1)
32 MESSAGE(FATAL_ERROR "Unsupported XNNPACK build with multiple iOS architectures (${IOS_ARCH}). "
33 "Specify a single architecture in IOS_ARCH and re-configure. ")
34 ENDIF()
35 IF(NOT IOS_ARCH MATCHES "^(i386|x86_64|armv7.*|arm64.*)$")
36 MESSAGE(FATAL_ERROR "Unrecognized IOS_ARCH = ${IOS_ARCH}")
37 ENDIF()
38 ELSE()
39 MESSAGE(FATAL_ERROR "CMAKE_SYSTEM_PROCESSOR is not defined")
40 ENDIF()
41ELSEIF(NOT CMAKE_SYSTEM_PROCESSOR MATCHES "^(i[3-6]86|x86_64|armv[5-8].*|aarch64)$")
42 MESSAGE(FATAL_ERROR "Unrecognized CMAKE_SYSTEM_PROCESSOR = ${CMAKE_SYSTEM_PROCESSOR}")
43ENDIF()
44
45IF(NOT CMAKE_SYSTEM_NAME)
46 MESSAGE(FATAL_ERROR "CMAKE_SYSTEM_NAME not defined")
47ELSEIF(NOT CMAKE_SYSTEM_NAME MATCHES "^(Darwin|Linux|Android)$")
48 MESSAGE(FATAL_ERROR "Unrecognized CMAKE_SYSTEM_NAME = ${CMAKE_SYSTEM_NAME}")
49ENDIF()
50
51# ---[ Download deps
52IF(NOT DEFINED CLOG_SOURCE_DIR)
53 MESSAGE(STATUS "Downloading clog to ${CMAKE_BINARY_DIR}/clog-source (define CLOG_SOURCE_DIR to avoid it)")
54 CONFIGURE_FILE(cmake/DownloadCLog.cmake "${CMAKE_BINARY_DIR}/clog-download/CMakeLists.txt")
55 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" -G "${CMAKE_GENERATOR}" .
56 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/clog-download")
57 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" --build .
58 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/clog-download")
59 SET(CLOG_SOURCE_DIR "${CMAKE_BINARY_DIR}/clog-source" CACHE STRING "clog source directory")
60ENDIF()
61
62IF(NOT DEFINED CPUINFO_SOURCE_DIR)
63 MESSAGE(STATUS "Downloading cpuinfo to ${CMAKE_BINARY_DIR}/cpuinfo-source (define CPUINFO_SOURCE_DIR to avoid it)")
64 CONFIGURE_FILE(cmake/DownloadCpuinfo.cmake "${CMAKE_BINARY_DIR}/cpuinfo-download/CMakeLists.txt")
65 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" -G "${CMAKE_GENERATOR}" .
66 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/cpuinfo-download")
67 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" --build .
68 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/cpuinfo-download")
69 SET(CPUINFO_SOURCE_DIR "${CMAKE_BINARY_DIR}/cpuinfo-source" CACHE STRING "cpuinfo source directory")
70ENDIF()
71
72IF(NOT DEFINED FP16_SOURCE_DIR)
73 MESSAGE(STATUS "Downloading FP16 to ${CMAKE_BINARY_DIR}/FP16-source (define FP16_SOURCE_DIR to avoid it)")
74 CONFIGURE_FILE(cmake/DownloadFP16.cmake "${CMAKE_BINARY_DIR}/FP16-download/CMakeLists.txt")
75 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" -G "${CMAKE_GENERATOR}" .
76 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/FP16-download")
77 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" --build .
78 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/FP16-download")
79 SET(FP16_SOURCE_DIR "${CMAKE_BINARY_DIR}/FP16-source" CACHE STRING "FP16 source directory")
80ENDIF()
81
82IF(NOT DEFINED FXDIV_SOURCE_DIR)
83 MESSAGE(STATUS "Downloading FXdiv to ${CMAKE_BINARY_DIR}/FXdiv-source (define FXDIV_SOURCE_DIR to avoid it)")
84 CONFIGURE_FILE(cmake/DownloadFXdiv.cmake "${CMAKE_BINARY_DIR}/FXdiv-download/CMakeLists.txt")
85 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" -G "${CMAKE_GENERATOR}" .
86 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/FXdiv-download")
87 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" --build .
88 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/FXdiv-download")
89 SET(FXDIV_SOURCE_DIR "${CMAKE_BINARY_DIR}/FXdiv-source" CACHE STRING "FXdiv source directory")
90ENDIF()
91
92IF(NOT DEFINED PSIMD_SOURCE_DIR)
93 MESSAGE(STATUS "Downloading PSimd to ${CMAKE_BINARY_DIR}/psimd-source (define PSIMD_SOURCE_DIR to avoid it)")
94 CONFIGURE_FILE(cmake/DownloadPSimd.cmake "${CMAKE_BINARY_DIR}/psimd-download/CMakeLists.txt")
95 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" -G "${CMAKE_GENERATOR}" .
96 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/psimd-download")
97 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" --build .
98 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/psimd-download")
99 SET(PSIMD_SOURCE_DIR "${CMAKE_BINARY_DIR}/psimd-source" CACHE STRING "PSimd source directory")
100ENDIF()
101
102IF(NOT DEFINED PTHREADPOOL_SOURCE_DIR)
103 MESSAGE(STATUS "Downloading pthreadpool to ${CMAKE_BINARY_DIR}/pthreadpool-source (define PTHREADPOOL_SOURCE_DIR to avoid it)")
104 CONFIGURE_FILE(cmake/DownloadPThreadPool.cmake "${CMAKE_BINARY_DIR}/pthreadpool-download/CMakeLists.txt")
105 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" -G "${CMAKE_GENERATOR}" .
106 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/pthreadpool-download")
107 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" --build .
108 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/pthreadpool-download")
109 SET(PTHREADPOOL_SOURCE_DIR "${CMAKE_BINARY_DIR}/pthreadpool-source" CACHE STRING "pthreadpool source directory")
110ENDIF()
111
112IF(XNNPACK_BUILD_TESTS AND NOT DEFINED GOOGLETEST_SOURCE_DIR)
113 MESSAGE(STATUS "Downloading Google Test to ${CMAKE_BINARY_DIR}/googletest-source (define GOOGLETEST_SOURCE_DIR to avoid it)")
114 CONFIGURE_FILE(cmake/DownloadGoogleTest.cmake "${CMAKE_BINARY_DIR}/googletest-download/CMakeLists.txt")
115 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" -G "${CMAKE_GENERATOR}" .
116 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/googletest-download")
117 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" --build .
118 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/googletest-download")
119 SET(GOOGLETEST_SOURCE_DIR "${CMAKE_BINARY_DIR}/googletest-source" CACHE STRING "Google Test source directory")
120ENDIF()
121
122IF(XNNPACK_BUILD_BENCHMARKS AND NOT DEFINED GOOGLEBENCHMARK_SOURCE_DIR)
123 MESSAGE(STATUS "Downloading Google Benchmark to ${CMAKE_BINARY_DIR}/googlebenchmark-source (define GOOGLEBENCHMARK_SOURCE_DIR to avoid it)")
124 CONFIGURE_FILE(cmake/DownloadGoogleBenchmark.cmake "${CMAKE_BINARY_DIR}/googlebenchmark-download/CMakeLists.txt")
125 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" -G "${CMAKE_GENERATOR}" .
126 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/googlebenchmark-download")
127 EXECUTE_PROCESS(COMMAND "${CMAKE_COMMAND}" --build .
128 WORKING_DIRECTORY "${CMAKE_BINARY_DIR}/googlebenchmark-download")
129 SET(GOOGLEBENCHMARK_SOURCE_DIR "${CMAKE_BINARY_DIR}/googlebenchmark-source" CACHE STRING "Google Benchmark source directory")
130ENDIF()
131
132# ---[ XNNPACK library
133SET(XNNPACK_OPERATOR_SRCS
Marat Dukhanefc47b82019-11-18 09:25:38 -0800134 src/add-nc.c
135 src/argmax-pooling-nhwc.c
136 src/average-pooling-nhwc.c
Marat Dukhanb1a0fc32019-12-02 19:32:02 -0800137 src/binary-elementwise-nd.c
Marat Dukhanefc47b82019-11-18 09:25:38 -0800138 src/channel-pad-nc.c
139 src/channel-shuffle-nc.c
140 src/clamp-nc.c
141 src/convolution-nchw.c
142 src/convolution-nhwc.c
143 src/deconvolution-nhwc.c
144 src/fully-connected-nc.c
145 src/global-average-pooling-ncw.c
146 src/global-average-pooling-nwc.c
147 src/hardswish-nc.c
148 src/leaky-relu-nc.c
149 src/max-pooling-nhwc.c
Marat Dukhanefc47b82019-11-18 09:25:38 -0800150 src/prelu-nc.c
151 src/resize-bilinear-nhwc.c
152 src/sigmoid-nc.c
153 src/softargmax-nc.c
154 src/unpooling-nhwc.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700155
156SET(XNNPACK_COLD_SRCS ${XNNPACK_OPERATOR_SRCS})
157LIST(APPEND XNNPACK_COLD_SRCS
158 src/init.c
Ashkan Aliabadi1f8a2b82019-11-20 11:27:00 -0800159 src/memory.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700160 src/operator-delete.c)
161
162SET(XNNPACK_HOT_SRCS
163 src/indirection.c
164 src/operator-run.c)
165
166SET(XNNPACK_SCALAR_MICROKERNEL_SRCS
Ashkan Aliabadif866a452019-11-20 10:58:51 -0800167 src/f32-argmaxpool/4x-scalar-c1.c
168 src/f32-argmaxpool/9p8x-scalar-c1.c
169 src/f32-argmaxpool/9x-scalar-c1.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700170 src/f32-avgpool/mp9p8q-scalar.c
171 src/f32-avgpool/up9-scalar.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800172 src/f32-bilinear/gen/scalar-c1.c
173 src/f32-bilinear/gen/scalar-c2.c
174 src/f32-bilinear/gen/scalar-c4.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700175 src/f32-clamp/scalar.c
Marat Dukhan6b7dfae2019-12-04 16:00:52 -0800176 src/f32-conv-hwc/3x3s2p1c3x4-scalar-1x1.c
Erich Elsen563df5f2019-10-23 08:02:21 -0700177 src/f32-conv-hwc2spchw/3x3s2p1c3x4-scalar-1x1.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
Marat Dukhan1e782c42019-11-21 17:02:40 -0800180 src/f32-dwconv-spchw/5x5p2-scalar.c
Erich Elsen38709a62019-11-08 11:58:45 -0800181 src/f32-dwconv-spchw/5x5s2p2-scalar.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800182 src/f32-dwconv/gen/up1x25-scalar-acc2.c
183 src/f32-dwconv/gen/up1x25-scalar.c
184 src/f32-dwconv/gen/up1x4-scalar-acc2.c
185 src/f32-dwconv/gen/up1x4-scalar.c
186 src/f32-dwconv/gen/up1x9-scalar-acc2.c
187 src/f32-dwconv/gen/up1x9-scalar.c
188 src/f32-dwconv/gen/up2x25-scalar-acc2.c
189 src/f32-dwconv/gen/up2x25-scalar.c
190 src/f32-dwconv/gen/up2x4-scalar-acc2.c
191 src/f32-dwconv/gen/up2x4-scalar.c
192 src/f32-dwconv/gen/up2x9-scalar-acc2.c
193 src/f32-dwconv/gen/up2x9-scalar.c
Erich Elsen34dc2c02019-10-16 05:11:41 -0700194 src/f32-gavgpool-spchw/scalar-x1.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700195 src/f32-gavgpool/mp7p7q-scalar.c
196 src/f32-gavgpool/up7-scalar.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800197 src/f32-gemm/gen/1x4-scalar.c
198 src/f32-gemm/gen/2x4-scalar.c
199 src/f32-gemm/gen/4x2-scalar.c
200 src/f32-gemm/gen/4x4-scalar.c
201 src/f32-gemm/gen-inc/1x4-scalar.c
202 src/f32-gemm/gen-inc/2x4-scalar.c
203 src/f32-gemm/gen-inc/4x4-scalar.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700204 src/f32-hswish/scalar.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800205 src/f32-igemm/gen/1x4-scalar.c
206 src/f32-igemm/gen/2x4-scalar.c
207 src/f32-igemm/gen/4x2-scalar.c
208 src/f32-igemm/gen/4x4-scalar.c
Marat Dukhan329da642019-11-19 21:44:39 -0800209 src/f32-maxpool/9p8x-scalar-c1.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700210 src/f32-pavgpool/mp9p8q-scalar.c
211 src/f32-pavgpool/up9-scalar.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800212 src/f32-ppmm/gen/2x4-scalar.c
213 src/f32-ppmm/gen/3x3-scalar.c
214 src/f32-ppmm/gen/4x2-scalar.c
215 src/f32-ppmm/gen/4x4-scalar.c
216 src/f32-prelu/gen/scalar-2x1.c
217 src/f32-prelu/gen/scalar-2x4.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700218 src/f32-rmax/scalar.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800219 src/f32-spmm/gen/1x1-scalar-pipelined.c
220 src/f32-spmm/gen/1x1-scalar.c
221 src/f32-spmm/gen/2x1-scalar-pipelined.c
222 src/f32-spmm/gen/2x1-scalar.c
223 src/f32-spmm/gen/4x1-scalar-pipelined.c
224 src/f32-spmm/gen/4x1-scalar.c
225 src/f32-spmm/gen/8x1-scalar-pipelined.c
226 src/f32-spmm/gen/8x1-scalar.c
227 src/f32-spmm/gen/8x2-scalar.c
228 src/f32-spmm/gen/8x4-scalar.c
229 src/f32-vbinary/gen/vadd-scalar-x1.c
230 src/f32-vbinary/gen/vadd-scalar-x2.c
231 src/f32-vbinary/gen/vadd-scalar-x4.c
232 src/f32-vbinary/gen/vaddc-scalar-x1.c
233 src/f32-vbinary/gen/vaddc-scalar-x2.c
234 src/f32-vbinary/gen/vaddc-scalar-x4.c
235 src/f32-vbinary/gen/vmul-scalar-x1.c
236 src/f32-vbinary/gen/vmul-scalar-x2.c
237 src/f32-vbinary/gen/vmul-scalar-x4.c
238 src/f32-vbinary/gen/vmulc-scalar-x1.c
239 src/f32-vbinary/gen/vmulc-scalar-x2.c
240 src/f32-vbinary/gen/vmulc-scalar-x4.c
241 src/f32-vbinary/gen/vrsubc-scalar-x1.c
242 src/f32-vbinary/gen/vrsubc-scalar-x2.c
243 src/f32-vbinary/gen/vrsubc-scalar-x4.c
244 src/f32-vbinary/gen/vsub-scalar-x1.c
245 src/f32-vbinary/gen/vsub-scalar-x2.c
246 src/f32-vbinary/gen/vsub-scalar-x4.c
247 src/f32-vbinary/gen/vsubc-scalar-x1.c
248 src/f32-vbinary/gen/vsubc-scalar-x2.c
249 src/f32-vbinary/gen/vsubc-scalar-x4.c
250 src/f32-vmulcaddc/gen/c1-scalar-2x.c
251 src/f32-vmulcaddc/gen/c2-scalar-2x.c
252 src/f32-vmulcaddc/gen/c4-scalar-2x.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700253 src/q8-avgpool/mp9p8q-scalar.c
254 src/q8-avgpool/up9-scalar.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700255 src/q8-dwconv/up1x9-scalar.c
256 src/q8-gavgpool/mp7p7q-scalar.c
257 src/q8-gavgpool/up7-scalar.c
258 src/q8-gemm/2x2-scalar.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800259 src/q8-igemm/2x2-scalar.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700260 src/q8-vadd/scalar.c
261 src/u8-clamp/scalar.c
262 src/u8-lut32norm/scalar.c
Marat Dukhan329da642019-11-19 21:44:39 -0800263 src/u8-maxpool/9p8x-scalar-c1.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700264 src/u8-rmax/scalar.c
265 src/x32-packx/x2-scalar.c
266 src/x32-packx/x3-scalar.c
267 src/x32-packx/x4-scalar.c
268 src/x32-pad/x2-scalar.c
269 src/x32-unpool/scalar.c
270 src/x32-zip/x2-scalar.c
271 src/x32-zip/x3-scalar.c
272 src/x32-zip/x4-scalar.c
273 src/x32-zip/xm-scalar.c
274 src/x8-lut/scalar.c
275 src/x8-zip/x2-scalar.c
276 src/x8-zip/x3-scalar.c
277 src/x8-zip/x4-scalar.c
278 src/x8-zip/xm-scalar.c)
279
280SET(XNNPACK_PSIMD_MICROKERNEL_SRCS
Ashkan Aliabadif866a452019-11-20 10:58:51 -0800281 src/f32-argmaxpool/4x-psimd-c4.c
282 src/f32-argmaxpool/9p8x-psimd-c4.c
283 src/f32-argmaxpool/9x-psimd-c4.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700284 src/f32-avgpool/mp9p8q-psimd.c
285 src/f32-avgpool/up9-psimd.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800286 src/f32-bilinear/gen/psimd-c4.c
287 src/f32-bilinear/gen/psimd-c8.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700288 src/f32-clamp/psimd.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800289 src/f32-dwconv/gen/up4x25-psimd-acc2.c
290 src/f32-dwconv/gen/up4x25-psimd.c
291 src/f32-dwconv/gen/up4x4-psimd-acc2.c
292 src/f32-dwconv/gen/up4x4-psimd.c
293 src/f32-dwconv/gen/up4x9-psimd-acc2.c
294 src/f32-dwconv/gen/up4x9-psimd.c
295 src/f32-dwconv/gen/up8x25-psimd-acc2.c
296 src/f32-dwconv/gen/up8x25-psimd.c
297 src/f32-dwconv/gen/up8x4-psimd-acc2.c
298 src/f32-dwconv/gen/up8x4-psimd.c
299 src/f32-dwconv/gen/up8x9-psimd-acc2.c
300 src/f32-dwconv/gen/up8x9-psimd.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700301 src/f32-gavgpool/mp7p7q-psimd.c
302 src/f32-gavgpool/up7-psimd.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800303 src/f32-gemm/gen/1x8-psimd-loadsplat.c
304 src/f32-gemm/gen/1x8-psimd-splat.c
305 src/f32-gemm/gen/1x8s4-psimd.c
306 src/f32-gemm/gen/4x8-psimd-loadsplat.c
307 src/f32-gemm/gen/4x8-psimd-splat.c
308 src/f32-gemm/gen/4x8s4-psimd.c
309 src/f32-gemm/gen/6x8-psimd-loadsplat.c
310 src/f32-gemm/gen/6x8-psimd-splat.c
311 src/f32-gemm/gen/6x8s4-psimd.c
312 src/f32-gemm/gen-inc/1x8-psimd-loadsplat.c
313 src/f32-gemm/gen-inc/1x8-psimd-splat.c
314 src/f32-gemm/gen-inc/1x8s4-psimd.c
315 src/f32-gemm/gen-inc/4x8-psimd-loadsplat.c
316 src/f32-gemm/gen-inc/4x8-psimd-splat.c
317 src/f32-gemm/gen-inc/4x8s4-psimd.c
318 src/f32-gemm/gen-inc/6x8-psimd-loadsplat.c
319 src/f32-gemm/gen-inc/6x8-psimd-splat.c
320 src/f32-gemm/gen-inc/6x8s4-psimd.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700321 src/f32-hswish/psimd.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800322 src/f32-igemm/gen/1x8-psimd-loadsplat.c
323 src/f32-igemm/gen/1x8-psimd-splat.c
324 src/f32-igemm/gen/1x8s4-psimd.c
325 src/f32-igemm/gen/4x2c4-psimd.c
326 src/f32-igemm/gen/4x8-psimd-loadsplat.c
327 src/f32-igemm/gen/4x8-psimd-splat.c
328 src/f32-igemm/gen/4x8s4-psimd.c
329 src/f32-igemm/gen/6x8-psimd-loadsplat.c
330 src/f32-igemm/gen/6x8-psimd-splat.c
331 src/f32-igemm/gen/6x8s4-psimd.c
Marat Dukhan329da642019-11-19 21:44:39 -0800332 src/f32-maxpool/9p8x-psimd-c4.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700333 src/f32-pavgpool/mp9p8q-psimd.c
334 src/f32-pavgpool/up9-psimd.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800335 src/f32-ppmm/gen/4x8-psimd.c
336 src/f32-prelu/gen/psimd-2x4.c
337 src/f32-prelu/gen/psimd-2x8.c
338 src/f32-vbinary/gen/vadd-psimd-x4.c
339 src/f32-vbinary/gen/vadd-psimd-x8.c
340 src/f32-vbinary/gen/vaddc-psimd-x4.c
341 src/f32-vbinary/gen/vaddc-psimd-x8.c
342 src/f32-vbinary/gen/vmul-psimd-x4.c
343 src/f32-vbinary/gen/vmul-psimd-x8.c
344 src/f32-vbinary/gen/vmulc-psimd-x4.c
345 src/f32-vbinary/gen/vmulc-psimd-x8.c
346 src/f32-vbinary/gen/vrsubc-psimd-x4.c
347 src/f32-vbinary/gen/vrsubc-psimd-x8.c
348 src/f32-vbinary/gen/vsub-psimd-x4.c
349 src/f32-vbinary/gen/vsub-psimd-x8.c
350 src/f32-vbinary/gen/vsubc-psimd-x4.c
351 src/f32-vbinary/gen/vsubc-psimd-x8.c
352 src/f32-vmulcaddc/gen/c4-psimd-2x.c
353 src/f32-vmulcaddc/gen/c8-psimd-2x.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700354 src/x32-packx/x4-psimd.c
355 src/x32-pad/x2-psimd.c
356 src/x32-unpool/psimd.c
357 src/x32-zip/x2-psimd.c
358 src/x32-zip/x3-psimd.c
359 src/x32-zip/x4-psimd.c
360 src/x32-zip/xm-psimd.c)
361
362SET(XNNPACK_NEON_MICROKERNEL_SRCS
Marat Dukhan36aecb52019-11-22 17:21:49 -0800363 src/f32-avgpool/mp9p8q-neon.c
364 src/f32-avgpool/up9-neon.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800365 src/f32-bilinear/gen/neon-c4.c
366 src/f32-bilinear/gen/neon-c8.c
Marat Dukhan36aecb52019-11-22 17:21:49 -0800367 src/f32-clamp/neon.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800368 src/f32-dwconv/gen/up4x9-neon.c
369 src/f32-dwconv/gen/up4x9-neon-acc2.c
370 src/f32-dwconv/gen/up8x9-neon.c
371 src/f32-dwconv/gen/up8x9-neon-acc2.c
Marat Dukhan36aecb52019-11-22 17:21:49 -0800372 src/f32-gavgpool-spchw/neon-x4.c
373 src/f32-gavgpool/mp7p7q-neon.c
374 src/f32-gavgpool/up7-neon.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800375 src/f32-gemm/gen/1x8-neon-lane-ld64.c
376 src/f32-gemm/gen/4x2-neon-lane-ld64.c
377 src/f32-gemm/gen/4x8-neon-lane-ld128.c
378 src/f32-gemm/gen/4x8-neon-lane-ld64.c
379 src/f32-gemm/gen/5x8-neon-lane-ld64.c
380 src/f32-gemm/gen/6x8-neon-lane-ld64.c
Frank Barchard69172d92019-11-26 16:22:39 -0800381 src/f32-gemm/gen/6x8-neon-lane-ld128.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800382 src/f32-gemm/gen/1x8-neon-dup-ld64.c
383 src/f32-gemm/gen/4x8-neon-dup-ld128.c
384 src/f32-gemm/gen/4x8-neon-dup-ld64.c
385 src/f32-gemm/gen/6x8-neon-dup-ld64.c
Frank Barchard69172d92019-11-26 16:22:39 -0800386 src/f32-gemm/gen/6x8-neon-dup-ld128.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800387 src/f32-gemm/gen/1x8s4-neon.c
388 src/f32-gemm/gen/4x8s4-neon.c
389 src/f32-gemm/gen/6x8s4-neon.c
390 src/f32-gemm/gen/8x8s4-neon.c
391 src/f32-gemm/gen-inc/1x8-neon-lane-ld64.c
392 src/f32-gemm/gen-inc/4x8-neon-lane-ld128.c
393 src/f32-gemm/gen-inc/4x8-neon-lane-ld64.c
394 src/f32-gemm/gen-inc/5x8-neon-lane-ld64.c
395 src/f32-gemm/gen-inc/6x8-neon-lane-ld64.c
Frank Barchard69172d92019-11-26 16:22:39 -0800396 src/f32-gemm/gen-inc/6x8-neon-lane-ld128.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800397 src/f32-gemm/gen-inc/1x8-neon-dup-ld64.c
398 src/f32-gemm/gen-inc/4x8-neon-dup-ld128.c
399 src/f32-gemm/gen-inc/4x8-neon-dup-ld64.c
400 src/f32-gemm/gen-inc/6x8-neon-dup-ld64.c
Frank Barchard69172d92019-11-26 16:22:39 -0800401 src/f32-gemm/gen-inc/6x8-neon-dup-ld128.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800402 src/f32-gemm/gen-inc/1x8s4-neon.c
403 src/f32-gemm/gen-inc/4x8s4-neon.c
404 src/f32-gemm/gen-inc/6x8s4-neon.c
405 src/f32-gemm/gen-inc/8x8s4-neon.c
Marat Dukhan36aecb52019-11-22 17:21:49 -0800406 src/f32-hswish/neon.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800407 src/f32-igemm/gen/1x8-neon-lane-ld64.c
408 src/f32-igemm/gen/4x2-neon-lane-ld64.c
409 src/f32-igemm/gen/4x4-neon-lane-ld64.c
410 src/f32-igemm/gen/4x8-neon-lane-ld128.c
411 src/f32-igemm/gen/4x8-neon-lane-ld64.c
412 src/f32-igemm/gen/6x8-neon-lane-ld64.c
Frank Barchard69172d92019-11-26 16:22:39 -0800413 src/f32-igemm/gen/6x8-neon-lane-ld128.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800414 src/f32-igemm/gen/1x8-neon-dup-ld64.c
415 src/f32-igemm/gen/4x8-neon-dup-ld128.c
416 src/f32-igemm/gen/4x8-neon-dup-ld64.c
417 src/f32-igemm/gen/6x8-neon-dup-ld64.c
Frank Barchard69172d92019-11-26 16:22:39 -0800418 src/f32-igemm/gen/6x8-neon-dup-ld128.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800419 src/f32-igemm/gen/1x8s4-neon.c
420 src/f32-igemm/gen/4x8s4-neon.c
421 src/f32-igemm/gen/6x8s4-neon.c
422 src/f32-igemm/gen/8x8s4-neon.c
Marat Dukhan36aecb52019-11-22 17:21:49 -0800423 src/f32-pavgpool/mp9p8q-neon.c
424 src/f32-pavgpool/up9-neon.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800425 src/f32-ppmm/gen/4x8-neon.c
426 src/f32-ppmm/gen/8x8-neon.c
427 src/f32-prelu/gen/neon-2x4.c
428 src/f32-prelu/gen/neon-2x8.c
Marat Dukhan36aecb52019-11-22 17:21:49 -0800429 src/f32-rmax/neon.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800430 src/f32-sigmoid/gen/neon-frac-p9-p10-nr1recps-x16.c
431 src/f32-vbinary/gen/vadd-neon-x4.c
432 src/f32-vbinary/gen/vadd-neon-x8.c
433 src/f32-vbinary/gen/vaddc-neon-x4.c
434 src/f32-vbinary/gen/vaddc-neon-x8.c
435 src/f32-vbinary/gen/vmul-neon-x4.c
436 src/f32-vbinary/gen/vmul-neon-x8.c
437 src/f32-vbinary/gen/vmulc-neon-x4.c
438 src/f32-vbinary/gen/vmulc-neon-x8.c
439 src/f32-vbinary/gen/vrsubc-neon-x4.c
440 src/f32-vbinary/gen/vrsubc-neon-x8.c
441 src/f32-vbinary/gen/vsub-neon-x4.c
442 src/f32-vbinary/gen/vsub-neon-x8.c
443 src/f32-vbinary/gen/vsubc-neon-x4.c
444 src/f32-vbinary/gen/vsubc-neon-x8.c
445 src/f32-vmulcaddc/gen/c4-neon-2x.c
446 src/f32-vmulcaddc/gen/c8-neon-2x.c
Marat Dukhan36aecb52019-11-22 17:21:49 -0800447 src/q8-avgpool/mp9p8q-neon.c
448 src/q8-avgpool/up9-neon.c
449 src/q8-dwconv/up8x9-neon.c
450 src/q8-gavgpool/mp7p7q-neon.c
451 src/q8-gavgpool/up7-neon.c
452 src/q8-gemm/4x8-neon.c
453 src/q8-gemm/8x8-neon.c
454 src/q8-igemm/4x8-neon.c
455 src/q8-igemm/8x8-neon.c
456 src/q8-vadd/neon.c
457 src/u8-clamp/neon.c
458 src/u8-maxpool/9p8x-neon-c16.c
459 src/u8-rmax/neon.c
460 src/x32-packx/x4-neon-st4.c
461 src/x32-pad/x2-neon.c
462 src/x32-zip/x2-neon.c
463 src/x32-zip/x3-neon.c
464 src/x32-zip/x4-neon.c
465 src/x32-zip/xm-neon.c
466 src/x8-zip/x2-neon.c
467 src/x8-zip/x3-neon.c
468 src/x8-zip/x4-neon.c
469 src/x8-zip/xm-neon.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700470
471SET(XNNPACK_NEONFMA_MICROKERNEL_SRCS
Marat Dukhan40a672f2019-11-25 03:08:22 -0800472 src/f32-bilinear/gen/neonfma-c4.c
473 src/f32-bilinear/gen/neonfma-c8.c
474 src/f32-igemm/gen/1x8-neonfma-dup-ld64.c
475 src/f32-igemm/gen/4x8-neonfma-dup-ld128.c
476 src/f32-igemm/gen/4x8-neonfma-dup-ld64.c
477 src/f32-igemm/gen/6x8-neonfma-dup-ld64.c
Frank Barchard69172d92019-11-26 16:22:39 -0800478 src/f32-igemm/gen/6x8-neonfma-dup-ld128.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800479 src/f32-igemm/gen/1x8s4-neonfma.c
480 src/f32-igemm/gen/4x8s4-neonfma.c
481 src/f32-igemm/gen/6x8s4-neonfma.c
482 src/f32-igemm/gen/8x8s4-neonfma.c
483 src/f32-dwconv/gen/up4x9-neonfma.c
484 src/f32-dwconv/gen/up4x9-neonfma-acc2.c
485 src/f32-dwconv/gen/up8x9-neonfma.c
486 src/f32-dwconv/gen/up8x9-neonfma-acc2.c
487 src/f32-gemm/gen/1x8-neonfma-dup-ld64.c
488 src/f32-gemm/gen/4x8-neonfma-dup-ld128.c
489 src/f32-gemm/gen/4x8-neonfma-dup-ld64.c
490 src/f32-gemm/gen/6x8-neonfma-dup-ld64.c
Frank Barchard69172d92019-11-26 16:22:39 -0800491 src/f32-gemm/gen/6x8-neonfma-dup-ld128.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800492 src/f32-gemm/gen/1x8s4-neonfma.c
493 src/f32-gemm/gen/4x8s4-neonfma.c
494 src/f32-gemm/gen/6x8s4-neonfma.c
495 src/f32-gemm/gen/8x8s4-neonfma.c
496 src/f32-gemm/gen-inc/1x8-neonfma-dup-ld64.c
497 src/f32-gemm/gen-inc/4x8-neonfma-dup-ld128.c
498 src/f32-gemm/gen-inc/4x8-neonfma-dup-ld64.c
499 src/f32-gemm/gen-inc/6x8-neonfma-dup-ld64.c
Frank Barchard69172d92019-11-26 16:22:39 -0800500 src/f32-gemm/gen-inc/6x8-neonfma-dup-ld128.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800501 src/f32-gemm/gen-inc/1x8s4-neonfma.c
502 src/f32-gemm/gen-inc/4x8s4-neonfma.c
503 src/f32-gemm/gen-inc/6x8s4-neonfma.c
504 src/f32-gemm/gen-inc/8x8s4-neonfma.c
Marat Dukhan36aecb52019-11-22 17:21:49 -0800505 src/f32-hswish/neonfma.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800506 src/f32-ppmm/gen/4x8-neonfma.c
507 src/f32-ppmm/gen/8x8-neonfma.c
508 src/f32-sigmoid/gen/neonfma-p5-nr2fma-x16.c
509 src/f32-vmulcaddc/gen/c4-neonfma-2x.c
510 src/f32-vmulcaddc/gen/c8-neonfma-2x.c
Marat Dukhan36aecb52019-11-22 17:21:49 -0800511 src/math/exp-neonfma-lut64-p2.c
512 src/math/exp-neonfma-p5.c
Marat Dukhan191e5cd2019-11-28 00:52:01 -0800513 src/math/expminus-neonfma-lut2048-p1.c
Marat Dukhan189ae802019-11-26 11:28:44 -0800514 src/math/expminus-neonfma-lut64-p2.c
Marat Dukhan36aecb52019-11-22 17:21:49 -0800515 src/math/expminus-neonfma-p5.c
Marat Dukhan91f8d862019-11-27 12:25:42 -0800516 src/math/sigmoid-neonfma-lut2048-p1-nr1recps1fma.c
517 src/math/sigmoid-neonfma-lut2048-p1-nr2fma.c
518 src/math/sigmoid-neonfma-lut2048-p1-nr2recps.c
Marat Dukhan36aecb52019-11-22 17:21:49 -0800519 src/math/sigmoid-neonfma-p5-nr1recps1fma.c
520 src/math/sigmoid-neonfma-p5-nr2fma.c
521 src/math/sigmoid-neonfma-p5-nr2recps.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700522
Marat Dukhan36aecb52019-11-22 17:21:49 -0800523SET(XNNPACK_AARCH64_NEONFMA_MICROKERNEL_SRCS
Marat Dukhan40a672f2019-11-25 03:08:22 -0800524 src/f32-gemm/gen/1x8-neonfma-lane-ld64.c
525 src/f32-gemm/gen/4x2-neonfma-lane-ld64.c
526 src/f32-gemm/gen/4x8-neonfma-lane-ld128.c
527 src/f32-gemm/gen/4x8-neonfma-lane-ld64.c
528 src/f32-gemm/gen/5x8-neonfma-lane-ld64.c
529 src/f32-gemm/gen/6x8-neonfma-lane-ld64.c
Frank Barchard69172d92019-11-26 16:22:39 -0800530 src/f32-gemm/gen/6x8-neonfma-lane-ld128.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800531 src/f32-gemm/gen-inc/1x8-neonfma-lane-ld64.c
532 src/f32-gemm/gen-inc/4x8-neonfma-lane-ld128.c
533 src/f32-gemm/gen-inc/4x8-neonfma-lane-ld64.c
534 src/f32-gemm/gen-inc/5x8-neonfma-lane-ld64.c
535 src/f32-gemm/gen-inc/6x8-neonfma-lane-ld64.c
Frank Barchard69172d92019-11-26 16:22:39 -0800536 src/f32-gemm/gen-inc/6x8-neonfma-lane-ld128.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800537 src/f32-igemm/gen/1x8-neonfma-lane-ld64.c
538 src/f32-igemm/gen/4x2-neonfma-lane-ld64.c
539 src/f32-igemm/gen/4x4-neonfma-lane-ld64.c
540 src/f32-igemm/gen/4x8-neonfma-lane-ld128.c
541 src/f32-igemm/gen/4x8-neonfma-lane-ld64.c
542 src/f32-igemm/gen/6x8-neonfma-lane-ld64.c
Frank Barchard69172d92019-11-26 16:22:39 -0800543 src/f32-igemm/gen/6x8-neonfma-lane-ld128.c
Marat Dukhan36aecb52019-11-22 17:21:49 -0800544 src/f32-conv-hwc/3x3s2p1c3x4-neonfma-2x2.c
545 src/f32-conv-hwc/3x3s2p1c3x8-neonfma-2x2.c
546 src/f32-conv-hwc2spchw/3x3s2p1c3x4-neonfma-2x2.c
547 src/f32-dwconv-spchw/3x3p1-neonfma.c
548 src/f32-dwconv-spchw/5x5p2-neonfma.c
549 src/f32-dwconv-spchw/3x3s2p1-neonfma.c
550 src/f32-dwconv-spchw/5x5s2p2-neonfma.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800551 src/f32-spmm/gen/12x1-neonfma.c
552 src/f32-spmm/gen/12x2-neonfma.c
553 src/f32-spmm/gen/12x4-neonfma.c
554 src/f32-spmm/gen/16x1-neonfma-pipelined.c
555 src/f32-spmm/gen/16x1-neonfma-unroll2.c
556 src/f32-spmm/gen/16x1-neonfma.c
557 src/f32-spmm/gen/16x2-neonfma.c
558 src/f32-spmm/gen/16x4-neonfma.c
559 src/f32-spmm/gen/4x1-neonfma-pipelined.c
560 src/f32-spmm/gen/4x1-neonfma-unroll2.c
561 src/f32-spmm/gen/4x1-neonfma.c
562 src/f32-spmm/gen/4x2-neonfma.c
563 src/f32-spmm/gen/4x4-neonfma.c
564 src/f32-spmm/gen/8x1-neonfma-pipelined.c
565 src/f32-spmm/gen/8x1-neonfma-unroll2.c
566 src/f32-spmm/gen/8x1-neonfma.c
567 src/f32-spmm/gen/8x2-neonfma.c
568 src/f32-spmm/gen/8x4-neonfma.c
Marat Dukhan91f8d862019-11-27 12:25:42 -0800569 src/math/sigmoid-neonfma-lut2048-p1-div.c
Marat Dukhan36aecb52019-11-22 17:21:49 -0800570 src/math/sigmoid-neonfma-p5-div.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700571
572SET(XNNPACK_AARCH64_NEONFP16ARITH_MICROKERNEL_SRCS
Marat Dukhan40a672f2019-11-25 03:08:22 -0800573 src/f16-gemm/gen/4x8-neonfp16arith-ld64.c
574 src/f16-gemm/gen/6x8-neonfp16arith-ld64.c
575 src/f16-gemm/gen/8x8-neonfp16arith-ld64.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700576
577SET(XNNPACK_SSE_MICROKERNEL_SRCS
578 src/f32-avgpool/mp9p8q-sse.c
579 src/f32-avgpool/up9-sse.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800580 src/f32-bilinear/gen/sse-c4.c
581 src/f32-bilinear/gen/sse-c8.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700582 src/f32-clamp/sse.c
Marat Dukhan1e782c42019-11-21 17:02:40 -0800583 src/f32-dwconv-spchw/3x3p1-sse.c
584 src/f32-dwconv-spchw/3x3s2p1-sse.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800585 src/f32-dwconv/gen/up4x25-sse-acc2.c
586 src/f32-dwconv/gen/up4x25-sse.c
587 src/f32-dwconv/gen/up4x4-sse-acc2.c
588 src/f32-dwconv/gen/up4x4-sse.c
589 src/f32-dwconv/gen/up4x9-sse-acc2.c
590 src/f32-dwconv/gen/up4x9-sse.c
591 src/f32-dwconv/gen/up8x25-sse-acc2.c
592 src/f32-dwconv/gen/up8x25-sse.c
593 src/f32-dwconv/gen/up8x4-sse-acc2.c
594 src/f32-dwconv/gen/up8x4-sse.c
595 src/f32-dwconv/gen/up8x9-sse-acc2.c
596 src/f32-dwconv/gen/up8x9-sse.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700597 src/f32-gavgpool-spchw/sse-x4.c
598 src/f32-gavgpool/mp7p7q-sse.c
599 src/f32-gavgpool/up7-sse.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800600 src/f32-gemm/gen/1x8-sse-dup.c
601 src/f32-gemm/gen/1x8-sse-load1.c
602 src/f32-gemm/gen/1x8s4-sse.c
603 src/f32-gemm/gen/4x8-sse-dup.c
604 src/f32-gemm/gen/4x8-sse-load1.c
605 src/f32-gemm/gen/4x8s4-sse.c
606 src/f32-gemm/gen-inc/1x8-sse-dup.c
607 src/f32-gemm/gen-inc/1x8-sse-load1.c
608 src/f32-gemm/gen-inc/1x8s4-sse.c
609 src/f32-gemm/gen-inc/4x8-sse-dup.c
610 src/f32-gemm/gen-inc/4x8-sse-load1.c
611 src/f32-gemm/gen-inc/4x8s4-sse.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700612 src/f32-hswish/sse.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800613 src/f32-igemm/gen/1x8-sse-dup.c
614 src/f32-igemm/gen/1x8-sse-load1.c
615 src/f32-igemm/gen/1x8s4-sse.c
616 src/f32-igemm/gen/4x2c4-sse.c
617 src/f32-igemm/gen/4x8-sse-dup.c
618 src/f32-igemm/gen/4x8-sse-load1.c
619 src/f32-igemm/gen/4x8s4-sse.c
Marat Dukhan329da642019-11-19 21:44:39 -0800620 src/f32-maxpool/9p8x-sse-c4.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700621 src/f32-pavgpool/mp9p8q-sse.c
622 src/f32-pavgpool/up9-sse.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800623 src/f32-ppmm/gen/4x8-sse.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700624 src/f32-rmax/sse.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800625 src/f32-spmm/gen/4x1-sse.c
626 src/f32-spmm/gen/8x1-sse.c
627 src/f32-vbinary/gen/vadd-sse-x4.c
628 src/f32-vbinary/gen/vadd-sse-x8.c
629 src/f32-vbinary/gen/vaddc-sse-x4.c
630 src/f32-vbinary/gen/vaddc-sse-x8.c
631 src/f32-vbinary/gen/vmul-sse-x4.c
632 src/f32-vbinary/gen/vmul-sse-x8.c
633 src/f32-vbinary/gen/vmulc-sse-x4.c
634 src/f32-vbinary/gen/vmulc-sse-x8.c
635 src/f32-vbinary/gen/vrsubc-sse-x4.c
636 src/f32-vbinary/gen/vrsubc-sse-x8.c
637 src/f32-vbinary/gen/vsub-sse-x4.c
638 src/f32-vbinary/gen/vsub-sse-x8.c
639 src/f32-vbinary/gen/vsubc-sse-x4.c
640 src/f32-vbinary/gen/vsubc-sse-x8.c
641 src/f32-vmulcaddc/gen/c4-sse-2x.c
642 src/f32-vmulcaddc/gen/c8-sse-2x.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700643 src/x32-packx/x4-sse.c)
644
645SET(XNNPACK_SSE2_MICROKERNEL_SRCS
Ashkan Aliabadif866a452019-11-20 10:58:51 -0800646 src/f32-argmaxpool/4x-sse2-c4.c
647 src/f32-argmaxpool/9p8x-sse2-c4.c
648 src/f32-argmaxpool/9x-sse2-c4.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800649 src/f32-prelu/gen/sse2-2x4.c
650 src/f32-prelu/gen/sse2-2x8.c
651 src/f32-sigmoid/gen/sse2-p5-div-x8.c
652 src/f32-sigmoid/gen/sse2-p5-div-x16.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700653 src/q8-avgpool/mp9p8q-sse2.c
654 src/q8-avgpool/up9-sse2.c
655 src/q8-igemm/4x4c2-sse2.c
656 src/q8-dwconv/up8x9-sse2.c
657 src/q8-gavgpool/mp7p7q-sse2.c
658 src/q8-gavgpool/up7-sse2.c
659 src/q8-gemm/2x4c8-sse2.c
660 src/q8-gemm/4x4c2-sse2.c
661 src/q8-vadd/sse2.c
662 src/u8-clamp/sse2.c
Marat Dukhan329da642019-11-19 21:44:39 -0800663 src/u8-maxpool/9p8x-sse2-c16.c
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700664 src/u8-rmax/sse2.c
665 src/x32-pad/x2-sse2.c
666 src/x32-zip/x2-sse2.c
667 src/x32-zip/x3-sse2.c
668 src/x32-zip/x4-sse2.c
669 src/x32-zip/xm-sse2.c
670 src/x8-zip/x2-sse2.c
671 src/x8-zip/x3-sse2.c
672 src/x8-zip/x4-sse2.c
Marat Dukhan9d501d52019-11-16 02:30:50 -0800673 src/x8-zip/xm-sse2.c
674 src/math/exp-sse2-p5.c
Marat Dukhan80bafd22019-11-18 10:16:01 -0800675 src/math/expminus-sse2-p5.c
676 src/math/sigmoid-sse2-p5-div.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700677
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800678SET(XNNPACK_SSE41_MICROKERNEL_SRCS
Marat Dukhan40a672f2019-11-25 03:08:22 -0800679 src/f32-prelu/gen/sse41-2x4.c
680 src/f32-prelu/gen/sse41-2x8.c
681 src/f32-sigmoid/gen/sse41-p5-div-x8.c
682 src/f32-sigmoid/gen/sse41-p5-div-x16.c)
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800683
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700684SET(XNNPACK_AVX_MICROKERNEL_SRCS
Marat Dukhane2c3f292019-11-27 15:40:54 -0800685 src/f32-clamp/avx.c
Marat Dukhan40a672f2019-11-25 03:08:22 -0800686 src/f32-dwconv/gen/up16x4-avx-acc2.c
687 src/f32-dwconv/gen/up16x4-avx.c
688 src/f32-dwconv/gen/up8x4-avx-acc2.c
689 src/f32-dwconv/gen/up8x4-avx.c
690 src/f32-dwconv/gen/up16x9-avx-acc2.c
691 src/f32-dwconv/gen/up16x9-avx.c
692 src/f32-dwconv/gen/up8x9-avx-acc2.c
693 src/f32-dwconv/gen/up8x9-avx.c
694 src/f32-dwconv/gen/up16x25-avx-acc2.c
695 src/f32-dwconv/gen/up16x25-avx.c
696 src/f32-dwconv/gen/up8x25-avx-acc2.c
697 src/f32-dwconv/gen/up8x25-avx.c
698 src/f32-gemm/gen/1x8-avx-broadcast.c
699 src/f32-gemm/gen/4x8-avx-broadcast.c
700 src/f32-gemm/gen/5x8-avx-broadcast.c
701 src/f32-gemm/gen/6x8-avx-broadcast.c
702 src/f32-gemm/gen/7x8-avx-broadcast.c
703 src/f32-gemm/gen-inc/1x8-avx-broadcast.c
704 src/f32-gemm/gen-inc/4x8-avx-broadcast.c
705 src/f32-gemm/gen-inc/5x8-avx-broadcast.c
706 src/f32-gemm/gen-inc/6x8-avx-broadcast.c
707 src/f32-gemm/gen-inc/7x8-avx-broadcast.c
708 src/f32-igemm/gen/1x8-avx-broadcast.c
709 src/f32-igemm/gen/4x8-avx-broadcast.c
710 src/f32-igemm/gen/5x8-avx-broadcast.c
711 src/f32-igemm/gen/6x8-avx-broadcast.c
712 src/f32-igemm/gen/7x8-avx-broadcast.c
Marat Dukhana5977df2019-10-30 22:02:15 -0700713 src/f32-rmax/avx.c
714 src/f32-vscale/avx-unroll32.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700715
Marat Dukhanfda12b82019-11-21 12:27:59 -0800716SET(XNNPACK_FMA3_MICROKERNEL_SRCS
Marat Dukhan40a672f2019-11-25 03:08:22 -0800717 src/f32-dwconv/gen/up16x4-fma3-acc2.c
718 src/f32-dwconv/gen/up16x4-fma3.c
719 src/f32-dwconv/gen/up8x4-fma3-acc2.c
720 src/f32-dwconv/gen/up8x4-fma3.c
721 src/f32-dwconv/gen/up16x9-fma3-acc2.c
722 src/f32-dwconv/gen/up16x9-fma3.c
723 src/f32-dwconv/gen/up8x9-fma3-acc2.c
724 src/f32-dwconv/gen/up8x9-fma3.c
725 src/f32-dwconv/gen/up16x25-fma3-acc2.c
726 src/f32-dwconv/gen/up16x25-fma3.c
727 src/f32-dwconv/gen/up8x25-fma3-acc2.c
728 src/f32-dwconv/gen/up8x25-fma3.c
729 src/f32-gemm/gen/1x8-fma3-broadcast.c
730 src/f32-gemm/gen/4x8-fma3-broadcast.c
731 src/f32-gemm/gen/5x8-fma3-broadcast.c
732 src/f32-gemm/gen/6x8-fma3-broadcast.c
733 src/f32-gemm/gen/7x8-fma3-broadcast.c
734 src/f32-gemm/gen/8x8-fma3-broadcast.c
735 src/f32-gemm/gen-inc/1x8-fma3-broadcast.c
736 src/f32-gemm/gen-inc/4x8-fma3-broadcast.c
737 src/f32-gemm/gen-inc/5x8-fma3-broadcast.c
738 src/f32-gemm/gen-inc/6x8-fma3-broadcast.c
739 src/f32-gemm/gen-inc/7x8-fma3-broadcast.c
740 src/f32-gemm/gen-inc/8x8-fma3-broadcast.c
741 src/f32-igemm/gen/1x8-fma3-broadcast.c
742 src/f32-igemm/gen/4x8-fma3-broadcast.c
743 src/f32-igemm/gen/5x8-fma3-broadcast.c
744 src/f32-igemm/gen/6x8-fma3-broadcast.c
745 src/f32-igemm/gen/7x8-fma3-broadcast.c
746 src/f32-igemm/gen/8x8-fma3-broadcast.c)
Marat Dukhanfda12b82019-11-21 12:27:59 -0800747
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700748SET(XNNPACK_AVX2_MICROKERNEL_SRCS
Marat Dukhan97579532019-10-18 16:40:39 -0700749 src/f32-raddexpminusmax/avx2-p5-unroll64.c
Marat Dukhan6f8d4d32019-10-25 17:07:09 -0700750 src/f32-raddextexp/avx2-p5-unroll64.c
Marat Dukhan97579532019-10-18 16:40:39 -0700751 src/f32-raddstoreexpminusmax/avx2-p5-unroll64.c
752 src/f32-vscaleexpminusmax/avx2-p5-unroll64.c
Marat Dukhan6f8d4d32019-10-25 17:07:09 -0700753 src/f32-vscaleextexp/avx2-p5-unroll64.c
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700754 src/math/exp-avx2-p5.c
755 src/math/exp-avx2-perm-p3.c
Marat Dukhan515c9772019-10-17 18:07:57 -0700756 src/math/exp-avx2-perm-p4.c
Marat Dukhan9d501d52019-11-16 02:30:50 -0800757 src/math/expminus-avx2-p5.c
758 src/math/extexp-avx2-p5.c)
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700759
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700760SET(XNNPACK_AVX512F_MICROKERNEL_SRCS
Marat Dukhane2c3f292019-11-27 15:40:54 -0800761 src/f32-clamp/avx512f.c
Marat Dukhan479f87e2019-11-27 15:17:06 -0800762 src/f32-dwconv/gen/up32x4-avx512f-acc2.c
763 src/f32-dwconv/gen/up32x4-avx512f.c
764 src/f32-dwconv/gen/up16x4-avx512f-acc2.c
765 src/f32-dwconv/gen/up16x4-avx512f.c
766 src/f32-dwconv/gen/up32x9-avx512f-acc2.c
767 src/f32-dwconv/gen/up32x9-avx512f.c
768 src/f32-dwconv/gen/up16x9-avx512f-acc2.c
769 src/f32-dwconv/gen/up16x9-avx512f.c
770 src/f32-dwconv/gen/up32x25-avx512f-acc2.c
771 src/f32-dwconv/gen/up32x25-avx512f.c
772 src/f32-dwconv/gen/up16x25-avx512f-acc2.c
773 src/f32-dwconv/gen/up16x25-avx512f.c
Marat Dukhan0f349c42019-11-27 11:58:54 -0800774 src/f32-gemm/gen/1x16-avx512f-broadcast.c
775 src/f32-gemm/gen/4x16-avx512f-broadcast.c
776 src/f32-gemm/gen/5x16-avx512f-broadcast.c
777 src/f32-gemm/gen/6x16-avx512f-broadcast.c
778 src/f32-gemm/gen/7x16-avx512f-broadcast.c
779 src/f32-gemm/gen/8x16-avx512f-broadcast.c
780 src/f32-gemm/gen-inc/1x16-avx512f-broadcast.c
781 src/f32-gemm/gen-inc/4x16-avx512f-broadcast.c
782 src/f32-gemm/gen-inc/5x16-avx512f-broadcast.c
783 src/f32-gemm/gen-inc/6x16-avx512f-broadcast.c
784 src/f32-gemm/gen-inc/7x16-avx512f-broadcast.c
785 src/f32-gemm/gen-inc/8x16-avx512f-broadcast.c
786 src/f32-igemm/gen/1x16-avx512f-broadcast.c
787 src/f32-igemm/gen/4x16-avx512f-broadcast.c
788 src/f32-igemm/gen/5x16-avx512f-broadcast.c
789 src/f32-igemm/gen/6x16-avx512f-broadcast.c
790 src/f32-igemm/gen/7x16-avx512f-broadcast.c
791 src/f32-igemm/gen/8x16-avx512f-broadcast.c
Marat Dukhan97579532019-10-18 16:40:39 -0700792 src/f32-raddexpminusmax/avx512f-p5-scalef-unroll128.c
Marat Dukhan6f8d4d32019-10-25 17:07:09 -0700793 src/f32-raddextexp/avx512f-p5-scalef-unroll128.c
Marat Dukhan97579532019-10-18 16:40:39 -0700794 src/f32-raddstoreexpminusmax/avx512f-p5-scalef-unroll128.c
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700795 src/f32-rmax/avx512f.c
Marat Dukhana5977df2019-10-30 22:02:15 -0700796 src/f32-vscale/avx512f-unroll64.c
Marat Dukhan97579532019-10-18 16:40:39 -0700797 src/f32-vscaleexpminusmax/avx512f-p5-scalef-unroll128.c
Marat Dukhan6f8d4d32019-10-25 17:07:09 -0700798 src/f32-vscaleextexp/avx512f-p5-scalef-unroll128.c
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700799 src/math/exp-avx512f-p5-scalef.c
800 src/math/exp-avx512f-p5.c
Marat Dukhanfeb49232019-10-28 11:03:31 -0700801 src/math/exp-avx512f-perm-p3.c
Marat Dukhan9d501d52019-11-16 02:30:50 -0800802 src/math/exp-avx512f-perm2-p2.c
803 src/math/extexp-avx512f-p5.c)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700804
805SET(XNNPACK_AARCH32_ASM_MICROKERNEL_SRCS
Frank Barchard8b0f0262019-11-27 23:18:40 -0800806 src/q8-dwconv/up8x9-aarch32-neon.S
807 src/f32-gemm/4x8-aarch32-neon-ld64.S)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700808
809SET(XNNPACK_AARCH64_ASM_MICROKERNEL_SRCS
810 src/f32-dwconv/up4x9-aarch64-neonfma-cortex-a55.S
811 src/f32-dwconv/up4x9-aarch64-neonfma.S
Marat Dukhan40a672f2019-11-25 03:08:22 -0800812 src/f32-gemm/gen/1x12-aarch64-neonfma-cortex-a53.S
813 src/f32-gemm/gen/1x8-aarch64-neonfma-cortex-a53.S
814 src/f32-gemm/gen/1x8-aarch64-neonfma-cortex-a57.S
815 src/f32-gemm/gen/1x8-aarch64-neonfma-cortex-a75.S
816 src/f32-gemm/gen/4x12-aarch64-neonfma-cortex-a53.S
817 src/f32-gemm/gen/4x8-aarch64-neonfma-cortex-a53.S
818 src/f32-gemm/gen/4x8-aarch64-neonfma-cortex-a57.S
819 src/f32-gemm/gen/4x8-aarch64-neonfma-cortex-a75.S
820 src/f32-gemm/gen/4x8-aarch64-neonfma-ld128.S
821 src/f32-gemm/gen/4x8-aarch64-neonfma-ld64.S
822 src/f32-gemm/gen/5x8-aarch64-neonfma-cortex-a75.S
823 src/f32-gemm/gen/6x8-aarch64-neonfma-cortex-a53.S
824 src/f32-gemm/gen/6x8-aarch64-neonfma-cortex-a57.S
825 src/f32-gemm/gen/6x8-aarch64-neonfma-cortex-a73.S
826 src/f32-gemm/gen/6x8-aarch64-neonfma-cortex-a75.S
827 src/f32-gemm/gen/6x8-aarch64-neonfma-ld128.S
828 src/f32-gemm/gen/6x8-aarch64-neonfma-ld64.S
829 src/f32-gemm/gen-inc/1x12-aarch64-neonfma-cortex-a53.S
830 src/f32-gemm/gen-inc/1x8-aarch64-neonfma-cortex-a53.S
831 src/f32-gemm/gen-inc/1x8-aarch64-neonfma-cortex-a57.S
832 src/f32-gemm/gen-inc/1x8-aarch64-neonfma-cortex-a75.S
833 src/f32-gemm/gen-inc/4x12-aarch64-neonfma-cortex-a53.S
834 src/f32-gemm/gen-inc/4x8-aarch64-neonfma-cortex-a53.S
835 src/f32-gemm/gen-inc/4x8-aarch64-neonfma-cortex-a57.S
836 src/f32-gemm/gen-inc/4x8-aarch64-neonfma-cortex-a75.S
837 src/f32-gemm/gen-inc/4x8-aarch64-neonfma-ld128.S
838 src/f32-gemm/gen-inc/4x8-aarch64-neonfma-ld64.S
839 src/f32-gemm/gen-inc/5x8-aarch64-neonfma-cortex-a75.S
840 src/f32-gemm/gen-inc/6x8-aarch64-neonfma-cortex-a53.S
841 src/f32-gemm/gen-inc/6x8-aarch64-neonfma-cortex-a57.S
842 src/f32-gemm/gen-inc/6x8-aarch64-neonfma-cortex-a73.S
843 src/f32-gemm/gen-inc/6x8-aarch64-neonfma-cortex-a75.S
844 src/f32-gemm/gen-inc/6x8-aarch64-neonfma-ld128.S
845 src/f32-gemm/gen-inc/6x8-aarch64-neonfma-ld64.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700846 src/f32-igemm/1x12-aarch64-neonfma-cortex-a53.S
Frank Barchard21be34f2019-10-09 19:32:19 -0700847 src/f32-igemm/1x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700848 src/f32-igemm/1x8-aarch64-neonfma-cortex-a57.S
849 src/f32-igemm/1x8-aarch64-neonfma-cortex-a75.S
850 src/f32-igemm/4x12-aarch64-neonfma-cortex-a53.S
851 src/f32-igemm/4x8-aarch64-neonfma-cortex-a75.S
852 src/f32-igemm/5x8-aarch64-neonfma-cortex-a75.S
Marat Dukhan8e6e9972019-10-31 14:15:31 -0700853 src/f32-igemm/6x8-aarch64-neonfma-cortex-a53.S
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700854 src/f32-igemm/6x8-aarch64-neonfma-cortex-a57.S
855 src/f32-igemm/6x8-aarch64-neonfma-cortex-a73.S
856 src/f32-igemm/6x8-aarch64-neonfma-cortex-a75.S)
857
858SET(XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SCALAR_MICROKERNEL_SRCS})
859IF(NOT CMAKE_SYSTEM_NAME STREQUAL "Emscripten")
860 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_PSIMD_MICROKERNEL_SRCS})
861ENDIF()
862IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^armv[5-8]" OR IOS_ARCH MATCHES "^armv7")
863 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEON_MICROKERNEL_SRCS})
864 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEONFMA_MICROKERNEL_SRCS})
865 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AARCH32_ASM_MICROKERNEL_SRCS})
866ENDIF()
867IF(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64" OR IOS_ARCH MATCHES "^arm64.*")
868 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEON_MICROKERNEL_SRCS})
869 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_NEONFMA_MICROKERNEL_SRCS})
Marat Dukhan36aecb52019-11-22 17:21:49 -0800870 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AARCH64_NEONFMA_MICROKERNEL_SRCS})
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700871 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AARCH64_NEONFP16ARITH_MICROKERNEL_SRCS})
872 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AARCH64_ASM_MICROKERNEL_SRCS})
873ENDIF()
874IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^(i[3-6]86|x86_64)$" OR IOS_ARCH MATCHES "^(i386|x86_64)$")
875 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SSE_MICROKERNEL_SRCS})
876 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SSE2_MICROKERNEL_SRCS})
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800877 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_SSE41_MICROKERNEL_SRCS})
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700878 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AVX_MICROKERNEL_SRCS})
Marat Dukhanfda12b82019-11-21 12:27:59 -0800879 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_FMA3_MICROKERNEL_SRCS})
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700880 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AVX2_MICROKERNEL_SRCS})
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700881 LIST(APPEND XNNPACK_MICROKERNEL_SRCS ${XNNPACK_AVX512F_MICROKERNEL_SRCS})
882ENDIF()
883
884IF(XNNPACK_LIBRARY_TYPE STREQUAL "default")
885 ADD_LIBRARY(XNNPACK ${XNNPACK_COLD_SRCS} ${XNNPACK_HOT_SRCS} ${XNNPACK_MICROKERNEL_SRCS})
886ELSEIF(XNNPACK_LIBRARY_TYPE STREQUAL "shared")
887 ADD_LIBRARY(XNNPACK SHARED ${XNNPACK_COLD_SRCS} ${XNNPACK_HOT_SRCS} ${XNNPACK_MICROKERNEL_SRCS})
888ELSEIF(XNNPACK_LIBRARY_TYPE STREQUAL "static")
889 ADD_LIBRARY(XNNPACK STATIC ${XNNPACK_COLD_SRCS} ${XNNPACK_HOT_SRCS} ${XNNPACK_MICROKERNEL_SRCS})
890ELSE()
891 MESSAGE(FATAL_ERROR "Unsupported XNNPACK library type \"${XNNPACK_LIBRARY_TYPE}\". Must be \"static\", \"shared\", or \"default\"")
892ENDIF()
893SET_TARGET_PROPERTIES(XNNPACK PROPERTIES
894 C_STANDARD 99
895 C_EXTENSIONS YES)
896IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^armv[5-8]" OR IOS_ARCH MATCHES "^armv7")
897 SET_PROPERTY(SOURCE ${XNNPACK_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -marm ")
898 SET_PROPERTY(SOURCE ${XNNPACK_PSIMD_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfpu=neon ")
899 SET_PROPERTY(SOURCE ${XNNPACK_NEON_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfpu=neon ")
900 SET_PROPERTY(SOURCE ${XNNPACK_NEONFMA_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfpu=neon-vfpv4 ")
901 IF(IOS)
902 SET_PROPERTY(SOURCE ${XNNPACK_AARCH32_ASM_UKERNELS} APPEND_STRING PROPERTY COMPILE_FLAGS " -arch ${IOS_ARCH} ")
903 ENDIF()
904ENDIF()
905IF(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64" OR IOS_ARCH MATCHES "^arm64.*")
906 SET_PROPERTY(SOURCE ${XNNPACK_AARCH64_NEONFP16ARITH_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -march=armv8.2-a+fp16 ")
907 IF(IOS)
908 SET_PROPERTY(SOURCE ${XNNPACK_AARCH64_ASM_UKERNELS} APPEND_STRING PROPERTY COMPILE_FLAGS " -arch ${IOS_ARCH} ")
909 ENDIF()
910ENDIF()
911IF(CMAKE_SYSTEM_PROCESSOR MATCHES "^(i[3-6]86|x86_64)$" OR IOS_ARCH MATCHES "^(i386|x86_64)$")
912 SET_PROPERTY(SOURCE ${XNNPACK_SSE_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -msse ")
913 SET_PROPERTY(SOURCE ${XNNPACK_SSE2_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -msse2 ")
Marat Dukhan69c3f2c2019-11-06 12:30:01 -0800914 SET_PROPERTY(SOURCE ${XNNPACK_SSE41_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -msse4.1 ")
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700915 SET_PROPERTY(SOURCE ${XNNPACK_AVX_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mavx ")
Marat Dukhanfda12b82019-11-21 12:27:59 -0800916 SET_PROPERTY(SOURCE ${XNNPACK_FMA3_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfma ")
Marat Dukhan6adff4e2019-10-14 18:32:07 -0700917 SET_PROPERTY(SOURCE ${XNNPACK_AVX2_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mfma -mavx2 ")
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700918 SET_PROPERTY(SOURCE ${XNNPACK_AVX512F_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -mavx512f ")
919ENDIF()
920IF(CMAKE_BUILD_TYPE STREQUAL "Debug")
921 TARGET_COMPILE_DEFINITIONS(XNNPACK PRIVATE XNN_LOG_LEVEL=5)
922ELSE()
923 TARGET_COMPILE_DEFINITIONS(XNNPACK PRIVATE XNN_LOG_LEVEL=0)
924 SET_PROPERTY(SOURCE ${XNNPACK_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -O2 ")
925 SET_PROPERTY(SOURCE ${XNNPACK_HOT_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -O2 ")
926 SET_PROPERTY(SOURCE ${XNNPACK_COLD_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -Os ")
927ENDIF()
928IF(NOT CMAKE_SYSTEM_NAME STREQUAL "Emscripten")
929 SET_PROPERTY(SOURCE ${XNNPACK_PSIMD_MICROKERNEL_SRCS} APPEND_STRING PROPERTY COMPILE_FLAGS " -O3 -ffast-math ")
930ENDIF()
931
932TARGET_INCLUDE_DIRECTORIES(XNNPACK PUBLIC include)
933TARGET_INCLUDE_DIRECTORIES(XNNPACK PRIVATE src)
Ashkan Aliabadid255a312019-12-03 12:44:34 -0800934SET_TARGET_PROPERTIES(XNNPACK PROPERTIES PUBLIC_HEADER include/xnnpack.h)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700935
936# ---[ Configure clog
937IF(NOT TARGET clog)
938 SET(CLOG_BUILD_TESTS OFF CACHE BOOL "")
939 SET(CLOG_RUNTIME_TYPE "${CPUINFO_RUNTIME_TYPE}" CACHE STRING "")
940 ADD_SUBDIRECTORY(
941 "${CLOG_SOURCE_DIR}/deps/clog"
942 "${CMAKE_BINARY_DIR}/clog")
943 # We build static version of clog but a dynamic library may indirectly depend on it
944 SET_PROPERTY(TARGET clog PROPERTY POSITION_INDEPENDENT_CODE ON)
945ENDIF()
946TARGET_LINK_LIBRARIES(XNNPACK PRIVATE clog)
947
948# ---[ Configure cpuinfo
949IF(NOT TARGET cpuinfo)
950 SET(CPUINFO_BUILD_TOOLS OFF CACHE BOOL "")
951 SET(CPUINFO_BUILD_UNIT_TESTS OFF CACHE BOOL "")
952 SET(CPUINFO_BUILD_MOCK_TESTS OFF CACHE BOOL "")
953 SET(CPUINFO_BUILD_BENCHMARKS OFF CACHE BOOL "")
954 ADD_SUBDIRECTORY(
955 "${CPUINFO_SOURCE_DIR}"
956 "${CMAKE_BINARY_DIR}/cpuinfo")
957ENDIF()
958TARGET_LINK_LIBRARIES(XNNPACK PRIVATE cpuinfo)
959
960# ---[ Configure pthreadpool
961IF(NOT TARGET pthreadpool)
962 SET(PTHREADPOOL_BUILD_TESTS OFF CACHE BOOL "")
963 SET(PTHREADPOOL_BUILD_BENCHMARKS OFF CACHE BOOL "")
Marat Dukhand70028a2019-10-29 09:53:16 -0700964 SET(PTHREADPOOL_ALLOW_DEPRECATED_API OFF CACHE BOOL "")
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -0700965 ADD_SUBDIRECTORY(
966 "${PTHREADPOOL_SOURCE_DIR}"
967 "${CMAKE_BINARY_DIR}/pthreadpool")
968ENDIF()
969TARGET_LINK_LIBRARIES(XNNPACK PUBLIC pthreadpool)
970
971# ---[ Configure FXdiv
972IF(NOT TARGET fxdiv)
973 SET(FXDIV_BUILD_TESTS OFF CACHE BOOL "")
974 SET(FXDIV_BUILD_BENCHMARKS OFF CACHE BOOL "")
975 ADD_SUBDIRECTORY(
976 "${FXDIV_SOURCE_DIR}"
977 "${CMAKE_BINARY_DIR}/FXdiv")
978ENDIF()
979TARGET_LINK_LIBRARIES(XNNPACK PRIVATE fxdiv)
980
981# ---[ Configure psimd
982IF(NOT TARGET psimd)
983 ADD_SUBDIRECTORY(
984 "${PSIMD_SOURCE_DIR}"
985 "${CMAKE_BINARY_DIR}/psimd")
986ENDIF()
987TARGET_LINK_LIBRARIES(XNNPACK PRIVATE psimd)
988
989# ---[ Configure FP16
990IF(NOT TARGET fp16)
991 SET(FP16_BUILD_TESTS OFF CACHE BOOL "")
992 SET(FP16_BUILD_BENCHMARKS OFF CACHE BOOL "")
993 ADD_SUBDIRECTORY(
994 "${FP16_SOURCE_DIR}"
995 "${CMAKE_BINARY_DIR}/FP16")
996ENDIF()
997TARGET_LINK_LIBRARIES(XNNPACK PRIVATE fp16)
998
999INSTALL(TARGETS XNNPACK
1000 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
1001 ARCHIVE DESTINATION ${CMAKE_INSTALL_LIBDIR}
1002 PUBLIC_HEADER DESTINATION ${CMAKE_INSTALL_INCLUDEDIR})
1003
1004# ---[ XNNPACK unit tests
1005IF(XNNPACK_BUILD_TESTS)
1006 # ---[ Build google test
1007 IF(NOT TARGET gtest)
1008 SET(gtest_force_shared_crt ON CACHE BOOL "" FORCE)
1009 ADD_SUBDIRECTORY(
1010 "${GOOGLETEST_SOURCE_DIR}"
1011 "${CMAKE_BINARY_DIR}/googletest")
1012 ENDIF()
1013
1014 # ---[ Build operator-level unit tests
Marat Dukhanefc47b82019-11-18 09:25:38 -08001015 ADD_EXECUTABLE(add-nc-test test/add-nc.cc)
1016 SET_TARGET_PROPERTIES(add-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001017 CXX_STANDARD 11
1018 CXX_STANDARD_REQUIRED YES
1019 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001020 TARGET_INCLUDE_DIRECTORIES(add-nc-test PRIVATE src test)
1021 TARGET_LINK_LIBRARIES(add-nc-test PRIVATE XNNPACK gtest gtest_main)
1022 ADD_TEST(add-nc-test add-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001023
Marat Dukhanb1a0fc32019-12-02 19:32:02 -08001024 ADD_EXECUTABLE(add-nd-test test/add-nd.cc)
1025 SET_TARGET_PROPERTIES(add-nd-test PROPERTIES
1026 CXX_STANDARD 11
1027 CXX_STANDARD_REQUIRED YES
1028 CXX_EXTENSIONS NO)
1029 TARGET_INCLUDE_DIRECTORIES(add-nd-test PRIVATE src test)
1030 TARGET_LINK_LIBRARIES(add-nd-test PRIVATE XNNPACK gtest gtest_main)
1031 ADD_TEST(add-nd-test add-nd-test)
1032
Marat Dukhanefc47b82019-11-18 09:25:38 -08001033 ADD_EXECUTABLE(argmax-pooling-nhwc-test test/argmax-pooling-nhwc.cc)
1034 SET_TARGET_PROPERTIES(argmax-pooling-nhwc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001035 CXX_STANDARD 11
1036 CXX_STANDARD_REQUIRED YES
1037 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001038 TARGET_INCLUDE_DIRECTORIES(argmax-pooling-nhwc-test PRIVATE src test)
1039 TARGET_LINK_LIBRARIES(argmax-pooling-nhwc-test PRIVATE XNNPACK gtest gtest_main)
1040 ADD_TEST(argmax-pooling-nhwc-test argmax-pooling-nhwc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001041
Marat Dukhanefc47b82019-11-18 09:25:38 -08001042 ADD_EXECUTABLE(average-pooling-nhwc-test test/average-pooling-nhwc.cc)
1043 SET_TARGET_PROPERTIES(average-pooling-nhwc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001044 CXX_STANDARD 11
1045 CXX_STANDARD_REQUIRED YES
1046 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001047 TARGET_INCLUDE_DIRECTORIES(average-pooling-nhwc-test PRIVATE src test)
1048 TARGET_LINK_LIBRARIES(average-pooling-nhwc-test PRIVATE XNNPACK gtest gtest_main)
1049 ADD_TEST(average-pooling-nhwc-test average-pooling-nhwc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001050
Marat Dukhanefc47b82019-11-18 09:25:38 -08001051 ADD_EXECUTABLE(channel-pad-nc-test test/channel-pad-nc.cc)
1052 SET_TARGET_PROPERTIES(channel-pad-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001053 CXX_STANDARD 11
1054 CXX_STANDARD_REQUIRED YES
1055 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001056 TARGET_INCLUDE_DIRECTORIES(channel-pad-nc-test PRIVATE src test)
1057 TARGET_LINK_LIBRARIES(channel-pad-nc-test PRIVATE XNNPACK gtest gtest_main)
1058 ADD_TEST(channel-pad-nc-test channel-pad-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001059
Marat Dukhanefc47b82019-11-18 09:25:38 -08001060 ADD_EXECUTABLE(channel-shuffle-nc-test test/channel-shuffle-nc.cc)
1061 SET_TARGET_PROPERTIES(channel-shuffle-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001062 CXX_STANDARD 11
1063 CXX_STANDARD_REQUIRED YES
1064 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001065 TARGET_INCLUDE_DIRECTORIES(channel-shuffle-nc-test PRIVATE src test)
1066 TARGET_LINK_LIBRARIES(channel-shuffle-nc-test PRIVATE XNNPACK gtest gtest_main)
1067 ADD_TEST(channel-shuffle-nc-test channel-shuffle-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001068
Marat Dukhanefc47b82019-11-18 09:25:38 -08001069 ADD_EXECUTABLE(clamp-nc-test test/clamp-nc.cc)
1070 SET_TARGET_PROPERTIES(clamp-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001071 CXX_STANDARD 11
1072 CXX_STANDARD_REQUIRED YES
1073 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001074 TARGET_INCLUDE_DIRECTORIES(clamp-nc-test PRIVATE src test)
1075 TARGET_LINK_LIBRARIES(clamp-nc-test PRIVATE XNNPACK gtest gtest_main)
1076 ADD_TEST(clamp-nc-test clamp-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001077
Marat Dukhanefc47b82019-11-18 09:25:38 -08001078 ADD_EXECUTABLE(convolution-nhwc-test test/convolution-nhwc.cc)
1079 SET_TARGET_PROPERTIES(convolution-nhwc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001080 CXX_STANDARD 11
1081 CXX_STANDARD_REQUIRED YES
1082 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001083 TARGET_INCLUDE_DIRECTORIES(convolution-nhwc-test PRIVATE src test)
1084 TARGET_LINK_LIBRARIES(convolution-nhwc-test PRIVATE XNNPACK gtest gtest_main)
1085 ADD_TEST(convolution-nhwc-test convolution-nhwc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001086
Marat Dukhanefc47b82019-11-18 09:25:38 -08001087 ADD_EXECUTABLE(convolution-nchw-test test/convolution-nchw.cc)
1088 SET_TARGET_PROPERTIES(convolution-nchw-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001089 CXX_STANDARD 11
1090 CXX_STANDARD_REQUIRED YES
1091 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001092 TARGET_INCLUDE_DIRECTORIES(convolution-nchw-test PRIVATE src test)
1093 TARGET_LINK_LIBRARIES(convolution-nchw-test PRIVATE XNNPACK gtest gtest_main)
1094 ADD_TEST(convolution-nchw-test convolution-nchw-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001095
Marat Dukhanefc47b82019-11-18 09:25:38 -08001096 ADD_EXECUTABLE(deconvolution-nhwc-test test/deconvolution-nhwc.cc)
1097 SET_TARGET_PROPERTIES(deconvolution-nhwc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001098 CXX_STANDARD 11
1099 CXX_STANDARD_REQUIRED YES
1100 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001101 TARGET_INCLUDE_DIRECTORIES(deconvolution-nhwc-test PRIVATE src test)
1102 TARGET_LINK_LIBRARIES(deconvolution-nhwc-test PRIVATE XNNPACK gtest gtest_main)
1103 ADD_TEST(deconvolution-nhwc-test deconvolution-nhwc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001104
Marat Dukhanefc47b82019-11-18 09:25:38 -08001105 ADD_EXECUTABLE(fully-connected-nc-test test/fully-connected-nc.cc)
1106 SET_TARGET_PROPERTIES(fully-connected-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001107 CXX_STANDARD 11
1108 CXX_STANDARD_REQUIRED YES
1109 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001110 TARGET_INCLUDE_DIRECTORIES(fully-connected-nc-test PRIVATE src test)
1111 TARGET_LINK_LIBRARIES(fully-connected-nc-test PRIVATE XNNPACK gtest gtest_main)
1112 ADD_TEST(fully-connected-nc-test fully-connected-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001113
Marat Dukhanefc47b82019-11-18 09:25:38 -08001114 ADD_EXECUTABLE(global-average-pooling-nwc-test test/global-average-pooling-nwc.cc)
1115 SET_TARGET_PROPERTIES(global-average-pooling-nwc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001116 CXX_STANDARD 11
1117 CXX_STANDARD_REQUIRED YES
1118 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001119 TARGET_INCLUDE_DIRECTORIES(global-average-pooling-nwc-test PRIVATE src test)
1120 TARGET_LINK_LIBRARIES(global-average-pooling-nwc-test PRIVATE XNNPACK gtest gtest_main)
1121 ADD_TEST(global-average-pooling-nwc-test global-average-pooling-nwc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001122
Marat Dukhanefc47b82019-11-18 09:25:38 -08001123 ADD_EXECUTABLE(global-average-pooling-ncw-test test/global-average-pooling-ncw.cc)
1124 SET_TARGET_PROPERTIES(global-average-pooling-ncw-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001125 CXX_STANDARD 11
1126 CXX_STANDARD_REQUIRED YES
1127 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001128 TARGET_INCLUDE_DIRECTORIES(global-average-pooling-ncw-test PRIVATE src test)
1129 TARGET_LINK_LIBRARIES(global-average-pooling-ncw-test PRIVATE XNNPACK gtest gtest_main)
1130 ADD_TEST(global-average-pooling-ncw-test global-average-pooling-ncw-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001131
Marat Dukhanefc47b82019-11-18 09:25:38 -08001132 ADD_EXECUTABLE(hardswish-nc-test test/hardswish-nc.cc)
1133 SET_TARGET_PROPERTIES(hardswish-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001134 CXX_STANDARD 11
1135 CXX_STANDARD_REQUIRED YES
1136 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001137 TARGET_INCLUDE_DIRECTORIES(hardswish-nc-test PRIVATE src test)
1138 TARGET_LINK_LIBRARIES(hardswish-nc-test PRIVATE XNNPACK gtest gtest_main)
1139 ADD_TEST(hardswish-nc-test hardswish-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001140
Marat Dukhanefc47b82019-11-18 09:25:38 -08001141 ADD_EXECUTABLE(leaky-relu-nc-test test/leaky-relu-nc.cc)
1142 SET_TARGET_PROPERTIES(leaky-relu-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001143 CXX_STANDARD 11
1144 CXX_STANDARD_REQUIRED YES
1145 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001146 TARGET_INCLUDE_DIRECTORIES(leaky-relu-nc-test PRIVATE src test)
1147 TARGET_LINK_LIBRARIES(leaky-relu-nc-test PRIVATE XNNPACK gtest gtest_main)
1148 ADD_TEST(leaky-relu-nc-test leaky-relu-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001149
Marat Dukhanefc47b82019-11-18 09:25:38 -08001150 ADD_EXECUTABLE(max-pooling-nhwc-test test/max-pooling-nhwc.cc)
1151 SET_TARGET_PROPERTIES(max-pooling-nhwc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001152 CXX_STANDARD 11
1153 CXX_STANDARD_REQUIRED YES
1154 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001155 TARGET_INCLUDE_DIRECTORIES(max-pooling-nhwc-test PRIVATE src test)
1156 TARGET_LINK_LIBRARIES(max-pooling-nhwc-test PRIVATE XNNPACK gtest gtest_main)
1157 ADD_TEST(max-pooling-nhwc-test max-pooling-nhwc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001158
Marat Dukhanefc47b82019-11-18 09:25:38 -08001159 ADD_EXECUTABLE(multiply-nd-test test/multiply-nd.cc)
1160 SET_TARGET_PROPERTIES(multiply-nd-test PROPERTIES
Marat Dukhan9d501d52019-11-16 02:30:50 -08001161 CXX_STANDARD 11
1162 CXX_STANDARD_REQUIRED YES
1163 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001164 TARGET_INCLUDE_DIRECTORIES(multiply-nd-test PRIVATE src test)
1165 TARGET_LINK_LIBRARIES(multiply-nd-test PRIVATE XNNPACK gtest gtest_main)
1166 ADD_TEST(multiply-nd-test multiply-nd-test)
Marat Dukhan9d501d52019-11-16 02:30:50 -08001167
Marat Dukhanefc47b82019-11-18 09:25:38 -08001168 ADD_EXECUTABLE(prelu-nc-test test/prelu-nc.cc)
1169 SET_TARGET_PROPERTIES(prelu-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001170 CXX_STANDARD 11
1171 CXX_STANDARD_REQUIRED YES
1172 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001173 TARGET_INCLUDE_DIRECTORIES(prelu-nc-test PRIVATE src test)
1174 TARGET_LINK_LIBRARIES(prelu-nc-test PRIVATE XNNPACK gtest gtest_main)
1175 ADD_TEST(prelu-nc-test prelu-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001176
Marat Dukhanefc47b82019-11-18 09:25:38 -08001177 ADD_EXECUTABLE(resize-bilinear-nhwc-test test/resize-bilinear-nhwc.cc)
1178 SET_TARGET_PROPERTIES(resize-bilinear-nhwc-test PROPERTIES
Marat Dukhan69722492019-11-11 19:55:50 -08001179 CXX_STANDARD 11
1180 CXX_STANDARD_REQUIRED YES
1181 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001182 TARGET_INCLUDE_DIRECTORIES(resize-bilinear-nhwc-test PRIVATE src test)
1183 TARGET_LINK_LIBRARIES(resize-bilinear-nhwc-test PRIVATE XNNPACK gtest gtest_main)
1184 ADD_TEST(resize-bilinear-nhwc-test resize-bilinear-nhwc-test)
Marat Dukhan69722492019-11-11 19:55:50 -08001185
Marat Dukhanefc47b82019-11-18 09:25:38 -08001186 ADD_EXECUTABLE(sigmoid-nc-test test/sigmoid-nc.cc)
1187 SET_TARGET_PROPERTIES(sigmoid-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001188 CXX_STANDARD 11
1189 CXX_STANDARD_REQUIRED YES
1190 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001191 TARGET_INCLUDE_DIRECTORIES(sigmoid-nc-test PRIVATE src test)
1192 TARGET_LINK_LIBRARIES(sigmoid-nc-test PRIVATE XNNPACK gtest gtest_main)
1193 ADD_TEST(sigmoid-nc-test sigmoid-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001194
Marat Dukhanefc47b82019-11-18 09:25:38 -08001195 ADD_EXECUTABLE(softargmax-nc-test test/softargmax-nc.cc)
1196 SET_TARGET_PROPERTIES(softargmax-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001197 CXX_STANDARD 11
1198 CXX_STANDARD_REQUIRED YES
1199 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001200 TARGET_INCLUDE_DIRECTORIES(softargmax-nc-test PRIVATE src test)
1201 TARGET_LINK_LIBRARIES(softargmax-nc-test PRIVATE XNNPACK gtest gtest_main)
1202 ADD_TEST(softargmax-nc-test softargmax-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001203
Marat Dukhan05f3f6d2019-12-03 15:13:53 -08001204 ADD_EXECUTABLE(subtract-nd-test test/subtract-nd.cc)
1205 SET_TARGET_PROPERTIES(subtract-nd-test PROPERTIES
1206 CXX_STANDARD 11
1207 CXX_STANDARD_REQUIRED YES
1208 CXX_EXTENSIONS NO)
1209 TARGET_INCLUDE_DIRECTORIES(subtract-nd-test PRIVATE src test)
1210 TARGET_LINK_LIBRARIES(subtract-nd-test PRIVATE XNNPACK gtest gtest_main)
1211 ADD_TEST(subtract-nd-test subtract-nd-test)
1212
Marat Dukhanefc47b82019-11-18 09:25:38 -08001213 ADD_EXECUTABLE(unpooling-nhwc-test test/unpooling-nhwc.cc)
1214 SET_TARGET_PROPERTIES(unpooling-nhwc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001215 CXX_STANDARD 11
1216 CXX_STANDARD_REQUIRED YES
1217 CXX_EXTENSIONS NO)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001218 TARGET_INCLUDE_DIRECTORIES(unpooling-nhwc-test PRIVATE src test)
1219 TARGET_LINK_LIBRARIES(unpooling-nhwc-test PRIVATE XNNPACK gtest gtest_main)
1220 ADD_TEST(unpooling-nhwc-test unpooling-nhwc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001221
1222 # ---[ Build microkernel-level unit tests
1223 ADD_EXECUTABLE(f16-gemm-test test/f16-gemm.cc)
1224 SET_TARGET_PROPERTIES(f16-gemm-test PROPERTIES
1225 CXX_STANDARD 11
1226 CXX_STANDARD_REQUIRED YES
1227 CXX_EXTENSIONS YES)
1228 TARGET_INCLUDE_DIRECTORIES(f16-gemm-test PRIVATE src test)
1229 TARGET_LINK_LIBRARIES(f16-gemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1230 ADD_TEST(f16-gemm-test f16-gemm-test)
1231
1232 ADD_EXECUTABLE(f32-argmaxpool-test test/f32-argmaxpool.cc)
1233 SET_TARGET_PROPERTIES(f32-argmaxpool-test PROPERTIES
1234 CXX_STANDARD 11
1235 CXX_STANDARD_REQUIRED YES
1236 CXX_EXTENSIONS YES)
1237 TARGET_INCLUDE_DIRECTORIES(f32-argmaxpool-test PRIVATE src test)
1238 TARGET_LINK_LIBRARIES(f32-argmaxpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1239 ADD_TEST(f32-argmaxpool-test f32-argmaxpool-test)
1240
1241 ADD_EXECUTABLE(f32-avgpool-test test/f32-avgpool.cc)
1242 SET_TARGET_PROPERTIES(f32-avgpool-test PROPERTIES
1243 CXX_STANDARD 11
1244 CXX_STANDARD_REQUIRED YES
1245 CXX_EXTENSIONS YES)
1246 TARGET_INCLUDE_DIRECTORIES(f32-avgpool-test PRIVATE src test)
1247 TARGET_LINK_LIBRARIES(f32-avgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1248 ADD_TEST(f32-avgpool-test f32-avgpool-test)
1249
Marat Dukhanc465fc22019-11-07 19:20:28 -08001250 ADD_EXECUTABLE(f32-bilinear-test test/f32-bilinear.cc)
1251 SET_TARGET_PROPERTIES(f32-bilinear-test PROPERTIES
1252 CXX_STANDARD 11
1253 CXX_STANDARD_REQUIRED YES
1254 CXX_EXTENSIONS YES)
1255 TARGET_INCLUDE_DIRECTORIES(f32-bilinear-test PRIVATE src test)
1256 TARGET_LINK_LIBRARIES(f32-bilinear-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1257 ADD_TEST(f32-bilinear-test f32-bilinear-test)
1258
Marat Dukhanefc47b82019-11-18 09:25:38 -08001259 ADD_EXECUTABLE(f32-clamp-nc-test test/f32-clamp.cc)
1260 SET_TARGET_PROPERTIES(f32-clamp-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001261 CXX_STANDARD 11
1262 CXX_STANDARD_REQUIRED YES
1263 CXX_EXTENSIONS YES)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001264 TARGET_INCLUDE_DIRECTORIES(f32-clamp-nc-test PRIVATE src test)
1265 TARGET_LINK_LIBRARIES(f32-clamp-nc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1266 ADD_TEST(f32-clamp-nc-test f32-clamp-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001267
1268 ADD_EXECUTABLE(f32-conv-hwc-test test/f32-conv-hwc.cc)
1269 SET_TARGET_PROPERTIES(f32-conv-hwc-test PROPERTIES
1270 CXX_STANDARD 11
1271 CXX_STANDARD_REQUIRED YES
1272 CXX_EXTENSIONS YES)
1273 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc-test PRIVATE src test)
1274 TARGET_LINK_LIBRARIES(f32-conv-hwc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1275 ADD_TEST(f32-conv-hwc-test f32-conv-hwc-test)
1276
1277 ADD_EXECUTABLE(f32-conv-hwc2spchw-test test/f32-conv-hwc2spchw.cc)
1278 SET_TARGET_PROPERTIES(f32-conv-hwc2spchw-test PROPERTIES
1279 CXX_STANDARD 11
1280 CXX_STANDARD_REQUIRED YES
1281 CXX_EXTENSIONS YES)
1282 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc2spchw-test PRIVATE src test)
1283 TARGET_LINK_LIBRARIES(f32-conv-hwc2spchw-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1284 ADD_TEST(f32-conv-hwc2spchw-test f32-conv-hwc2spchw-test)
1285
1286 ADD_EXECUTABLE(f32-dwconv-spchw-test test/f32-dwconv-spchw.cc)
1287 SET_TARGET_PROPERTIES(f32-dwconv-spchw-test PROPERTIES
1288 CXX_STANDARD 11
1289 CXX_STANDARD_REQUIRED YES
1290 CXX_EXTENSIONS YES)
1291 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-spchw-test PRIVATE src test)
1292 TARGET_LINK_LIBRARIES(f32-dwconv-spchw-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1293 ADD_TEST(f32-dwconv-spchw-test f32-dwconv-spchw-test)
1294
1295 ADD_EXECUTABLE(f32-dwconv-test test/f32-dwconv.cc)
1296 SET_TARGET_PROPERTIES(f32-dwconv-test PROPERTIES
1297 CXX_STANDARD 11
1298 CXX_STANDARD_REQUIRED YES
1299 CXX_EXTENSIONS YES)
1300 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-test PRIVATE src test)
1301 TARGET_LINK_LIBRARIES(f32-dwconv-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1302 ADD_TEST(f32-dwconv-test f32-dwconv-test)
1303
1304 ADD_EXECUTABLE(f32-gavgpool-spchw-test test/f32-gavgpool-spchw.cc)
1305 SET_TARGET_PROPERTIES(f32-gavgpool-spchw-test PROPERTIES
1306 CXX_STANDARD 11
1307 CXX_STANDARD_REQUIRED YES
1308 CXX_EXTENSIONS YES)
1309 TARGET_INCLUDE_DIRECTORIES(f32-gavgpool-spchw-test PRIVATE src test)
1310 TARGET_LINK_LIBRARIES(f32-gavgpool-spchw-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1311 ADD_TEST(f32-gavgpool-spchw-test f32-gavgpool-spchw-test)
1312
1313 ADD_EXECUTABLE(f32-gavgpool-test test/f32-gavgpool.cc)
1314 SET_TARGET_PROPERTIES(f32-gavgpool-test PROPERTIES
1315 CXX_STANDARD 11
1316 CXX_STANDARD_REQUIRED YES
1317 CXX_EXTENSIONS YES)
1318 TARGET_INCLUDE_DIRECTORIES(f32-gavgpool-test PRIVATE src test)
1319 TARGET_LINK_LIBRARIES(f32-gavgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1320 ADD_TEST(f32-gavgpool-test f32-gavgpool-test)
1321
1322 ADD_EXECUTABLE(f32-gemm-test test/f32-gemm.cc)
1323 SET_TARGET_PROPERTIES(f32-gemm-test PROPERTIES
1324 CXX_STANDARD 11
1325 CXX_STANDARD_REQUIRED YES
1326 CXX_EXTENSIONS YES)
1327 TARGET_INCLUDE_DIRECTORIES(f32-gemm-test PRIVATE src test)
1328 TARGET_LINK_LIBRARIES(f32-gemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1329 ADD_TEST(f32-gemm-test f32-gemm-test)
1330
1331 ADD_EXECUTABLE(f32-gemminc-test test/f32-gemminc.cc)
1332 SET_TARGET_PROPERTIES(f32-gemminc-test PROPERTIES
1333 CXX_STANDARD 11
1334 CXX_STANDARD_REQUIRED YES
1335 CXX_EXTENSIONS YES)
1336 TARGET_INCLUDE_DIRECTORIES(f32-gemminc-test PRIVATE src test)
1337 TARGET_LINK_LIBRARIES(f32-gemminc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1338 ADD_TEST(f32-gemminc-test f32-gemminc-test)
1339
1340 ADD_EXECUTABLE(f32-hswish-test test/f32-hswish.cc)
1341 SET_TARGET_PROPERTIES(f32-hswish-test PROPERTIES
1342 CXX_STANDARD 11
1343 CXX_STANDARD_REQUIRED YES
1344 CXX_EXTENSIONS YES)
1345 TARGET_INCLUDE_DIRECTORIES(f32-hswish-test PRIVATE src test)
1346 TARGET_LINK_LIBRARIES(f32-hswish-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1347 ADD_TEST(f32-hswish-test f32-hswish-test)
1348
1349 ADD_EXECUTABLE(f32-igemm-test test/f32-igemm.cc)
1350 SET_TARGET_PROPERTIES(f32-igemm-test PROPERTIES
1351 CXX_STANDARD 11
1352 CXX_STANDARD_REQUIRED YES
1353 CXX_EXTENSIONS YES)
1354 TARGET_INCLUDE_DIRECTORIES(f32-igemm-test PRIVATE src test)
1355 TARGET_LINK_LIBRARIES(f32-igemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1356 ADD_TEST(f32-igemm-test f32-igemm-test)
1357
1358 ADD_EXECUTABLE(f32-maxpool-test test/f32-maxpool.cc)
1359 SET_TARGET_PROPERTIES(f32-maxpool-test PROPERTIES
1360 CXX_STANDARD 11
1361 CXX_STANDARD_REQUIRED YES
1362 CXX_EXTENSIONS YES)
1363 TARGET_INCLUDE_DIRECTORIES(f32-maxpool-test PRIVATE src test)
1364 TARGET_LINK_LIBRARIES(f32-maxpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1365 ADD_TEST(f32-maxpool-test f32-maxpool-test)
1366
1367 ADD_EXECUTABLE(f32-pavgpool-test test/f32-pavgpool.cc)
1368 SET_TARGET_PROPERTIES(f32-pavgpool-test PROPERTIES
1369 CXX_STANDARD 11
1370 CXX_STANDARD_REQUIRED YES
1371 CXX_EXTENSIONS YES)
1372 TARGET_INCLUDE_DIRECTORIES(f32-pavgpool-test PRIVATE src test)
1373 TARGET_LINK_LIBRARIES(f32-pavgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1374 ADD_TEST(f32-pavgpool-test f32-pavgpool-test)
1375
1376 ADD_EXECUTABLE(f32-ppmm-test test/f32-ppmm.cc)
1377 SET_TARGET_PROPERTIES(f32-ppmm-test PROPERTIES
1378 CXX_STANDARD 11
1379 CXX_STANDARD_REQUIRED YES
1380 CXX_EXTENSIONS YES)
1381 TARGET_INCLUDE_DIRECTORIES(f32-ppmm-test PRIVATE src test)
1382 TARGET_LINK_LIBRARIES(f32-ppmm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1383 ADD_TEST(f32-ppmm-test f32-ppmm-test)
1384
Marat Dukhanefc47b82019-11-18 09:25:38 -08001385 ADD_EXECUTABLE(f32-prelu-nc-test test/f32-prelu.cc)
1386 SET_TARGET_PROPERTIES(f32-prelu-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001387 CXX_STANDARD 11
1388 CXX_STANDARD_REQUIRED YES
1389 CXX_EXTENSIONS YES)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001390 TARGET_INCLUDE_DIRECTORIES(f32-prelu-nc-test PRIVATE src test)
1391 TARGET_LINK_LIBRARIES(f32-prelu-nc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1392 ADD_TEST(f32-prelu-nc-test f32-prelu-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001393
Marat Dukhan97579532019-10-18 16:40:39 -07001394 ADD_EXECUTABLE(f32-raddexpminusmax-test test/f32-raddexpminusmax.cc)
1395 SET_TARGET_PROPERTIES(f32-raddexpminusmax-test PROPERTIES
1396 CXX_STANDARD 11
1397 CXX_STANDARD_REQUIRED YES
1398 CXX_EXTENSIONS YES)
1399 TARGET_INCLUDE_DIRECTORIES(f32-raddexpminusmax-test PRIVATE src test)
1400 TARGET_LINK_LIBRARIES(f32-raddexpminusmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1401 ADD_TEST(f32-raddexpminusmax-test f32-raddexpminusmax-test)
1402
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001403 ADD_EXECUTABLE(f32-raddextexp-test test/f32-raddextexp.cc)
1404 SET_TARGET_PROPERTIES(f32-raddextexp-test PROPERTIES
1405 CXX_STANDARD 11
1406 CXX_STANDARD_REQUIRED YES
1407 CXX_EXTENSIONS YES)
1408 TARGET_INCLUDE_DIRECTORIES(f32-raddextexp-test PRIVATE src test)
Marat Dukhana5977df2019-10-30 22:02:15 -07001409 TARGET_LINK_LIBRARIES(f32-raddextexp-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001410 ADD_TEST(f32-raddextexp-test f32-raddextexp-test)
1411
Marat Dukhan97579532019-10-18 16:40:39 -07001412 ADD_EXECUTABLE(f32-raddstoreexpminusmax-test test/f32-raddstoreexpminusmax.cc)
1413 SET_TARGET_PROPERTIES(f32-raddstoreexpminusmax-test PROPERTIES
1414 CXX_STANDARD 11
1415 CXX_STANDARD_REQUIRED YES
1416 CXX_EXTENSIONS YES)
1417 TARGET_INCLUDE_DIRECTORIES(f32-raddstoreexpminusmax-test PRIVATE src test)
1418 TARGET_LINK_LIBRARIES(f32-raddstoreexpminusmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1419 ADD_TEST(f32-raddstoreexpminusmax-test f32-raddstoreexpminusmax-test)
1420
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001421 ADD_EXECUTABLE(f32-rmax-test test/f32-rmax.cc)
1422 SET_TARGET_PROPERTIES(f32-rmax-test PROPERTIES
1423 CXX_STANDARD 11
1424 CXX_STANDARD_REQUIRED YES
1425 CXX_EXTENSIONS YES)
1426 TARGET_INCLUDE_DIRECTORIES(f32-rmax-test PRIVATE src test)
1427 TARGET_LINK_LIBRARIES(f32-rmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1428 ADD_TEST(f32-rmax-test f32-rmax-test)
1429
Marat Dukhan581c1ac2019-11-18 14:57:54 -08001430 ADD_EXECUTABLE(f32-sigmoid-test test/f32-sigmoid.cc)
1431 SET_TARGET_PROPERTIES(f32-sigmoid-test PROPERTIES
1432 CXX_STANDARD 11
1433 CXX_STANDARD_REQUIRED YES
1434 CXX_EXTENSIONS YES)
1435 TARGET_INCLUDE_DIRECTORIES(f32-sigmoid-test PRIVATE src test)
1436 TARGET_LINK_LIBRARIES(f32-sigmoid-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1437 ADD_TEST(f32-sigmoid-test f32-sigmoid-test)
1438
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001439 ADD_EXECUTABLE(f32-spmm-test test/f32-spmm.cc)
1440 SET_TARGET_PROPERTIES(f32-spmm-test PROPERTIES
1441 CXX_STANDARD 11
1442 CXX_STANDARD_REQUIRED YES
1443 CXX_EXTENSIONS YES)
1444 TARGET_INCLUDE_DIRECTORIES(f32-spmm-test PRIVATE src test)
1445 TARGET_LINK_LIBRARIES(f32-spmm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1446 ADD_TEST(f32-spmm-test f32-spmm-test)
1447
Marat Dukhanefc47b82019-11-18 09:25:38 -08001448 ADD_EXECUTABLE(f32-vadd-nc-test test/f32-vadd.cc)
1449 SET_TARGET_PROPERTIES(f32-vadd-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001450 CXX_STANDARD 11
1451 CXX_STANDARD_REQUIRED YES
1452 CXX_EXTENSIONS YES)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001453 TARGET_INCLUDE_DIRECTORIES(f32-vadd-nc-test PRIVATE src test)
1454 TARGET_LINK_LIBRARIES(f32-vadd-nc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1455 ADD_TEST(f32-vadd-nc-test f32-vadd-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001456
Marat Dukhanc76065f2019-11-14 18:13:33 -08001457 ADD_EXECUTABLE(f32-vaddc-test test/f32-vaddc.cc)
1458 SET_TARGET_PROPERTIES(f32-vaddc-test PROPERTIES
1459 CXX_STANDARD 11
1460 CXX_STANDARD_REQUIRED YES
1461 CXX_EXTENSIONS YES)
1462 TARGET_INCLUDE_DIRECTORIES(f32-vaddc-test PRIVATE src test)
1463 TARGET_LINK_LIBRARIES(f32-vaddc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1464 ADD_TEST(f32-vaddc-test f32-vaddc-test)
1465
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001466 ADD_EXECUTABLE(f32-vmul-test test/f32-vmul.cc)
1467 SET_TARGET_PROPERTIES(f32-vmul-test PROPERTIES
1468 CXX_STANDARD 11
1469 CXX_STANDARD_REQUIRED YES
1470 CXX_EXTENSIONS YES)
1471 TARGET_INCLUDE_DIRECTORIES(f32-vmul-test PRIVATE src test)
1472 TARGET_LINK_LIBRARIES(f32-vmul-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1473 ADD_TEST(f32-vmul-test f32-vmul-test)
1474
Marat Dukhanc76065f2019-11-14 18:13:33 -08001475 ADD_EXECUTABLE(f32-vmulc-test test/f32-vmulc.cc)
1476 SET_TARGET_PROPERTIES(f32-vmulc-test PROPERTIES
1477 CXX_STANDARD 11
1478 CXX_STANDARD_REQUIRED YES
1479 CXX_EXTENSIONS YES)
1480 TARGET_INCLUDE_DIRECTORIES(f32-vmulc-test PRIVATE src test)
1481 TARGET_LINK_LIBRARIES(f32-vmulc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1482 ADD_TEST(f32-vmulc-test f32-vmulc-test)
1483
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001484 ADD_EXECUTABLE(f32-vmulcaddc-test test/f32-vmulcaddc.cc)
1485 SET_TARGET_PROPERTIES(f32-vmulcaddc-test PROPERTIES
1486 CXX_STANDARD 11
1487 CXX_STANDARD_REQUIRED YES
1488 CXX_EXTENSIONS YES)
1489 TARGET_INCLUDE_DIRECTORIES(f32-vmulcaddc-test PRIVATE src test)
1490 TARGET_LINK_LIBRARIES(f32-vmulcaddc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1491 ADD_TEST(f32-vmulcaddc-test f32-vmulcaddc-test)
1492
Marat Dukhan97579532019-10-18 16:40:39 -07001493 ADD_EXECUTABLE(f32-vscaleexpminusmax-test test/f32-vscaleexpminusmax.cc)
1494 SET_TARGET_PROPERTIES(f32-vscaleexpminusmax-test PROPERTIES
1495 CXX_STANDARD 11
1496 CXX_STANDARD_REQUIRED YES
1497 CXX_EXTENSIONS YES)
1498 TARGET_INCLUDE_DIRECTORIES(f32-vscaleexpminusmax-test PRIVATE src test)
1499 TARGET_LINK_LIBRARIES(f32-vscaleexpminusmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1500 ADD_TEST(f32-vscaleexpminusmax-test f32-vscaleexpminusmax-test)
1501
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001502 ADD_EXECUTABLE(f32-vscaleextexp-test test/f32-vscaleextexp.cc)
1503 SET_TARGET_PROPERTIES(f32-vscaleexpminusmax-test PROPERTIES
1504 CXX_STANDARD 11
1505 CXX_STANDARD_REQUIRED YES
1506 CXX_EXTENSIONS YES)
1507 TARGET_INCLUDE_DIRECTORIES(f32-vscaleextexp-test PRIVATE src test)
Marat Dukhana5977df2019-10-30 22:02:15 -07001508 TARGET_LINK_LIBRARIES(f32-vscaleextexp-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001509 ADD_TEST(f32-vscaleextexp-test f32-vscaleextexp-test)
1510
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001511 ADD_EXECUTABLE(f32-vsub-test test/f32-vsub.cc)
1512 SET_TARGET_PROPERTIES(f32-vsub-test PROPERTIES
1513 CXX_STANDARD 11
1514 CXX_STANDARD_REQUIRED YES
1515 CXX_EXTENSIONS YES)
1516 TARGET_INCLUDE_DIRECTORIES(f32-vsub-test PRIVATE src test)
1517 TARGET_LINK_LIBRARIES(f32-vsub-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1518 ADD_TEST(f32-vsub-test f32-vsub-test)
1519
Marat Dukhanc76065f2019-11-14 18:13:33 -08001520 ADD_EXECUTABLE(f32-vsubc-test test/f32-vsubc.cc)
1521 SET_TARGET_PROPERTIES(f32-vsubc-test PROPERTIES
1522 CXX_STANDARD 11
1523 CXX_STANDARD_REQUIRED YES
1524 CXX_EXTENSIONS YES)
1525 TARGET_INCLUDE_DIRECTORIES(f32-vsubc-test PRIVATE src test)
1526 TARGET_LINK_LIBRARIES(f32-vsubc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1527 ADD_TEST(f32-vsubc-test f32-vsubc-test)
1528
1529 ADD_EXECUTABLE(f32-vrsubc-test test/f32-vrsubc.cc)
1530 SET_TARGET_PROPERTIES(f32-vrsubc-test PROPERTIES
1531 CXX_STANDARD 11
1532 CXX_STANDARD_REQUIRED YES
1533 CXX_EXTENSIONS YES)
1534 TARGET_INCLUDE_DIRECTORIES(f32-vrsubc-test PRIVATE src test)
1535 TARGET_LINK_LIBRARIES(f32-vrsubc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1536 ADD_TEST(f32-vrsubc-test f32-vrsubc-test)
1537
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001538 ADD_EXECUTABLE(q8-avgpool-test test/q8-avgpool.cc)
1539 SET_TARGET_PROPERTIES(q8-avgpool-test PROPERTIES
1540 CXX_STANDARD 11
1541 CXX_STANDARD_REQUIRED YES
1542 CXX_EXTENSIONS YES)
1543 TARGET_INCLUDE_DIRECTORIES(q8-avgpool-test PRIVATE src test)
1544 TARGET_LINK_LIBRARIES(q8-avgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1545 ADD_TEST(q8-avgpool-test q8-avgpool-test)
1546
1547 ADD_EXECUTABLE(q8-dwconv-test test/q8-dwconv.cc)
1548 SET_TARGET_PROPERTIES(q8-dwconv-test PROPERTIES
1549 CXX_STANDARD 11
1550 CXX_STANDARD_REQUIRED YES
1551 CXX_EXTENSIONS YES)
1552 TARGET_INCLUDE_DIRECTORIES(q8-dwconv-test PRIVATE src test)
1553 TARGET_LINK_LIBRARIES(q8-dwconv-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1554 ADD_TEST(q8-dwconv-test q8-dwconv-test)
1555
1556 ADD_EXECUTABLE(q8-gavgpool-test test/q8-gavgpool.cc)
1557 SET_TARGET_PROPERTIES(q8-gavgpool-test PROPERTIES
1558 CXX_STANDARD 11
1559 CXX_STANDARD_REQUIRED YES
1560 CXX_EXTENSIONS YES)
1561 TARGET_INCLUDE_DIRECTORIES(q8-gavgpool-test PRIVATE src test)
1562 TARGET_LINK_LIBRARIES(q8-gavgpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1563 ADD_TEST(q8-gavgpool-test q8-gavgpool-test)
1564
1565 ADD_EXECUTABLE(q8-gemm-test test/q8-gemm.cc)
1566 SET_TARGET_PROPERTIES(q8-gemm-test PROPERTIES
1567 CXX_STANDARD 11
1568 CXX_STANDARD_REQUIRED YES
1569 CXX_EXTENSIONS YES)
1570 TARGET_INCLUDE_DIRECTORIES(q8-gemm-test PRIVATE src test)
1571 TARGET_LINK_LIBRARIES(q8-gemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1572 ADD_TEST(q8-gemm-test q8-gemm-test)
1573
1574 ADD_EXECUTABLE(q8-igemm-test test/q8-igemm.cc)
1575 SET_TARGET_PROPERTIES(q8-igemm-test PROPERTIES
1576 CXX_STANDARD 11
1577 CXX_STANDARD_REQUIRED YES
1578 CXX_EXTENSIONS YES)
1579 TARGET_INCLUDE_DIRECTORIES(q8-igemm-test PRIVATE src test)
1580 TARGET_LINK_LIBRARIES(q8-igemm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1581 ADD_TEST(q8-igemm-test q8-igemm-test)
1582
Marat Dukhanefc47b82019-11-18 09:25:38 -08001583 ADD_EXECUTABLE(q8-vadd-nc-test test/q8-vadd.cc)
1584 SET_TARGET_PROPERTIES(q8-vadd-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001585 CXX_STANDARD 11
1586 CXX_STANDARD_REQUIRED YES
1587 CXX_EXTENSIONS YES)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001588 TARGET_INCLUDE_DIRECTORIES(q8-vadd-nc-test PRIVATE src test)
1589 TARGET_LINK_LIBRARIES(q8-vadd-nc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1590 ADD_TEST(q8-vadd-nc-test q8-vadd-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001591
Marat Dukhanefc47b82019-11-18 09:25:38 -08001592 ADD_EXECUTABLE(u8-clamp-nc-test test/u8-clamp.cc)
1593 SET_TARGET_PROPERTIES(u8-clamp-nc-test PROPERTIES
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001594 CXX_STANDARD 11
1595 CXX_STANDARD_REQUIRED YES
1596 CXX_EXTENSIONS YES)
Marat Dukhanefc47b82019-11-18 09:25:38 -08001597 TARGET_INCLUDE_DIRECTORIES(u8-clamp-nc-test PRIVATE src test)
1598 TARGET_LINK_LIBRARIES(u8-clamp-nc-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1599 ADD_TEST(u8-clamp-nc-test u8-clamp-nc-test)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001600
1601 ADD_EXECUTABLE(u8-lut32norm-test test/u8-lut32norm.cc)
1602 SET_TARGET_PROPERTIES(u8-lut32norm-test PROPERTIES
1603 CXX_STANDARD 11
1604 CXX_STANDARD_REQUIRED YES
1605 CXX_EXTENSIONS YES)
1606 TARGET_INCLUDE_DIRECTORIES(u8-lut32norm-test PRIVATE src test)
1607 TARGET_LINK_LIBRARIES(u8-lut32norm-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1608 ADD_TEST(u8-lut32norm-test u8-lut32norm-test)
1609
1610 ADD_EXECUTABLE(u8-maxpool-test test/u8-maxpool.cc)
1611 SET_TARGET_PROPERTIES(u8-maxpool-test PROPERTIES
1612 CXX_STANDARD 11
1613 CXX_STANDARD_REQUIRED YES
1614 CXX_EXTENSIONS YES)
1615 TARGET_INCLUDE_DIRECTORIES(u8-maxpool-test PRIVATE src test)
1616 TARGET_LINK_LIBRARIES(u8-maxpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1617 ADD_TEST(u8-maxpool-test u8-maxpool-test)
1618
1619 ADD_EXECUTABLE(u8-rmax-test test/u8-rmax.cc)
1620 SET_TARGET_PROPERTIES(u8-rmax-test PROPERTIES
1621 CXX_STANDARD 11
1622 CXX_STANDARD_REQUIRED YES
1623 CXX_EXTENSIONS YES)
1624 TARGET_INCLUDE_DIRECTORIES(u8-rmax-test PRIVATE src test)
1625 TARGET_LINK_LIBRARIES(u8-rmax-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1626 ADD_TEST(u8-rmax-test u8-rmax-test)
1627
1628 ADD_EXECUTABLE(x32-packx-test test/x32-packx.cc)
1629 SET_TARGET_PROPERTIES(x32-packx-test PROPERTIES
1630 CXX_STANDARD 11
1631 CXX_STANDARD_REQUIRED YES
1632 CXX_EXTENSIONS YES)
1633 TARGET_INCLUDE_DIRECTORIES(x32-packx-test PRIVATE src test)
1634 TARGET_LINK_LIBRARIES(x32-packx-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1635 ADD_TEST(x32-packx-test x32-packx-test)
1636
1637 ADD_EXECUTABLE(x32-pad-test test/x32-pad.cc)
1638 SET_TARGET_PROPERTIES(x32-pad-test PROPERTIES
1639 CXX_STANDARD 11
1640 CXX_STANDARD_REQUIRED YES
1641 CXX_EXTENSIONS YES)
1642 TARGET_INCLUDE_DIRECTORIES(x32-pad-test PRIVATE src test)
1643 TARGET_LINK_LIBRARIES(x32-pad-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1644 ADD_TEST(x32-pad-test x32-pad-test)
1645
1646 ADD_EXECUTABLE(x32-unpool-test test/x32-unpool.cc)
1647 SET_TARGET_PROPERTIES(x32-unpool-test PROPERTIES
1648 CXX_STANDARD 11
1649 CXX_STANDARD_REQUIRED YES
1650 CXX_EXTENSIONS YES)
1651 TARGET_INCLUDE_DIRECTORIES(x32-unpool-test PRIVATE src test)
1652 TARGET_LINK_LIBRARIES(x32-unpool-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1653 ADD_TEST(x32-unpool-test x32-unpool-test)
1654
1655 ADD_EXECUTABLE(x32-zip-test test/x32-zip.cc)
1656 SET_TARGET_PROPERTIES(x32-zip-test PROPERTIES
1657 CXX_STANDARD 11
1658 CXX_STANDARD_REQUIRED YES
1659 CXX_EXTENSIONS YES)
1660 TARGET_INCLUDE_DIRECTORIES(x32-zip-test PRIVATE src test)
1661 TARGET_LINK_LIBRARIES(x32-zip-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1662 ADD_TEST(x32-zip-test x32-zip-test)
1663
1664 ADD_EXECUTABLE(x8-lut-test test/x8-lut.cc)
1665 SET_TARGET_PROPERTIES(x8-lut-test PROPERTIES
1666 CXX_STANDARD 11
1667 CXX_STANDARD_REQUIRED YES
1668 CXX_EXTENSIONS YES)
1669 TARGET_INCLUDE_DIRECTORIES(x8-lut-test PRIVATE src test)
1670 TARGET_LINK_LIBRARIES(x8-lut-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1671 ADD_TEST(x8-lut-test x8-lut-test)
1672
1673 ADD_EXECUTABLE(x8-zip-test test/x8-zip.cc)
1674 SET_TARGET_PROPERTIES(x8-zip-test PROPERTIES
1675 CXX_STANDARD 11
1676 CXX_STANDARD_REQUIRED YES
1677 CXX_EXTENSIONS YES)
1678 TARGET_INCLUDE_DIRECTORIES(x8-zip-test PRIVATE src test)
1679 TARGET_LINK_LIBRARIES(x8-zip-test PRIVATE XNNPACK cpuinfo fp16 gtest gtest_main)
1680 ADD_TEST(x8-zip-test x8-zip-test)
1681ENDIF()
1682
1683# ---[ XNNPACK microbenchmarks
1684IF(XNNPACK_BUILD_BENCHMARKS)
1685 # ---[ Build google benchmark
1686 IF(NOT TARGET benchmark)
1687 SET(BENCHMARK_ENABLE_TESTING OFF CACHE BOOL "")
1688 ADD_SUBDIRECTORY(
1689 "${GOOGLEBENCHMARK_SOURCE_DIR}"
1690 "${CONFU_DEPENDENCIES_BINARY_DIR}/googlebenchmark")
1691 ENDIF()
1692
1693 ADD_LIBRARY(bench-utils bench/utils.cc)
1694 SET_TARGET_PROPERTIES(bench-utils PROPERTIES
1695 CXX_STANDARD 11
1696 CXX_STANDARD_REQUIRED YES
1697 CXX_EXTENSIONS NO)
1698 TARGET_INCLUDE_DIRECTORIES(bench-utils PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
Marat Dukhanbad48fe2019-11-04 10:35:22 -08001699 TARGET_LINK_LIBRARIES(bench-utils PRIVATE benchmark cpuinfo)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001700
Marat Dukhancb10f262019-11-16 15:52:37 -08001701 # ---[ Build accuracy microbenchmarks
1702 ADD_EXECUTABLE(f32-exp-eval eval/f32-exp.cc)
1703 SET_TARGET_PROPERTIES(f32-exp-eval PROPERTIES
1704 CXX_STANDARD 11
1705 CXX_STANDARD_REQUIRED YES
1706 CXX_EXTENSIONS NO)
1707 TARGET_INCLUDE_DIRECTORIES(f32-exp-eval PRIVATE src)
1708 TARGET_LINK_LIBRARIES(f32-exp-eval PRIVATE XNNPACK benchmark fp16)
1709
1710 ADD_EXECUTABLE(f32-expminus-eval eval/f32-expminus.cc)
1711 SET_TARGET_PROPERTIES(f32-expminus-eval PROPERTIES
1712 CXX_STANDARD 11
1713 CXX_STANDARD_REQUIRED YES
1714 CXX_EXTENSIONS NO)
1715 TARGET_INCLUDE_DIRECTORIES(f32-expminus-eval PRIVATE src)
1716 TARGET_LINK_LIBRARIES(f32-expminus-eval PRIVATE XNNPACK benchmark fp16)
1717
1718 ADD_EXECUTABLE(f32-extexp-eval eval/f32-extexp.cc)
1719 SET_TARGET_PROPERTIES(f32-extexp-eval PROPERTIES
1720 CXX_STANDARD 11
1721 CXX_STANDARD_REQUIRED YES
1722 CXX_EXTENSIONS YES)
1723 TARGET_INCLUDE_DIRECTORIES(f32-extexp-eval PRIVATE src)
1724 TARGET_LINK_LIBRARIES(f32-extexp-eval PRIVATE XNNPACK benchmark fp16)
1725
1726 ADD_EXECUTABLE(f32-sigmoid-eval eval/f32-sigmoid.cc)
1727 SET_TARGET_PROPERTIES(f32-sigmoid-eval PROPERTIES
1728 CXX_STANDARD 11
1729 CXX_STANDARD_REQUIRED YES
1730 CXX_EXTENSIONS NO)
1731 TARGET_INCLUDE_DIRECTORIES(f32-sigmoid-eval PRIVATE src)
1732 TARGET_LINK_LIBRARIES(f32-sigmoid-eval PRIVATE XNNPACK benchmark fp16)
1733
Marat Dukhan5f18d262019-10-31 10:24:14 -07001734 # ---[ Build end-to-end microbenchmarks
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001735 ADD_EXECUTABLE(end2end-bench bench/end2end.cc models/mobilenet-v1.cc models/mobilenet-v2.cc)
1736 SET_TARGET_PROPERTIES(end2end-bench PROPERTIES
1737 CXX_STANDARD 11
1738 CXX_STANDARD_REQUIRED YES
1739 CXX_EXTENSIONS NO)
1740 TARGET_INCLUDE_DIRECTORIES(end2end-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
Frank Barcharde72e2872019-10-31 11:12:15 -07001741 TARGET_LINK_LIBRARIES(end2end-bench PRIVATE XNNPACK benchmark bench-utils)
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001742
Marat Dukhanef4416e2019-10-31 13:44:40 -07001743 ADD_EXECUTABLE(f32-dwconv-e2e-bench bench/f32-dwconv-e2e.cc models/mobilenet-v1.cc models/mobilenet-v2.cc)
1744 SET_TARGET_PROPERTIES(f32-dwconv-e2e-bench PROPERTIES
1745 CXX_STANDARD 11
1746 CXX_STANDARD_REQUIRED YES
1747 CXX_EXTENSIONS NO)
Marat Dukhanc9d2f3f2019-10-31 14:02:07 -07001748 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-e2e-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}" src)
Marat Dukhanef4416e2019-10-31 13:44:40 -07001749 TARGET_LINK_LIBRARIES(f32-dwconv-e2e-bench PRIVATE XNNPACK benchmark bench-utils)
1750
Marat Dukhan0a5a53f2019-10-31 10:58:31 -07001751 ADD_EXECUTABLE(f32-gemm-e2e-bench bench/f32-gemm-e2e.cc models/mobilenet-v1.cc models/mobilenet-v2.cc)
1752 SET_TARGET_PROPERTIES(f32-gemm-e2e-bench PROPERTIES
Marat Dukhan5f18d262019-10-31 10:24:14 -07001753 CXX_STANDARD 11
1754 CXX_STANDARD_REQUIRED YES
1755 CXX_EXTENSIONS NO)
Marat Dukhanc9d2f3f2019-10-31 14:02:07 -07001756 TARGET_INCLUDE_DIRECTORIES(f32-gemm-e2e-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}" src)
Frank Barchardc712fa42019-10-31 14:00:21 -07001757 TARGET_LINK_LIBRARIES(f32-gemm-e2e-bench PRIVATE XNNPACK benchmark bench-utils)
Marat Dukhan5f18d262019-10-31 10:24:14 -07001758
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001759 # ---[ Build operator-level microbenchmarks
1760 ADD_EXECUTABLE(add-bench bench/add.cc)
1761 SET_TARGET_PROPERTIES(add-bench PROPERTIES
1762 CXX_STANDARD 11
1763 CXX_STANDARD_REQUIRED YES
1764 CXX_EXTENSIONS NO)
1765 TARGET_INCLUDE_DIRECTORIES(add-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1766 TARGET_LINK_LIBRARIES(add-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1767
1768 ADD_EXECUTABLE(average-pooling-bench bench/average-pooling.cc)
1769 SET_TARGET_PROPERTIES(average-pooling-bench PROPERTIES
1770 CXX_STANDARD 11
1771 CXX_STANDARD_REQUIRED YES
1772 CXX_EXTENSIONS NO)
1773 TARGET_INCLUDE_DIRECTORIES(average-pooling-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1774 TARGET_LINK_LIBRARIES(average-pooling-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1775
1776 ADD_EXECUTABLE(channel-shuffle-bench bench/channel-shuffle.cc)
1777 SET_TARGET_PROPERTIES(channel-shuffle-bench PROPERTIES
1778 CXX_STANDARD 11
1779 CXX_STANDARD_REQUIRED YES
1780 CXX_EXTENSIONS NO)
1781 TARGET_INCLUDE_DIRECTORIES(channel-shuffle-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1782 TARGET_LINK_LIBRARIES(channel-shuffle-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1783
1784 ADD_EXECUTABLE(convolution-bench bench/convolution.cc)
1785 SET_TARGET_PROPERTIES(convolution-bench PROPERTIES
1786 CXX_STANDARD 11
1787 CXX_STANDARD_REQUIRED YES
1788 CXX_EXTENSIONS NO)
1789 TARGET_INCLUDE_DIRECTORIES(convolution-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1790 TARGET_LINK_LIBRARIES(convolution-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1791
1792 ADD_EXECUTABLE(deconvolution-bench bench/deconvolution.cc)
1793 SET_TARGET_PROPERTIES(deconvolution-bench PROPERTIES
1794 CXX_STANDARD 11
1795 CXX_STANDARD_REQUIRED YES
1796 CXX_EXTENSIONS NO)
1797 TARGET_INCLUDE_DIRECTORIES(deconvolution-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1798 TARGET_LINK_LIBRARIES(deconvolution-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1799
1800 ADD_EXECUTABLE(global-average-pooling-bench bench/global-average-pooling.cc)
1801 SET_TARGET_PROPERTIES(global-average-pooling-bench PROPERTIES
1802 CXX_STANDARD 11
1803 CXX_STANDARD_REQUIRED YES
1804 CXX_EXTENSIONS NO)
1805 TARGET_INCLUDE_DIRECTORIES(global-average-pooling-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1806 TARGET_LINK_LIBRARIES(global-average-pooling-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1807
1808 ADD_EXECUTABLE(max-pooling-bench bench/max-pooling.cc)
1809 SET_TARGET_PROPERTIES(max-pooling-bench PROPERTIES
1810 CXX_STANDARD 11
1811 CXX_STANDARD_REQUIRED YES
1812 CXX_EXTENSIONS NO)
1813 TARGET_INCLUDE_DIRECTORIES(max-pooling-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1814 TARGET_LINK_LIBRARIES(max-pooling-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1815
Marat Dukhan95b22432019-10-30 16:30:14 -07001816 ADD_EXECUTABLE(prelu-bench bench/prelu.cc)
1817 SET_TARGET_PROPERTIES(prelu-bench PROPERTIES
1818 CXX_STANDARD 11
1819 CXX_STANDARD_REQUIRED YES
1820 CXX_EXTENSIONS NO)
1821 TARGET_INCLUDE_DIRECTORIES(prelu-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1822 TARGET_LINK_LIBRARIES(prelu-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1823
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001824 ADD_EXECUTABLE(sigmoid-bench bench/sigmoid.cc)
1825 SET_TARGET_PROPERTIES(sigmoid-bench PROPERTIES
1826 CXX_STANDARD 11
1827 CXX_STANDARD_REQUIRED YES
1828 CXX_EXTENSIONS NO)
1829 TARGET_INCLUDE_DIRECTORIES(sigmoid-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1830 TARGET_LINK_LIBRARIES(sigmoid-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1831
1832 ADD_EXECUTABLE(softargmax-bench bench/softargmax.cc)
1833 SET_TARGET_PROPERTIES(softargmax-bench PROPERTIES
1834 CXX_STANDARD 11
1835 CXX_STANDARD_REQUIRED YES
1836 CXX_EXTENSIONS NO)
1837 TARGET_INCLUDE_DIRECTORIES(softargmax-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1838 TARGET_LINK_LIBRARIES(softargmax-bench PRIVATE XNNPACK cpuinfo benchmark bench-utils)
1839
1840 # ---[ Build microkernel-level microbenchmarks
1841 ADD_EXECUTABLE(f16-gemm-bench bench/f16-gemm.cc)
1842 SET_TARGET_PROPERTIES(f16-gemm-bench PROPERTIES
1843 CXX_STANDARD 11
1844 CXX_STANDARD_REQUIRED YES
1845 CXX_EXTENSIONS YES)
1846 TARGET_INCLUDE_DIRECTORIES(f16-gemm-bench PRIVATE src)
1847 TARGET_INCLUDE_DIRECTORIES(f16-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1848 TARGET_LINK_LIBRARIES(f16-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1849
1850 ADD_EXECUTABLE(f32-conv-hwc-bench bench/f32-conv-hwc.cc)
1851 SET_TARGET_PROPERTIES(f32-conv-hwc-bench PROPERTIES
1852 CXX_STANDARD 11
1853 CXX_STANDARD_REQUIRED YES
1854 CXX_EXTENSIONS YES)
1855 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc-bench PRIVATE src)
1856 TARGET_INCLUDE_DIRECTORIES(f32-conv-hwc-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1857 TARGET_LINK_LIBRARIES(f32-conv-hwc-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1858
1859 ADD_EXECUTABLE(f32-dwconv-spchw-bench bench/f32-dwconv-spchw.cc)
1860 SET_TARGET_PROPERTIES(f32-dwconv-spchw-bench PROPERTIES
1861 CXX_STANDARD 11
1862 CXX_STANDARD_REQUIRED YES
1863 CXX_EXTENSIONS YES)
1864 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-spchw-bench PRIVATE src)
1865 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-spchw-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1866 TARGET_LINK_LIBRARIES(f32-dwconv-spchw-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1867
1868 ADD_EXECUTABLE(f32-dwconv-bench bench/f32-dwconv.cc)
1869 SET_TARGET_PROPERTIES(f32-dwconv-bench PROPERTIES
1870 CXX_STANDARD 11
1871 CXX_STANDARD_REQUIRED YES
1872 CXX_EXTENSIONS YES)
1873 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-bench PRIVATE src)
1874 TARGET_INCLUDE_DIRECTORIES(f32-dwconv-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1875 TARGET_LINK_LIBRARIES(f32-dwconv-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1876
1877 ADD_EXECUTABLE(f32-gemm-bench bench/f32-gemm.cc)
1878 SET_TARGET_PROPERTIES(f32-gemm-bench PROPERTIES
1879 CXX_STANDARD 11
1880 CXX_STANDARD_REQUIRED YES
1881 CXX_EXTENSIONS YES)
1882 TARGET_INCLUDE_DIRECTORIES(f32-gemm-bench PRIVATE src)
1883 TARGET_INCLUDE_DIRECTORIES(f32-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1884 TARGET_LINK_LIBRARIES(f32-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1885
1886 ADD_EXECUTABLE(f32-igemm-bench bench/f32-igemm.cc)
1887 SET_TARGET_PROPERTIES(f32-igemm-bench PROPERTIES
1888 CXX_STANDARD 11
1889 CXX_STANDARD_REQUIRED YES
1890 CXX_EXTENSIONS YES)
1891 TARGET_INCLUDE_DIRECTORIES(f32-igemm-bench PRIVATE src)
1892 TARGET_INCLUDE_DIRECTORIES(f32-igemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1893 TARGET_LINK_LIBRARIES(f32-igemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1894
1895 ADD_EXECUTABLE(f32-im2col-gemm-bench bench/f32-im2col-gemm.cc src/im2col.c)
1896 SET_TARGET_PROPERTIES(f32-im2col-gemm-bench PROPERTIES
1897 C_STANDARD 99
1898 C_STANDARD_REQUIRED YES
1899 C_EXTENSIONS NO
1900 CXX_STANDARD 11
1901 CXX_STANDARD_REQUIRED YES
1902 CXX_EXTENSIONS YES)
1903 TARGET_INCLUDE_DIRECTORIES(f32-im2col-gemm-bench PRIVATE src)
1904 TARGET_INCLUDE_DIRECTORIES(f32-im2col-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1905 TARGET_LINK_LIBRARIES(f32-im2col-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1906
1907 ADD_EXECUTABLE(f32-rmax-bench bench/f32-rmax.cc)
1908 SET_TARGET_PROPERTIES(f32-rmax-bench PROPERTIES
1909 CXX_STANDARD 11
1910 CXX_STANDARD_REQUIRED YES
1911 CXX_EXTENSIONS YES)
1912 TARGET_INCLUDE_DIRECTORIES(f32-rmax-bench PRIVATE src)
1913 TARGET_INCLUDE_DIRECTORIES(f32-rmax-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1914 TARGET_LINK_LIBRARIES(f32-rmax-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1915
Marat Dukhan581c1ac2019-11-18 14:57:54 -08001916 ADD_EXECUTABLE(f32-sigmoid-bench bench/f32-sigmoid.cc)
1917 SET_TARGET_PROPERTIES(f32-sigmoid-bench PROPERTIES
1918 CXX_STANDARD 11
1919 CXX_STANDARD_REQUIRED YES
1920 CXX_EXTENSIONS YES)
1921 TARGET_INCLUDE_DIRECTORIES(f32-sigmoid-bench PRIVATE src)
1922 TARGET_INCLUDE_DIRECTORIES(f32-sigmoid-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1923 TARGET_LINK_LIBRARIES(f32-sigmoid-bench PRIVATE XNNPACK benchmark bench-utils)
1924
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001925 ADD_EXECUTABLE(f32-spmm-bench bench/f32-spmm.cc)
1926 SET_TARGET_PROPERTIES(f32-spmm-bench PROPERTIES
1927 CXX_STANDARD 11
1928 CXX_STANDARD_REQUIRED YES
1929 CXX_EXTENSIONS YES)
1930 TARGET_INCLUDE_DIRECTORIES(f32-spmm-bench PRIVATE src)
1931 TARGET_INCLUDE_DIRECTORIES(f32-spmm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1932 TARGET_LINK_LIBRARIES(f32-spmm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1933
Ashkan Aliabadi7892d972019-10-24 16:00:52 -07001934 ADD_EXECUTABLE(f32-softargmax-bench bench/f32-softargmax.cc)
1935 SET_TARGET_PROPERTIES(f32-softargmax-bench PROPERTIES
Marat Dukhan4a4a7fa2019-10-21 13:46:14 -07001936 CXX_STANDARD 11
1937 CXX_STANDARD_REQUIRED YES
1938 CXX_EXTENSIONS YES)
Ashkan Aliabadi7892d972019-10-24 16:00:52 -07001939 TARGET_INCLUDE_DIRECTORIES(f32-softargmax-bench PRIVATE src)
1940 TARGET_INCLUDE_DIRECTORIES(f32-softargmax-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1941 TARGET_LINK_LIBRARIES(f32-softargmax-bench PRIVATE XNNPACK fp16 benchmark bench-utils)
Marat Dukhan4a4a7fa2019-10-21 13:46:14 -07001942
Marat Dukhan2dbdc2f2019-10-08 16:36:23 -07001943 ADD_EXECUTABLE(q8-gemm-bench bench/q8-gemm.cc)
1944 SET_TARGET_PROPERTIES(q8-gemm-bench PROPERTIES
1945 CXX_STANDARD 11
1946 CXX_STANDARD_REQUIRED YES
1947 CXX_EXTENSIONS YES)
1948 TARGET_INCLUDE_DIRECTORIES(q8-gemm-bench PRIVATE src)
1949 TARGET_INCLUDE_DIRECTORIES(q8-gemm-bench PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}")
1950 TARGET_LINK_LIBRARIES(q8-gemm-bench PRIVATE XNNPACK cpuinfo fp16 benchmark bench-utils)
1951ENDIF()