blob: e79afa48490ffe1e65b3b71aec10b4d258617eb4 [file] [log] [blame]
Marat Dukhan08c4a432019-10-03 09:29:21 -07001# Copyright 2019 Google LLC
2#
3# This source code is licensed under the BSD-style license found in the
4# LICENSE file in the root directory of this source tree.
5#
6# Description:
7# XNNPACK - optimized floating-point neural network operators library
8
Marat Dukhan10a38082020-04-17 03:58:35 -07009load(":build_defs.bzl", "xnnpack_aggregate_library", "xnnpack_benchmark", "xnnpack_binary", "xnnpack_cc_library", "xnnpack_gcc_std_copts", "xnnpack_min_size_copts", "xnnpack_msvc_std_copts", "xnnpack_optional_armcl_copts", "xnnpack_optional_armcl_deps", "xnnpack_optional_dnnl_copts", "xnnpack_optional_dnnl_deps", "xnnpack_optional_gemmlowp_copts", "xnnpack_optional_gemmlowp_deps", "xnnpack_optional_ruy_copts", "xnnpack_optional_ruy_deps", "xnnpack_optional_tflite_copts", "xnnpack_optional_tflite_deps", "xnnpack_std_cxxopts", "xnnpack_unit_test", "xnnpack_visibility")
Marat Dukhan69c3f2c2019-11-06 12:30:01 -080010
Marat Dukhan08c4a432019-10-03 09:29:21 -070011licenses(["notice"])
12
13exports_files(["LICENSE"])
14
Marat Dukhan1b354632020-03-23 12:50:22 -070015OPERATOR_BENCHMARK_DEPS = [
16 ":XNNPACK",
17 ":bench_utils",
18 "@cpuinfo",
19 "@pthreadpool",
20]
21
Marat Dukhan08c4a432019-10-03 09:29:21 -070022MICROKERNEL_BENCHMARK_DEPS = [
23 ":ukernels",
24 ":bench_utils",
Frank Barchard7e955972019-10-11 10:34:25 -070025 ":enable_assembly",
Marat Dukhan08c4a432019-10-03 09:29:21 -070026 "@cpuinfo",
27 "@FP16",
28 "@pthreadpool",
29]
30
Marat Dukhan6adff4e2019-10-14 18:32:07 -070031ACCURACY_EVAL_DEPS = [
32 ":XNNPACK",
33 ":ukernels",
34 "@FP16",
35 "@pthreadpool",
36]
37
Marat Dukhan08c4a432019-10-03 09:29:21 -070038MICROKERNEL_TEST_DEPS = [
Marat Dukhan33fcf782020-05-24 14:27:15 -070039 ":ukernels_test_mode",
Frank Barchard7e955972019-10-11 10:34:25 -070040 ":enable_assembly",
Marat Dukhan08c4a432019-10-03 09:29:21 -070041 "@cpuinfo",
42 "@FP16",
43 "@pthreadpool",
44]
45
Marat Dukhan1b354632020-03-23 12:50:22 -070046OPERATOR_TEST_DEPS = [
Marat Dukhan33fcf782020-05-24 14:27:15 -070047 ":XNNPACK_test_mode",
Marat Dukhan1b354632020-03-23 12:50:22 -070048 "@pthreadpool",
49 "@FP16",
50]
51
Marat Dukhan08c4a432019-10-03 09:29:21 -070052OPERATOR_SRCS = [
Marat Dukhane8265432020-04-28 18:42:59 -070053 "src/operators/add-nc.c",
54 "src/operators/argmax-pooling-nhwc.c",
55 "src/operators/average-pooling-nhwc.c",
56 "src/operators/binary-elementwise-nd.c",
57 "src/operators/channel-pad-nc.c",
58 "src/operators/channel-shuffle-nc.c",
59 "src/operators/clamp-nc.c",
Marat Dukhan065b11e2020-05-22 09:49:41 -070060 "src/operators/constant-pad-nd.c",
Marat Dukhane8265432020-04-28 18:42:59 -070061 "src/operators/convolution-nchw.c",
62 "src/operators/convolution-nhwc.c",
63 "src/operators/deconvolution-nhwc.c",
64 "src/operators/fully-connected-nc.c",
65 "src/operators/global-average-pooling-ncw.c",
66 "src/operators/global-average-pooling-nwc.c",
67 "src/operators/hardswish-nc.c",
68 "src/operators/leaky-relu-nc.c",
69 "src/operators/max-pooling-nhwc.c",
70 "src/operators/prelu-nc.c",
71 "src/operators/resize-bilinear-nhwc.c",
72 "src/operators/sigmoid-nc.c",
73 "src/operators/softmax-nc.c",
74 "src/operators/unpooling-nhwc.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -070075]
76
Marat Dukhan3a77ea72019-12-23 12:10:24 -080077TABLE_SRCS = [
78 "src/tables/exp2-k-over-64.c",
79 "src/tables/exp2-k-over-2048.c",
80]
81
Marat Dukhan08c4a432019-10-03 09:29:21 -070082SCALAR_UKERNELS = [
Marat Dukhan329da642019-11-19 21:44:39 -080083 "src/f32-argmaxpool/4x-scalar-c1.c",
Marat Dukhan1e782c42019-11-21 17:02:40 -080084 "src/f32-argmaxpool/9p8x-scalar-c1.c",
Marat Dukhan329da642019-11-19 21:44:39 -080085 "src/f32-argmaxpool/9x-scalar-c1.c",
Marat Dukhan99936602020-04-11 16:47:01 -070086 "src/f32-avgpool/9p8x-minmax-scalar-c1.c",
87 "src/f32-avgpool/9x-minmax-scalar-c1.c",
Marat Dukhan5c5fa962020-03-10 18:38:33 -070088 "src/f32-clamp/gen/scalar-x1.c",
89 "src/f32-clamp/gen/scalar-x2.c",
90 "src/f32-clamp/gen/scalar-x4.c",
Marat Dukhan441e2212019-12-04 18:30:49 -080091 "src/f32-conv-hwc/3x3s2p0p1c3x4-scalar-1x1.c",
Marat Dukhan6b7dfae2019-12-04 16:00:52 -080092 "src/f32-conv-hwc/3x3s2p1c3x4-scalar-1x1.c",
Marat Dukhan1f29b802020-05-15 23:46:39 -070093 "src/f32-conv-hwc2chw/3x3s2p1c3x4-scalar-1x1.c",
94 "src/f32-dwconv-chw/3x3p1-scalar.c",
95 "src/f32-dwconv-chw/3x3s2p1-scalar.c",
96 "src/f32-dwconv-chw/5x5p2-scalar.c",
97 "src/f32-dwconv-chw/5x5s2p2-scalar.c",
Marat Dukhan163a7e62020-04-09 04:19:26 -070098 "src/f32-dwconv/gen/up1x4-scalar-acc2.c",
99 "src/f32-dwconv/gen/up1x4-scalar.c",
100 "src/f32-dwconv/gen/up1x9-scalar-acc2.c",
101 "src/f32-dwconv/gen/up1x9-scalar.c",
102 "src/f32-dwconv/gen/up1x25-scalar-acc2.c",
103 "src/f32-dwconv/gen/up1x25-scalar.c",
104 "src/f32-dwconv/gen/up2x4-scalar-acc2.c",
105 "src/f32-dwconv/gen/up2x4-scalar.c",
106 "src/f32-dwconv/gen/up2x9-scalar-acc2.c",
107 "src/f32-dwconv/gen/up2x9-scalar.c",
108 "src/f32-dwconv/gen/up2x25-scalar-acc2.c",
109 "src/f32-dwconv/gen/up2x25-scalar.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700110 "src/f32-dwconv/gen/up1x4-minmax-scalar-acc2.c",
111 "src/f32-dwconv/gen/up1x4-minmax-scalar.c",
112 "src/f32-dwconv/gen/up1x9-minmax-scalar-acc2.c",
113 "src/f32-dwconv/gen/up1x9-minmax-scalar.c",
Marat Dukhan163a7e62020-04-09 04:19:26 -0700114 "src/f32-dwconv/gen/up1x25-minmax-scalar-acc2.c",
115 "src/f32-dwconv/gen/up1x25-minmax-scalar.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700116 "src/f32-dwconv/gen/up2x4-minmax-scalar-acc2.c",
117 "src/f32-dwconv/gen/up2x4-minmax-scalar.c",
118 "src/f32-dwconv/gen/up2x9-minmax-scalar-acc2.c",
119 "src/f32-dwconv/gen/up2x9-minmax-scalar.c",
Marat Dukhan163a7e62020-04-09 04:19:26 -0700120 "src/f32-dwconv/gen/up2x25-minmax-scalar-acc2.c",
121 "src/f32-dwconv/gen/up2x25-minmax-scalar.c",
Marat Dukhan1f29b802020-05-15 23:46:39 -0700122 "src/f32-gavgpool-cw/scalar-x1.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700123 "src/f32-gavgpool/7p7x-minmax-scalar-c1.c",
124 "src/f32-gavgpool/7x-minmax-scalar-c1.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700125 "src/f32-gemm/gen-inc/1x4inc-minmax-scalar.c",
126 "src/f32-gemm/gen-inc/2x4inc-minmax-scalar.c",
127 "src/f32-gemm/gen-inc/4x4inc-minmax-scalar.c",
Marat Dukhan163a7e62020-04-09 04:19:26 -0700128 "src/f32-gemm/gen/1x4-scalar.c",
129 "src/f32-gemm/gen/2x4-scalar.c",
130 "src/f32-gemm/gen/4x2-scalar.c",
131 "src/f32-gemm/gen/4x4-scalar.c",
Marat Dukhan467f6362020-05-22 23:21:55 -0700132 "src/f32-gemm/gen/1x4-relu-scalar.c",
133 "src/f32-gemm/gen/2x4-relu-scalar.c",
134 "src/f32-gemm/gen/4x2-relu-scalar.c",
135 "src/f32-gemm/gen/4x4-relu-scalar.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700136 "src/f32-gemm/gen/1x4-minmax-scalar.c",
137 "src/f32-gemm/gen/2x4-minmax-scalar.c",
138 "src/f32-gemm/gen/4x2-minmax-scalar.c",
139 "src/f32-gemm/gen/4x4-minmax-scalar.c",
Marat Dukhan662faa02019-12-09 22:48:16 -0800140 "src/f32-hswish/gen/scalar-x1.c",
141 "src/f32-hswish/gen/scalar-x2.c",
142 "src/f32-hswish/gen/scalar-x4.c",
Marat Dukhan660fd192020-03-10 04:55:30 -0700143 "src/f32-ibilinear/gen/scalar-c1.c",
144 "src/f32-ibilinear/gen/scalar-c2.c",
145 "src/f32-ibilinear/gen/scalar-c4.c",
Marat Dukhan163a7e62020-04-09 04:19:26 -0700146 "src/f32-igemm/gen/1x4-scalar.c",
147 "src/f32-igemm/gen/2x4-scalar.c",
148 "src/f32-igemm/gen/4x2-scalar.c",
149 "src/f32-igemm/gen/4x4-scalar.c",
Marat Dukhan467f6362020-05-22 23:21:55 -0700150 "src/f32-igemm/gen/1x4-relu-scalar.c",
151 "src/f32-igemm/gen/2x4-relu-scalar.c",
152 "src/f32-igemm/gen/4x2-relu-scalar.c",
153 "src/f32-igemm/gen/4x4-relu-scalar.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700154 "src/f32-igemm/gen/1x4-minmax-scalar.c",
155 "src/f32-igemm/gen/2x4-minmax-scalar.c",
156 "src/f32-igemm/gen/4x2-minmax-scalar.c",
157 "src/f32-igemm/gen/4x4-minmax-scalar.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700158 "src/f32-maxpool/9p8x-minmax-scalar-c1.c",
159 "src/f32-pavgpool/9p8x-minmax-scalar-c1.c",
160 "src/f32-pavgpool/9x-minmax-scalar-c1.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700161 "src/f32-ppmm/gen/2x4-minmax-scalar.c",
162 "src/f32-ppmm/gen/3x3-minmax-scalar.c",
163 "src/f32-ppmm/gen/4x2-minmax-scalar.c",
164 "src/f32-ppmm/gen/4x4-minmax-scalar.c",
Marat Dukhan40a672f2019-11-25 03:08:22 -0800165 "src/f32-prelu/gen/scalar-2x1.c",
166 "src/f32-prelu/gen/scalar-2x4.c",
Marat Dukhanf46f6752020-01-21 11:03:49 -0800167 "src/f32-raddstoreexpminusmax/gen/scalar-lut64-p2-x1.c",
168 "src/f32-raddstoreexpminusmax/gen/scalar-lut64-p2-x2.c",
169 "src/f32-raddstoreexpminusmax/gen/scalar-lut64-p2-x2-acc2.c",
170 "src/f32-raddstoreexpminusmax/gen/scalar-lut64-p2-x4.c",
171 "src/f32-raddstoreexpminusmax/gen/scalar-lut64-p2-x4-acc2.c",
172 "src/f32-raddstoreexpminusmax/gen/scalar-lut64-p2-x4-acc4.c",
173 "src/f32-raddstoreexpminusmax/gen/scalar-p5-x1.c",
174 "src/f32-raddstoreexpminusmax/gen/scalar-p5-x2.c",
175 "src/f32-raddstoreexpminusmax/gen/scalar-p5-x2-acc2.c",
176 "src/f32-raddstoreexpminusmax/gen/scalar-p5-x4.c",
177 "src/f32-raddstoreexpminusmax/gen/scalar-p5-x4-acc2.c",
178 "src/f32-raddstoreexpminusmax/gen/scalar-p5-x4-acc4.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -0700179 "src/f32-rmax/scalar.c",
Marat Dukhan3a77ea72019-12-23 12:10:24 -0800180 "src/f32-sigmoid/gen/scalar-lut2048-p1-div-x1.c",
181 "src/f32-sigmoid/gen/scalar-lut2048-p1-div-x2.c",
182 "src/f32-sigmoid/gen/scalar-lut2048-p1-div-x4.c",
183 "src/f32-sigmoid/gen/scalar-lut64-p2-div-x1.c",
184 "src/f32-sigmoid/gen/scalar-lut64-p2-div-x2.c",
185 "src/f32-sigmoid/gen/scalar-lut64-p2-div-x4.c",
186 "src/f32-sigmoid/gen/scalar-p5-div-x1.c",
187 "src/f32-sigmoid/gen/scalar-p5-div-x2.c",
188 "src/f32-sigmoid/gen/scalar-p5-div-x4.c",
Marat Dukhan355ab432020-04-09 19:01:52 -0700189 "src/f32-spmm/gen/1x1-minmax-scalar-pipelined.c",
190 "src/f32-spmm/gen/1x1-minmax-scalar.c",
191 "src/f32-spmm/gen/2x1-minmax-scalar-pipelined.c",
192 "src/f32-spmm/gen/2x1-minmax-scalar.c",
193 "src/f32-spmm/gen/4x1-minmax-scalar-pipelined.c",
194 "src/f32-spmm/gen/4x1-minmax-scalar.c",
195 "src/f32-spmm/gen/8x1-minmax-scalar-pipelined.c",
196 "src/f32-spmm/gen/8x1-minmax-scalar.c",
197 "src/f32-spmm/gen/8x2-minmax-scalar.c",
198 "src/f32-spmm/gen/8x4-minmax-scalar.c",
Marat Dukhan91cd2b72020-04-09 23:57:31 -0700199 "src/f32-vbinary/gen/vadd-minmax-scalar-x1.c",
200 "src/f32-vbinary/gen/vadd-minmax-scalar-x2.c",
201 "src/f32-vbinary/gen/vadd-minmax-scalar-x4.c",
202 "src/f32-vbinary/gen/vaddc-minmax-scalar-x1.c",
203 "src/f32-vbinary/gen/vaddc-minmax-scalar-x2.c",
204 "src/f32-vbinary/gen/vaddc-minmax-scalar-x4.c",
205 "src/f32-vbinary/gen/vdiv-minmax-scalar-x1.c",
206 "src/f32-vbinary/gen/vdiv-minmax-scalar-x2.c",
207 "src/f32-vbinary/gen/vdiv-minmax-scalar-x4.c",
208 "src/f32-vbinary/gen/vdivc-minmax-scalar-x1.c",
209 "src/f32-vbinary/gen/vdivc-minmax-scalar-x2.c",
210 "src/f32-vbinary/gen/vdivc-minmax-scalar-x4.c",
Marat Dukhan403b7d42019-12-05 12:49:11 -0800211 "src/f32-vbinary/gen/vmax-scalar-x1.c",
212 "src/f32-vbinary/gen/vmax-scalar-x2.c",
213 "src/f32-vbinary/gen/vmax-scalar-x4.c",
214 "src/f32-vbinary/gen/vmaxc-scalar-x1.c",
215 "src/f32-vbinary/gen/vmaxc-scalar-x2.c",
216 "src/f32-vbinary/gen/vmaxc-scalar-x4.c",
217 "src/f32-vbinary/gen/vmin-scalar-x1.c",
218 "src/f32-vbinary/gen/vmin-scalar-x2.c",
219 "src/f32-vbinary/gen/vmin-scalar-x4.c",
220 "src/f32-vbinary/gen/vminc-scalar-x1.c",
221 "src/f32-vbinary/gen/vminc-scalar-x2.c",
222 "src/f32-vbinary/gen/vminc-scalar-x4.c",
Marat Dukhan91cd2b72020-04-09 23:57:31 -0700223 "src/f32-vbinary/gen/vmul-minmax-scalar-x1.c",
224 "src/f32-vbinary/gen/vmul-minmax-scalar-x2.c",
225 "src/f32-vbinary/gen/vmul-minmax-scalar-x4.c",
226 "src/f32-vbinary/gen/vmulc-minmax-scalar-x1.c",
227 "src/f32-vbinary/gen/vmulc-minmax-scalar-x2.c",
228 "src/f32-vbinary/gen/vmulc-minmax-scalar-x4.c",
229 "src/f32-vbinary/gen/vrdivc-minmax-scalar-x1.c",
230 "src/f32-vbinary/gen/vrdivc-minmax-scalar-x2.c",
231 "src/f32-vbinary/gen/vrdivc-minmax-scalar-x4.c",
232 "src/f32-vbinary/gen/vrsubc-minmax-scalar-x1.c",
233 "src/f32-vbinary/gen/vrsubc-minmax-scalar-x2.c",
234 "src/f32-vbinary/gen/vrsubc-minmax-scalar-x4.c",
235 "src/f32-vbinary/gen/vsub-minmax-scalar-x1.c",
236 "src/f32-vbinary/gen/vsub-minmax-scalar-x2.c",
237 "src/f32-vbinary/gen/vsub-minmax-scalar-x4.c",
238 "src/f32-vbinary/gen/vsubc-minmax-scalar-x1.c",
239 "src/f32-vbinary/gen/vsubc-minmax-scalar-x2.c",
240 "src/f32-vbinary/gen/vsubc-minmax-scalar-x4.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700241 "src/f32-vmulcaddc/gen/c1-minmax-scalar-2x.c",
242 "src/f32-vmulcaddc/gen/c2-minmax-scalar-2x.c",
243 "src/f32-vmulcaddc/gen/c4-minmax-scalar-2x.c",
Marat Dukhan5739f702019-12-22 19:45:09 -0800244 "src/math/expminus-scalar-lut2048-p1.c",
245 "src/math/expminus-scalar-lut64-p2.c",
246 "src/math/expminus-scalar-p5.c",
Marat Dukhan075088a2020-05-12 19:42:12 -0700247 "src/math/roundne-scalar-addsub.c",
Marat Dukhanffbf96a2020-05-14 02:59:08 -0700248 "src/math/roundne-scalar-nearbyint.c",
249 "src/math/roundne-scalar-rint.c",
Marat Dukhanc9852ba2020-05-13 17:21:29 -0700250 "src/math/roundd-scalar-addsub.c",
251 "src/math/roundd-scalar-cvt.c",
Marat Dukhanffbf96a2020-05-14 02:59:08 -0700252 "src/math/roundd-scalar-floor.c",
Marat Dukhanc9852ba2020-05-13 17:21:29 -0700253 "src/math/roundu-scalar-addsub.c",
Marat Dukhanffbf96a2020-05-14 02:59:08 -0700254 "src/math/roundu-scalar-ceil.c",
Marat Dukhanc9852ba2020-05-13 17:21:29 -0700255 "src/math/roundu-scalar-cvt.c",
Marat Dukhan2dbb9442020-05-12 20:43:43 -0700256 "src/math/roundz-scalar-addsub.c",
257 "src/math/roundz-scalar-cvt.c",
Marat Dukhanffbf96a2020-05-14 02:59:08 -0700258 "src/math/roundz-scalar-trunc.c",
Marat Dukhan5739f702019-12-22 19:45:09 -0800259 "src/math/sigmoid-scalar-lut2048-p1-div.c",
260 "src/math/sigmoid-scalar-lut64-p2-div.c",
261 "src/math/sigmoid-scalar-p5-div.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700262 "src/q8-avgpool/9p8x-minmax-scalar-c1.c",
263 "src/q8-avgpool/9x-minmax-scalar-c1.c",
Marat Dukhande06f492020-04-09 00:19:31 -0700264 "src/q8-dwconv/up1x9-minmax-scalar.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700265 "src/q8-gavgpool/7p7x-minmax-scalar-c1.c",
266 "src/q8-gavgpool/7x-minmax-scalar-c1.c",
Marat Dukhande06f492020-04-09 00:19:31 -0700267 "src/q8-gemm/2x2-minmax-scalar.c",
268 "src/q8-igemm/2x2-minmax-scalar.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700269 "src/q8-vadd/minmax-scalar.c",
Marat Dukhan5c5fa962020-03-10 18:38:33 -0700270 "src/u8-clamp/scalar-x4.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -0700271 "src/u8-lut32norm/scalar.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700272 "src/u8-maxpool/9p8x-minmax-scalar-c1.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -0700273 "src/u8-rmax/scalar.c",
Marat Dukhan3bb3bfc2020-05-19 17:42:46 -0700274 "src/x32-fill/scalar-float.c",
275 "src/x32-fill/scalar-int.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -0700276 "src/x32-packx/x2-scalar.c",
277 "src/x32-packx/x3-scalar.c",
278 "src/x32-packx/x4-scalar.c",
Marat Dukhan63523d42020-05-22 17:07:33 -0700279 "src/x32-pad/scalar-int.c",
280 "src/x32-pad/scalar-float.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -0700281 "src/x32-unpool/scalar.c",
282 "src/x32-zip/x2-scalar.c",
283 "src/x32-zip/x3-scalar.c",
284 "src/x32-zip/x4-scalar.c",
285 "src/x32-zip/xm-scalar.c",
286 "src/x8-lut/scalar.c",
287 "src/x8-zip/x2-scalar.c",
288 "src/x8-zip/x3-scalar.c",
289 "src/x8-zip/x4-scalar.c",
290 "src/x8-zip/xm-scalar.c",
Marat Dukhanfe7acb62020-03-09 19:30:05 -0700291 "src/requantization/precise-scalar.c",
292 "src/requantization/fp32-scalar.c",
293 "src/requantization/q31-scalar.c",
294 "src/requantization/gemmlowp-scalar.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -0700295]
296
Marat Dukhan436ebe62019-12-04 15:10:12 -0800297WASM_UKERNELS = [
Marat Dukhan99936602020-04-11 16:47:01 -0700298 "src/f32-avgpool/9p8x-minmax-wasm-c1.c",
299 "src/f32-avgpool/9x-minmax-wasm-c1.c",
Marat Dukhan5c5fa962020-03-10 18:38:33 -0700300 "src/f32-clamp/gen/wasm-x1.c",
301 "src/f32-clamp/gen/wasm-x2.c",
302 "src/f32-clamp/gen/wasm-x4.c",
Marat Dukhan163a7e62020-04-09 04:19:26 -0700303 "src/f32-dwconv/gen/up1x4-wasm-acc2.c",
304 "src/f32-dwconv/gen/up1x4-wasm.c",
305 "src/f32-dwconv/gen/up1x9-wasm-acc2.c",
306 "src/f32-dwconv/gen/up1x9-wasm.c",
307 "src/f32-dwconv/gen/up1x25-wasm-acc2.c",
308 "src/f32-dwconv/gen/up1x25-wasm.c",
309 "src/f32-dwconv/gen/up2x4-wasm-acc2.c",
310 "src/f32-dwconv/gen/up2x4-wasm.c",
311 "src/f32-dwconv/gen/up2x9-wasm-acc2.c",
312 "src/f32-dwconv/gen/up2x9-wasm.c",
313 "src/f32-dwconv/gen/up2x25-wasm-acc2.c",
314 "src/f32-dwconv/gen/up2x25-wasm.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700315 "src/f32-dwconv/gen/up1x4-minmax-wasm-acc2.c",
316 "src/f32-dwconv/gen/up1x4-minmax-wasm.c",
317 "src/f32-dwconv/gen/up1x9-minmax-wasm-acc2.c",
318 "src/f32-dwconv/gen/up1x9-minmax-wasm.c",
Marat Dukhan163a7e62020-04-09 04:19:26 -0700319 "src/f32-dwconv/gen/up1x25-minmax-wasm-acc2.c",
320 "src/f32-dwconv/gen/up1x25-minmax-wasm.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700321 "src/f32-dwconv/gen/up2x4-minmax-wasm-acc2.c",
322 "src/f32-dwconv/gen/up2x4-minmax-wasm.c",
323 "src/f32-dwconv/gen/up2x9-minmax-wasm-acc2.c",
324 "src/f32-dwconv/gen/up2x9-minmax-wasm.c",
Marat Dukhan163a7e62020-04-09 04:19:26 -0700325 "src/f32-dwconv/gen/up2x25-minmax-wasm-acc2.c",
326 "src/f32-dwconv/gen/up2x25-minmax-wasm.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700327 "src/f32-gavgpool/7p7x-minmax-wasm-c1.c",
328 "src/f32-gavgpool/7x-minmax-wasm-c1.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700329 "src/f32-gemm/gen-inc/1x4inc-minmax-wasm.c",
330 "src/f32-gemm/gen-inc/2x4inc-minmax-wasm.c",
331 "src/f32-gemm/gen-inc/4x4inc-minmax-wasm.c",
Marat Dukhan163a7e62020-04-09 04:19:26 -0700332 "src/f32-gemm/gen/1x4-wasm.c",
333 "src/f32-gemm/gen/2x4-wasm.c",
334 "src/f32-gemm/gen/4x2-wasm.c",
335 "src/f32-gemm/gen/4x4-wasm.c",
Marat Dukhan467f6362020-05-22 23:21:55 -0700336 "src/f32-gemm/gen/1x4-relu-wasm.c",
337 "src/f32-gemm/gen/2x4-relu-wasm.c",
338 "src/f32-gemm/gen/4x2-relu-wasm.c",
339 "src/f32-gemm/gen/4x4-relu-wasm.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700340 "src/f32-gemm/gen/1x4-minmax-wasm.c",
341 "src/f32-gemm/gen/2x4-minmax-wasm.c",
342 "src/f32-gemm/gen/4x2-minmax-wasm.c",
343 "src/f32-gemm/gen/4x4-minmax-wasm.c",
Marat Dukhan662faa02019-12-09 22:48:16 -0800344 "src/f32-hswish/gen/wasm-x1.c",
345 "src/f32-hswish/gen/wasm-x2.c",
346 "src/f32-hswish/gen/wasm-x4.c",
Marat Dukhan163a7e62020-04-09 04:19:26 -0700347 "src/f32-igemm/gen/1x4-wasm.c",
348 "src/f32-igemm/gen/2x4-wasm.c",
349 "src/f32-igemm/gen/4x2-wasm.c",
350 "src/f32-igemm/gen/4x4-wasm.c",
Marat Dukhan467f6362020-05-22 23:21:55 -0700351 "src/f32-igemm/gen/1x4-relu-wasm.c",
352 "src/f32-igemm/gen/2x4-relu-wasm.c",
353 "src/f32-igemm/gen/4x2-relu-wasm.c",
354 "src/f32-igemm/gen/4x4-relu-wasm.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700355 "src/f32-igemm/gen/1x4-minmax-wasm.c",
356 "src/f32-igemm/gen/2x4-minmax-wasm.c",
357 "src/f32-igemm/gen/4x2-minmax-wasm.c",
358 "src/f32-igemm/gen/4x4-minmax-wasm.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700359 "src/f32-maxpool/9p8x-minmax-wasm-c1.c",
360 "src/f32-pavgpool/9p8x-minmax-wasm-c1.c",
361 "src/f32-pavgpool/9x-minmax-wasm-c1.c",
Marat Dukhan91cd2b72020-04-09 23:57:31 -0700362 "src/f32-vbinary/gen/vadd-minmax-wasm-x1.c",
363 "src/f32-vbinary/gen/vadd-minmax-wasm-x2.c",
364 "src/f32-vbinary/gen/vadd-minmax-wasm-x4.c",
365 "src/f32-vbinary/gen/vaddc-minmax-wasm-x1.c",
366 "src/f32-vbinary/gen/vaddc-minmax-wasm-x2.c",
367 "src/f32-vbinary/gen/vaddc-minmax-wasm-x4.c",
368 "src/f32-vbinary/gen/vdiv-minmax-wasm-x1.c",
369 "src/f32-vbinary/gen/vdiv-minmax-wasm-x2.c",
370 "src/f32-vbinary/gen/vdiv-minmax-wasm-x4.c",
371 "src/f32-vbinary/gen/vdivc-minmax-wasm-x1.c",
372 "src/f32-vbinary/gen/vdivc-minmax-wasm-x2.c",
373 "src/f32-vbinary/gen/vdivc-minmax-wasm-x4.c",
Marat Dukhan403b7d42019-12-05 12:49:11 -0800374 "src/f32-vbinary/gen/vmax-wasm-x1.c",
375 "src/f32-vbinary/gen/vmax-wasm-x2.c",
376 "src/f32-vbinary/gen/vmax-wasm-x4.c",
377 "src/f32-vbinary/gen/vmaxc-wasm-x1.c",
378 "src/f32-vbinary/gen/vmaxc-wasm-x2.c",
379 "src/f32-vbinary/gen/vmaxc-wasm-x4.c",
380 "src/f32-vbinary/gen/vmin-wasm-x1.c",
381 "src/f32-vbinary/gen/vmin-wasm-x2.c",
382 "src/f32-vbinary/gen/vmin-wasm-x4.c",
383 "src/f32-vbinary/gen/vminc-wasm-x1.c",
384 "src/f32-vbinary/gen/vminc-wasm-x2.c",
385 "src/f32-vbinary/gen/vminc-wasm-x4.c",
Marat Dukhan91cd2b72020-04-09 23:57:31 -0700386 "src/f32-vbinary/gen/vmul-minmax-wasm-x1.c",
387 "src/f32-vbinary/gen/vmul-minmax-wasm-x2.c",
388 "src/f32-vbinary/gen/vmul-minmax-wasm-x4.c",
389 "src/f32-vbinary/gen/vmulc-minmax-wasm-x1.c",
390 "src/f32-vbinary/gen/vmulc-minmax-wasm-x2.c",
391 "src/f32-vbinary/gen/vmulc-minmax-wasm-x4.c",
392 "src/f32-vbinary/gen/vrdivc-minmax-wasm-x1.c",
393 "src/f32-vbinary/gen/vrdivc-minmax-wasm-x2.c",
394 "src/f32-vbinary/gen/vrdivc-minmax-wasm-x4.c",
395 "src/f32-vbinary/gen/vrsubc-minmax-wasm-x1.c",
396 "src/f32-vbinary/gen/vrsubc-minmax-wasm-x2.c",
397 "src/f32-vbinary/gen/vrsubc-minmax-wasm-x4.c",
398 "src/f32-vbinary/gen/vsub-minmax-wasm-x1.c",
399 "src/f32-vbinary/gen/vsub-minmax-wasm-x2.c",
400 "src/f32-vbinary/gen/vsub-minmax-wasm-x4.c",
401 "src/f32-vbinary/gen/vsubc-minmax-wasm-x1.c",
402 "src/f32-vbinary/gen/vsubc-minmax-wasm-x2.c",
403 "src/f32-vbinary/gen/vsubc-minmax-wasm-x4.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700404 "src/f32-vmulcaddc/gen/c1-minmax-wasm-2x.c",
405 "src/f32-vmulcaddc/gen/c2-minmax-wasm-2x.c",
406 "src/f32-vmulcaddc/gen/c4-minmax-wasm-2x.c",
Marat Dukhan436ebe62019-12-04 15:10:12 -0800407]
408
Marat Dukhan8d3c07e2020-01-02 01:20:59 -0800409PSIMD_FASTMATH_UKERNELS = [
Marat Dukhan329da642019-11-19 21:44:39 -0800410 "src/f32-argmaxpool/4x-psimd-c4.c",
Marat Dukhan1e782c42019-11-21 17:02:40 -0800411 "src/f32-argmaxpool/9p8x-psimd-c4.c",
Marat Dukhan329da642019-11-19 21:44:39 -0800412 "src/f32-argmaxpool/9x-psimd-c4.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700413 "src/f32-avgpool/9p8x-minmax-psimd-c4.c",
414 "src/f32-avgpool/9x-minmax-psimd-c4.c",
Marat Dukhan5c5fa962020-03-10 18:38:33 -0700415 "src/f32-clamp/gen/psimd-x4.c",
416 "src/f32-clamp/gen/psimd-x8.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700417 "src/f32-dwconv/gen/up4x25-minmax-psimd-acc2.c",
418 "src/f32-dwconv/gen/up4x25-minmax-psimd.c",
419 "src/f32-dwconv/gen/up4x4-minmax-psimd-acc2.c",
420 "src/f32-dwconv/gen/up4x4-minmax-psimd.c",
421 "src/f32-dwconv/gen/up4x9-minmax-psimd-acc2.c",
422 "src/f32-dwconv/gen/up4x9-minmax-psimd.c",
423 "src/f32-dwconv/gen/up8x25-minmax-psimd-acc2.c",
424 "src/f32-dwconv/gen/up8x25-minmax-psimd.c",
425 "src/f32-dwconv/gen/up8x4-minmax-psimd-acc2.c",
426 "src/f32-dwconv/gen/up8x4-minmax-psimd.c",
427 "src/f32-dwconv/gen/up8x9-minmax-psimd-acc2.c",
428 "src/f32-dwconv/gen/up8x9-minmax-psimd.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700429 "src/f32-gavgpool/7p7x-minmax-psimd-c4.c",
430 "src/f32-gavgpool/7x-minmax-psimd-c4.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700431 "src/f32-gemm/gen/1x8-minmax-psimd-loadsplat.c",
432 "src/f32-gemm/gen/1x8-minmax-psimd-splat.c",
433 "src/f32-gemm/gen/1x8s4-minmax-psimd.c",
434 "src/f32-gemm/gen/4x2c4-minmax-psimd.c",
435 "src/f32-gemm/gen/4x8-minmax-psimd-loadsplat.c",
436 "src/f32-gemm/gen/4x8-minmax-psimd-splat.c",
437 "src/f32-gemm/gen/4x8s4-minmax-psimd.c",
438 "src/f32-gemm/gen/6x8-minmax-psimd-loadsplat.c",
439 "src/f32-gemm/gen/6x8-minmax-psimd-splat.c",
440 "src/f32-gemm/gen/6x8s4-minmax-psimd.c",
441 "src/f32-gemm/gen-inc/1x8inc-minmax-psimd-loadsplat.c",
442 "src/f32-gemm/gen-inc/1x8inc-minmax-psimd-splat.c",
443 "src/f32-gemm/gen-inc/1x8s4inc-minmax-psimd.c",
444 "src/f32-gemm/gen-inc/4x8inc-minmax-psimd-loadsplat.c",
445 "src/f32-gemm/gen-inc/4x8inc-minmax-psimd-splat.c",
446 "src/f32-gemm/gen-inc/4x8s4inc-minmax-psimd.c",
447 "src/f32-gemm/gen-inc/6x8inc-minmax-psimd-loadsplat.c",
448 "src/f32-gemm/gen-inc/6x8inc-minmax-psimd-splat.c",
449 "src/f32-gemm/gen-inc/6x8s4inc-minmax-psimd.c",
Marat Dukhan662faa02019-12-09 22:48:16 -0800450 "src/f32-hswish/gen/psimd-x4.c",
451 "src/f32-hswish/gen/psimd-x8.c",
Marat Dukhan660fd192020-03-10 04:55:30 -0700452 "src/f32-ibilinear/gen/psimd-c4.c",
453 "src/f32-ibilinear/gen/psimd-c8.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700454 "src/f32-igemm/gen/1x8-minmax-psimd-loadsplat.c",
455 "src/f32-igemm/gen/1x8-minmax-psimd-splat.c",
456 "src/f32-igemm/gen/1x8s4-minmax-psimd.c",
457 "src/f32-igemm/gen/4x2c4-minmax-psimd.c",
458 "src/f32-igemm/gen/4x8-minmax-psimd-loadsplat.c",
459 "src/f32-igemm/gen/4x8-minmax-psimd-splat.c",
460 "src/f32-igemm/gen/4x8s4-minmax-psimd.c",
461 "src/f32-igemm/gen/6x8-minmax-psimd-loadsplat.c",
462 "src/f32-igemm/gen/6x8-minmax-psimd-splat.c",
463 "src/f32-igemm/gen/6x8s4-minmax-psimd.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700464 "src/f32-maxpool/9p8x-minmax-psimd-c4.c",
465 "src/f32-pavgpool/9p8x-minmax-psimd-c4.c",
466 "src/f32-pavgpool/9x-minmax-psimd-c4.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700467 "src/f32-ppmm/gen/4x8-minmax-psimd.c",
Marat Dukhan40a672f2019-11-25 03:08:22 -0800468 "src/f32-prelu/gen/psimd-2x4.c",
469 "src/f32-prelu/gen/psimd-2x8.c",
Marat Dukhanb39689d2020-01-24 13:32:20 -0800470 "src/f32-rmax/psimd.c",
Marat Dukhan91cd2b72020-04-09 23:57:31 -0700471 "src/f32-vbinary/gen/vadd-minmax-psimd-x4.c",
472 "src/f32-vbinary/gen/vadd-minmax-psimd-x8.c",
473 "src/f32-vbinary/gen/vaddc-minmax-psimd-x4.c",
474 "src/f32-vbinary/gen/vaddc-minmax-psimd-x8.c",
475 "src/f32-vbinary/gen/vdiv-minmax-psimd-x4.c",
476 "src/f32-vbinary/gen/vdiv-minmax-psimd-x8.c",
477 "src/f32-vbinary/gen/vdivc-minmax-psimd-x4.c",
478 "src/f32-vbinary/gen/vdivc-minmax-psimd-x8.c",
Marat Dukhan403b7d42019-12-05 12:49:11 -0800479 "src/f32-vbinary/gen/vmax-psimd-x4.c",
480 "src/f32-vbinary/gen/vmax-psimd-x8.c",
481 "src/f32-vbinary/gen/vmaxc-psimd-x4.c",
482 "src/f32-vbinary/gen/vmaxc-psimd-x8.c",
483 "src/f32-vbinary/gen/vmin-psimd-x4.c",
484 "src/f32-vbinary/gen/vmin-psimd-x8.c",
485 "src/f32-vbinary/gen/vminc-psimd-x4.c",
486 "src/f32-vbinary/gen/vminc-psimd-x8.c",
Marat Dukhan91cd2b72020-04-09 23:57:31 -0700487 "src/f32-vbinary/gen/vmul-minmax-psimd-x4.c",
488 "src/f32-vbinary/gen/vmul-minmax-psimd-x8.c",
489 "src/f32-vbinary/gen/vmulc-minmax-psimd-x4.c",
490 "src/f32-vbinary/gen/vmulc-minmax-psimd-x8.c",
491 "src/f32-vbinary/gen/vrdivc-minmax-psimd-x4.c",
492 "src/f32-vbinary/gen/vrdivc-minmax-psimd-x8.c",
493 "src/f32-vbinary/gen/vrsubc-minmax-psimd-x4.c",
494 "src/f32-vbinary/gen/vrsubc-minmax-psimd-x8.c",
495 "src/f32-vbinary/gen/vsub-minmax-psimd-x4.c",
496 "src/f32-vbinary/gen/vsub-minmax-psimd-x8.c",
497 "src/f32-vbinary/gen/vsubc-minmax-psimd-x4.c",
498 "src/f32-vbinary/gen/vsubc-minmax-psimd-x8.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700499 "src/f32-vmulcaddc/gen/c4-minmax-psimd-2x.c",
500 "src/f32-vmulcaddc/gen/c8-minmax-psimd-2x.c",
Marat Dukhan3bb3bfc2020-05-19 17:42:46 -0700501 "src/x32-fill/psimd.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -0700502 "src/x32-packx/x4-psimd.c",
Marat Dukhan63523d42020-05-22 17:07:33 -0700503 "src/x32-pad/psimd.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -0700504 "src/x32-unpool/psimd.c",
505 "src/x32-zip/x2-psimd.c",
506 "src/x32-zip/x3-psimd.c",
507 "src/x32-zip/x4-psimd.c",
508 "src/x32-zip/xm-psimd.c",
Marat Dukhanfe7acb62020-03-09 19:30:05 -0700509 "src/requantization/precise-psimd.c",
510 "src/requantization/fp32-psimd.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -0700511]
512
Marat Dukhan8d3c07e2020-01-02 01:20:59 -0800513PSIMD_ACCMATH_UKERNELS = [
Marat Dukhanb39689d2020-01-24 13:32:20 -0800514 "src/f32-raddstoreexpminusmax/gen/psimd-p5-x4.c",
515 "src/f32-raddstoreexpminusmax/gen/psimd-p5-x8.c",
516 "src/f32-raddstoreexpminusmax/gen/psimd-p5-x8-acc2.c",
517 "src/f32-raddstoreexpminusmax/gen/psimd-p5-x12.c",
518 "src/f32-raddstoreexpminusmax/gen/psimd-p5-x12-acc2.c",
519 "src/f32-raddstoreexpminusmax/gen/psimd-p5-x12-acc3.c",
520 "src/f32-raddstoreexpminusmax/gen/psimd-p5-x16.c",
521 "src/f32-raddstoreexpminusmax/gen/psimd-p5-x16-acc2.c",
522 "src/f32-raddstoreexpminusmax/gen/psimd-p5-x16-acc4.c",
523 "src/f32-raddstoreexpminusmax/gen/psimd-p5-x20.c",
524 "src/f32-raddstoreexpminusmax/gen/psimd-p5-x20-acc2.c",
525 "src/f32-raddstoreexpminusmax/gen/psimd-p5-x20-acc5.c",
Marat Dukhan8d3c07e2020-01-02 01:20:59 -0800526 "src/f32-sigmoid/gen/psimd-p5-div-x4.c",
527 "src/f32-sigmoid/gen/psimd-p5-div-x8.c",
528 "src/f32-sigmoid/gen/psimd-p5-div-x12.c",
529 "src/f32-sigmoid/gen/psimd-p5-div-x16.c",
530 "src/f32-sigmoid/gen/psimd-p5-div-x20.c",
531 "src/f32-sigmoid/gen/psimd-p5-div-x24.c",
Marat Dukhan075088a2020-05-12 19:42:12 -0700532 "src/math/roundne-psimd-addsub.c",
Marat Dukhanc9852ba2020-05-13 17:21:29 -0700533 "src/math/roundd-psimd-addsub.c",
534 "src/math/roundu-psimd-addsub.c",
Marat Dukhan2dbb9442020-05-12 20:43:43 -0700535 "src/math/roundz-psimd-addsub.c",
Marat Dukhan8d3c07e2020-01-02 01:20:59 -0800536 "src/math/sigmoid-psimd-p5-div.c",
537]
538
Marat Dukhan08c4a432019-10-03 09:29:21 -0700539# ISA-specific micro-kernels
540NEON_UKERNELS = [
Marat Dukhan99936602020-04-11 16:47:01 -0700541 "src/f32-avgpool/9p8x-minmax-neon-c4.c",
542 "src/f32-avgpool/9x-minmax-neon-c4.c",
Marat Dukhan5c5fa962020-03-10 18:38:33 -0700543 "src/f32-clamp/gen/neon-x4.c",
544 "src/f32-clamp/gen/neon-x8.c",
Marat Dukhan56b10cd2020-05-18 09:35:49 -0700545 "src/f32-conv-hwc/gen/3x3s2p0p1c3x4-neon-2x1.c",
546 "src/f32-conv-hwc/gen/3x3s2p0p1c3x8-neon-2x1.c",
Marat Dukhance7a3f82020-05-17 21:46:44 -0700547 "src/f32-conv-hwc/gen/3x3s2p0p1c3x4-neon-2x2.c",
548 "src/f32-conv-hwc/gen/3x3s2p0p1c3x8-neon-2x2.c",
Marat Dukhan56b10cd2020-05-18 09:35:49 -0700549 "src/f32-conv-hwc/gen/3x3s2p1c3x4-neon-2x1.c",
550 "src/f32-conv-hwc/gen/3x3s2p1c3x8-neon-2x1.c",
Marat Dukhance7a3f82020-05-17 21:46:44 -0700551 "src/f32-conv-hwc/gen/3x3s2p1c3x4-neon-2x2.c",
552 "src/f32-conv-hwc/gen/3x3s2p1c3x8-neon-2x2.c",
Marat Dukhanf5425ea2020-04-24 01:46:00 -0700553 "src/f32-dwconv/gen/up4x4-minmax-neon.c",
554 "src/f32-dwconv/gen/up4x4-minmax-neon-acc2.c",
555 "src/f32-dwconv/gen/up8x4-minmax-neon.c",
556 "src/f32-dwconv/gen/up8x4-minmax-neon-acc2.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700557 "src/f32-dwconv/gen/up4x9-minmax-neon.c",
558 "src/f32-dwconv/gen/up4x9-minmax-neon-acc2.c",
559 "src/f32-dwconv/gen/up8x9-minmax-neon.c",
560 "src/f32-dwconv/gen/up8x9-minmax-neon-acc2.c",
Marat Dukhanf5425ea2020-04-24 01:46:00 -0700561 "src/f32-dwconv/gen/up4x25-minmax-neon.c",
562 "src/f32-dwconv/gen/up4x25-minmax-neon-acc2.c",
563 "src/f32-dwconv/gen/up8x25-minmax-neon.c",
564 "src/f32-dwconv/gen/up8x25-minmax-neon-acc2.c",
Marat Dukhan1f29b802020-05-15 23:46:39 -0700565 "src/f32-gavgpool-cw/neon-x4.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700566 "src/f32-gavgpool/7p7x-minmax-neon-c4.c",
567 "src/f32-gavgpool/7x-minmax-neon-c4.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700568 "src/f32-gemm/gen/1x8-minmax-neon-lane-ld64.c",
569 "src/f32-gemm/gen/4x2-minmax-neon-lane-ld64.c",
570 "src/f32-gemm/gen/4x8-minmax-neon-lane-ld128.c",
571 "src/f32-gemm/gen/4x8-minmax-neon-lane-ld64.c",
572 "src/f32-gemm/gen/5x8-minmax-neon-lane-ld64.c",
573 "src/f32-gemm/gen/6x8-minmax-neon-lane-ld64.c",
574 "src/f32-gemm/gen/6x8-minmax-neon-lane-ld128.c",
575 "src/f32-gemm/gen/1x8-minmax-neon-dup-ld64.c",
576 "src/f32-gemm/gen/4x8-minmax-neon-dup-ld128.c",
577 "src/f32-gemm/gen/4x8-minmax-neon-dup-ld64.c",
578 "src/f32-gemm/gen/6x8-minmax-neon-dup-ld64.c",
579 "src/f32-gemm/gen/6x8-minmax-neon-dup-ld128.c",
580 "src/f32-gemm/gen/1x8s4-minmax-neon.c",
581 "src/f32-gemm/gen/4x8s4-minmax-neon.c",
582 "src/f32-gemm/gen/6x8s4-minmax-neon.c",
583 "src/f32-gemm/gen/8x8s4-minmax-neon.c",
584 "src/f32-gemm/gen-inc/1x8inc-minmax-neon-lane-ld64.c",
585 "src/f32-gemm/gen-inc/4x8inc-minmax-neon-lane-ld128.c",
586 "src/f32-gemm/gen-inc/4x8inc-minmax-neon-lane-ld64.c",
587 "src/f32-gemm/gen-inc/5x8inc-minmax-neon-lane-ld64.c",
588 "src/f32-gemm/gen-inc/6x8inc-minmax-neon-lane-ld64.c",
589 "src/f32-gemm/gen-inc/6x8inc-minmax-neon-lane-ld128.c",
590 "src/f32-gemm/gen-inc/1x8inc-minmax-neon-dup-ld64.c",
591 "src/f32-gemm/gen-inc/4x8inc-minmax-neon-dup-ld128.c",
592 "src/f32-gemm/gen-inc/4x8inc-minmax-neon-dup-ld64.c",
593 "src/f32-gemm/gen-inc/6x8inc-minmax-neon-dup-ld64.c",
594 "src/f32-gemm/gen-inc/6x8inc-minmax-neon-dup-ld128.c",
595 "src/f32-gemm/gen-inc/1x8s4inc-minmax-neon.c",
596 "src/f32-gemm/gen-inc/4x8s4inc-minmax-neon.c",
597 "src/f32-gemm/gen-inc/6x8s4inc-minmax-neon.c",
598 "src/f32-gemm/gen-inc/8x8s4inc-minmax-neon.c",
Marat Dukhan662faa02019-12-09 22:48:16 -0800599 "src/f32-hswish/gen/neon-x4.c",
600 "src/f32-hswish/gen/neon-x8.c",
Marat Dukhan660fd192020-03-10 04:55:30 -0700601 "src/f32-ibilinear/gen/neon-c4.c",
602 "src/f32-ibilinear/gen/neon-c8.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700603 "src/f32-igemm/gen/1x8-minmax-neon-lane-ld64.c",
604 "src/f32-igemm/gen/4x2-minmax-neon-lane-ld64.c",
605 "src/f32-igemm/gen/4x4-minmax-neon-lane-ld64.c",
606 "src/f32-igemm/gen/4x8-minmax-neon-lane-ld128.c",
607 "src/f32-igemm/gen/4x8-minmax-neon-lane-ld64.c",
608 "src/f32-igemm/gen/6x8-minmax-neon-lane-ld64.c",
609 "src/f32-igemm/gen/6x8-minmax-neon-lane-ld128.c",
610 "src/f32-igemm/gen/1x8-minmax-neon-dup-ld64.c",
611 "src/f32-igemm/gen/4x8-minmax-neon-dup-ld128.c",
612 "src/f32-igemm/gen/4x8-minmax-neon-dup-ld64.c",
613 "src/f32-igemm/gen/6x8-minmax-neon-dup-ld64.c",
614 "src/f32-igemm/gen/6x8-minmax-neon-dup-ld128.c",
615 "src/f32-igemm/gen/1x8s4-minmax-neon.c",
616 "src/f32-igemm/gen/4x8s4-minmax-neon.c",
617 "src/f32-igemm/gen/6x8s4-minmax-neon.c",
618 "src/f32-igemm/gen/8x8s4-minmax-neon.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700619 "src/f32-maxpool/9p8x-minmax-neon-c4.c",
620 "src/f32-pavgpool/9p8x-minmax-neon-c4.c",
621 "src/f32-pavgpool/9x-minmax-neon-c4.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700622 "src/f32-ppmm/gen/4x8-minmax-neon.c",
623 "src/f32-ppmm/gen/8x8-minmax-neon.c",
Marat Dukhan40a672f2019-11-25 03:08:22 -0800624 "src/f32-prelu/gen/neon-2x4.c",
625 "src/f32-prelu/gen/neon-2x8.c",
Marat Dukhan8137e4c2020-01-25 12:56:58 -0800626 "src/f32-raddstoreexpminusmax/gen/neon-p5-x4.c",
627 "src/f32-raddstoreexpminusmax/gen/neon-p5-x8.c",
628 "src/f32-raddstoreexpminusmax/gen/neon-p5-x8-acc2.c",
629 "src/f32-raddstoreexpminusmax/gen/neon-p5-x12.c",
630 "src/f32-raddstoreexpminusmax/gen/neon-p5-x12-acc2.c",
631 "src/f32-raddstoreexpminusmax/gen/neon-p5-x12-acc3.c",
632 "src/f32-raddstoreexpminusmax/gen/neon-p5-x16.c",
633 "src/f32-raddstoreexpminusmax/gen/neon-p5-x16-acc2.c",
634 "src/f32-raddstoreexpminusmax/gen/neon-p5-x16-acc4.c",
635 "src/f32-raddstoreexpminusmax/gen/neon-p5-x20.c",
636 "src/f32-raddstoreexpminusmax/gen/neon-p5-x20-acc2.c",
637 "src/f32-raddstoreexpminusmax/gen/neon-p5-x20-acc5.c",
638 "src/f32-raddstoreexpminusmax/gen/neon-lut64-p2-x4.c",
639 "src/f32-raddstoreexpminusmax/gen/neon-lut64-p2-x8.c",
640 "src/f32-raddstoreexpminusmax/gen/neon-lut64-p2-x8-acc2.c",
641 "src/f32-raddstoreexpminusmax/gen/neon-lut64-p2-x12.c",
642 "src/f32-raddstoreexpminusmax/gen/neon-lut64-p2-x12-acc2.c",
643 "src/f32-raddstoreexpminusmax/gen/neon-lut64-p2-x12-acc3.c",
644 "src/f32-raddstoreexpminusmax/gen/neon-lut64-p2-x16.c",
645 "src/f32-raddstoreexpminusmax/gen/neon-lut64-p2-x16-acc2.c",
646 "src/f32-raddstoreexpminusmax/gen/neon-lut64-p2-x16-acc4.c",
647 "src/f32-raddstoreexpminusmax/gen/neon-lut64-p2-x20.c",
648 "src/f32-raddstoreexpminusmax/gen/neon-lut64-p2-x20-acc2.c",
649 "src/f32-raddstoreexpminusmax/gen/neon-lut64-p2-x20-acc5.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -0700650 "src/f32-rmax/neon.c",
Marat Dukhan40a672f2019-11-25 03:08:22 -0800651 "src/f32-sigmoid/gen/neon-frac-p9-p10-nr1recps-x16.c",
Marat Dukhan4a24a582020-01-06 13:30:00 -0800652 "src/f32-sigmoid/gen/neon-rr2-p5-nr2recps-x4.c",
653 "src/f32-sigmoid/gen/neon-rr2-p5-nr2recps-x8.c",
654 "src/f32-sigmoid/gen/neon-rr2-p5-nr2recps-x12.c",
655 "src/f32-sigmoid/gen/neon-rr2-p5-nr2recps-x16.c",
656 "src/f32-sigmoid/gen/neon-rr2-p5-nr2recps-x20.c",
657 "src/f32-sigmoid/gen/neon-rr2-p5-nr2recps-x24.c",
658 "src/f32-sigmoid/gen/neon-rr2-lut64-p2-nr2recps-x4.c",
659 "src/f32-sigmoid/gen/neon-rr2-lut64-p2-nr2recps-x8.c",
660 "src/f32-sigmoid/gen/neon-rr2-lut64-p2-nr2recps-x12.c",
661 "src/f32-sigmoid/gen/neon-rr2-lut64-p2-nr2recps-x16.c",
662 "src/f32-sigmoid/gen/neon-rr2-lut64-p2-nr2recps-x20.c",
663 "src/f32-sigmoid/gen/neon-rr2-lut64-p2-nr2recps-x24.c",
664 "src/f32-sigmoid/gen/neon-rr2-lut2048-p1-nr2recps-x4.c",
665 "src/f32-sigmoid/gen/neon-rr2-lut2048-p1-nr2recps-x8.c",
666 "src/f32-sigmoid/gen/neon-rr2-lut2048-p1-nr2recps-x12.c",
667 "src/f32-sigmoid/gen/neon-rr2-lut2048-p1-nr2recps-x16.c",
668 "src/f32-sigmoid/gen/neon-rr2-lut2048-p1-nr2recps-x20.c",
669 "src/f32-sigmoid/gen/neon-rr2-lut2048-p1-nr2recps-x24.c",
Marat Dukhan91cd2b72020-04-09 23:57:31 -0700670 "src/f32-vbinary/gen/vadd-minmax-neon-x4.c",
671 "src/f32-vbinary/gen/vadd-minmax-neon-x8.c",
672 "src/f32-vbinary/gen/vaddc-minmax-neon-x4.c",
673 "src/f32-vbinary/gen/vaddc-minmax-neon-x8.c",
Marat Dukhan403b7d42019-12-05 12:49:11 -0800674 "src/f32-vbinary/gen/vmax-neon-x4.c",
675 "src/f32-vbinary/gen/vmax-neon-x8.c",
676 "src/f32-vbinary/gen/vmaxc-neon-x4.c",
677 "src/f32-vbinary/gen/vmaxc-neon-x8.c",
678 "src/f32-vbinary/gen/vmin-neon-x4.c",
679 "src/f32-vbinary/gen/vmin-neon-x8.c",
680 "src/f32-vbinary/gen/vminc-neon-x4.c",
681 "src/f32-vbinary/gen/vminc-neon-x8.c",
Marat Dukhan91cd2b72020-04-09 23:57:31 -0700682 "src/f32-vbinary/gen/vmul-minmax-neon-x4.c",
683 "src/f32-vbinary/gen/vmul-minmax-neon-x8.c",
684 "src/f32-vbinary/gen/vmulc-minmax-neon-x4.c",
685 "src/f32-vbinary/gen/vmulc-minmax-neon-x8.c",
686 "src/f32-vbinary/gen/vrsubc-minmax-neon-x4.c",
687 "src/f32-vbinary/gen/vrsubc-minmax-neon-x8.c",
688 "src/f32-vbinary/gen/vsub-minmax-neon-x4.c",
689 "src/f32-vbinary/gen/vsub-minmax-neon-x8.c",
690 "src/f32-vbinary/gen/vsubc-minmax-neon-x4.c",
691 "src/f32-vbinary/gen/vsubc-minmax-neon-x8.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700692 "src/f32-vmulcaddc/gen/c4-minmax-neon-2x.c",
693 "src/f32-vmulcaddc/gen/c8-minmax-neon-2x.c",
694 "src/q8-avgpool/9p8x-minmax-neon-c8.c",
695 "src/q8-avgpool/9x-minmax-neon-c8.c",
Marat Dukhande06f492020-04-09 00:19:31 -0700696 "src/q8-dwconv/up8x9-minmax-neon.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700697 "src/q8-gavgpool/7p7x-minmax-neon-c8.c",
698 "src/q8-gavgpool/7x-minmax-neon-c8.c",
Marat Dukhande06f492020-04-09 00:19:31 -0700699 "src/q8-gemm/4x8-minmax-neon.c",
700 "src/q8-gemm/8x8-minmax-neon.c",
701 "src/q8-igemm/4x8-minmax-neon.c",
702 "src/q8-igemm/8x8-minmax-neon.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700703 "src/q8-vadd/minmax-neon.c",
Marat Dukhan5c5fa962020-03-10 18:38:33 -0700704 "src/u8-clamp/neon-x64.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700705 "src/u8-maxpool/9p8x-minmax-neon-c16.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -0700706 "src/u8-rmax/neon.c",
Marat Dukhan3bb3bfc2020-05-19 17:42:46 -0700707 "src/x32-fill/neon.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -0700708 "src/x32-packx/x4-neon-st4.c",
Marat Dukhan63523d42020-05-22 17:07:33 -0700709 "src/x32-pad/neon.c",
Marat Dukhan57dccd82020-04-14 00:53:10 -0700710 "src/x32-unpool/neon.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -0700711 "src/x32-zip/x2-neon.c",
712 "src/x32-zip/x3-neon.c",
713 "src/x32-zip/x4-neon.c",
714 "src/x32-zip/xm-neon.c",
715 "src/x8-zip/x2-neon.c",
716 "src/x8-zip/x3-neon.c",
717 "src/x8-zip/x4-neon.c",
718 "src/x8-zip/xm-neon.c",
Marat Dukhan075088a2020-05-12 19:42:12 -0700719 "src/math/roundne-neon-addsub.c",
Marat Dukhanc9852ba2020-05-13 17:21:29 -0700720 "src/math/roundd-neon-addsub.c",
721 "src/math/roundd-neon-cvt.c",
722 "src/math/roundu-neon-addsub.c",
723 "src/math/roundu-neon-cvt.c",
Marat Dukhan2dbb9442020-05-12 20:43:43 -0700724 "src/math/roundz-neon-addsub.c",
725 "src/math/roundz-neon-cvt.c",
Marat Dukhan68b3b452020-01-02 10:11:15 -0800726 "src/math/sigmoid-neon-frac-p9-p10-nr1recps.c",
Marat Dukhan77221d32020-01-06 10:04:39 -0800727 "src/math/sigmoid-neon-rr1-lut2048-p1-nr2recps.c",
728 "src/math/sigmoid-neon-rr1-lut64-p2-nr2recps.c",
729 "src/math/sigmoid-neon-rr1-p5-nr2recps.c",
730 "src/math/sigmoid-neon-rr2-lut2048-p1-nr2recps.c",
731 "src/math/sigmoid-neon-rr2-lut64-p2-nr2recps.c",
732 "src/math/sigmoid-neon-rr2-p5-nr2recps.c",
Marat Dukhanfe7acb62020-03-09 19:30:05 -0700733 "src/requantization/precise-neon.c",
734 "src/requantization/fp32-neon.c",
735 "src/requantization/q31-neon.c",
736 "src/requantization/gemmlowp-neon.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -0700737]
738
739NEONFMA_UKERNELS = [
Marat Dukhan660fd192020-03-10 04:55:30 -0700740 "src/f32-ibilinear/gen/neonfma-c4.c",
741 "src/f32-ibilinear/gen/neonfma-c8.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700742 "src/f32-igemm/gen/1x8-minmax-neonfma-dup-ld64.c",
743 "src/f32-igemm/gen/4x8-minmax-neonfma-dup-ld128.c",
744 "src/f32-igemm/gen/4x8-minmax-neonfma-dup-ld64.c",
745 "src/f32-igemm/gen/6x8-minmax-neonfma-dup-ld64.c",
746 "src/f32-igemm/gen/6x8-minmax-neonfma-dup-ld128.c",
747 "src/f32-igemm/gen/1x8s4-minmax-neonfma.c",
748 "src/f32-igemm/gen/4x8s4-minmax-neonfma.c",
749 "src/f32-igemm/gen/6x8s4-minmax-neonfma.c",
750 "src/f32-igemm/gen/8x8s4-minmax-neonfma.c",
Marat Dukhanf5425ea2020-04-24 01:46:00 -0700751 "src/f32-dwconv/gen/up4x4-minmax-neonfma.c",
752 "src/f32-dwconv/gen/up4x4-minmax-neonfma-acc2.c",
753 "src/f32-dwconv/gen/up8x4-minmax-neonfma.c",
754 "src/f32-dwconv/gen/up8x4-minmax-neonfma-acc2.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700755 "src/f32-dwconv/gen/up4x9-minmax-neonfma.c",
756 "src/f32-dwconv/gen/up4x9-minmax-neonfma-acc2.c",
757 "src/f32-dwconv/gen/up8x9-minmax-neonfma.c",
758 "src/f32-dwconv/gen/up8x9-minmax-neonfma-acc2.c",
Marat Dukhanf5425ea2020-04-24 01:46:00 -0700759 "src/f32-dwconv/gen/up4x25-minmax-neonfma.c",
760 "src/f32-dwconv/gen/up4x25-minmax-neonfma-acc2.c",
761 "src/f32-dwconv/gen/up8x25-minmax-neonfma.c",
762 "src/f32-dwconv/gen/up8x25-minmax-neonfma-acc2.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700763 "src/f32-gemm/gen/1x8-minmax-neonfma-dup-ld64.c",
764 "src/f32-gemm/gen/4x8-minmax-neonfma-dup-ld128.c",
765 "src/f32-gemm/gen/4x8-minmax-neonfma-dup-ld64.c",
766 "src/f32-gemm/gen/6x8-minmax-neonfma-dup-ld64.c",
767 "src/f32-gemm/gen/6x8-minmax-neonfma-dup-ld128.c",
768 "src/f32-gemm/gen/1x8s4-minmax-neonfma.c",
769 "src/f32-gemm/gen/4x8s4-minmax-neonfma.c",
770 "src/f32-gemm/gen/6x8s4-minmax-neonfma.c",
771 "src/f32-gemm/gen/8x8s4-minmax-neonfma.c",
772 "src/f32-gemm/gen-inc/1x8inc-minmax-neonfma-dup-ld64.c",
773 "src/f32-gemm/gen-inc/4x8inc-minmax-neonfma-dup-ld128.c",
774 "src/f32-gemm/gen-inc/4x8inc-minmax-neonfma-dup-ld64.c",
775 "src/f32-gemm/gen-inc/6x8inc-minmax-neonfma-dup-ld64.c",
776 "src/f32-gemm/gen-inc/6x8inc-minmax-neonfma-dup-ld128.c",
777 "src/f32-gemm/gen-inc/1x8s4inc-minmax-neonfma.c",
778 "src/f32-gemm/gen-inc/4x8s4inc-minmax-neonfma.c",
779 "src/f32-gemm/gen-inc/6x8s4inc-minmax-neonfma.c",
780 "src/f32-gemm/gen-inc/8x8s4inc-minmax-neonfma.c",
Marat Dukhan662faa02019-12-09 22:48:16 -0800781 "src/f32-hswish/gen/neonfma-x4.c",
782 "src/f32-hswish/gen/neonfma-x8.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700783 "src/f32-ppmm/gen/4x8-minmax-neonfma.c",
784 "src/f32-ppmm/gen/8x8-minmax-neonfma.c",
Marat Dukhan8137e4c2020-01-25 12:56:58 -0800785 "src/f32-raddstoreexpminusmax/gen/neonfma-p5-x4.c",
786 "src/f32-raddstoreexpminusmax/gen/neonfma-p5-x8.c",
787 "src/f32-raddstoreexpminusmax/gen/neonfma-p5-x8-acc2.c",
788 "src/f32-raddstoreexpminusmax/gen/neonfma-p5-x12.c",
789 "src/f32-raddstoreexpminusmax/gen/neonfma-p5-x12-acc2.c",
790 "src/f32-raddstoreexpminusmax/gen/neonfma-p5-x12-acc3.c",
791 "src/f32-raddstoreexpminusmax/gen/neonfma-p5-x16.c",
792 "src/f32-raddstoreexpminusmax/gen/neonfma-p5-x16-acc2.c",
793 "src/f32-raddstoreexpminusmax/gen/neonfma-p5-x16-acc4.c",
794 "src/f32-raddstoreexpminusmax/gen/neonfma-p5-x20.c",
795 "src/f32-raddstoreexpminusmax/gen/neonfma-p5-x20-acc2.c",
796 "src/f32-raddstoreexpminusmax/gen/neonfma-p5-x20-acc5.c",
797 "src/f32-raddstoreexpminusmax/gen/neonfma-lut64-p2-x4.c",
798 "src/f32-raddstoreexpminusmax/gen/neonfma-lut64-p2-x8.c",
799 "src/f32-raddstoreexpminusmax/gen/neonfma-lut64-p2-x8-acc2.c",
800 "src/f32-raddstoreexpminusmax/gen/neonfma-lut64-p2-x12.c",
801 "src/f32-raddstoreexpminusmax/gen/neonfma-lut64-p2-x12-acc2.c",
802 "src/f32-raddstoreexpminusmax/gen/neonfma-lut64-p2-x12-acc3.c",
803 "src/f32-raddstoreexpminusmax/gen/neonfma-lut64-p2-x16.c",
804 "src/f32-raddstoreexpminusmax/gen/neonfma-lut64-p2-x16-acc2.c",
805 "src/f32-raddstoreexpminusmax/gen/neonfma-lut64-p2-x16-acc4.c",
806 "src/f32-raddstoreexpminusmax/gen/neonfma-lut64-p2-x20.c",
807 "src/f32-raddstoreexpminusmax/gen/neonfma-lut64-p2-x20-acc2.c",
808 "src/f32-raddstoreexpminusmax/gen/neonfma-lut64-p2-x20-acc5.c",
Marat Dukhan4a24a582020-01-06 13:30:00 -0800809 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr2fma-x4.c",
810 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr2fma-x8.c",
811 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr2fma-x12.c",
812 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr2fma-x16.c",
813 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr2fma-x20.c",
814 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr2fma-x24.c",
815 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr1recps1fma-x4.c",
816 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr1recps1fma-x8.c",
817 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr1recps1fma-x12.c",
818 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr1recps1fma-x16.c",
819 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr1recps1fma-x20.c",
820 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr1recps1fma-x24.c",
821 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr2recps-x4.c",
822 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr2recps-x8.c",
823 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr2recps-x12.c",
824 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr2recps-x16.c",
825 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr2recps-x20.c",
826 "src/f32-sigmoid/gen/neonfma-rr1-p5-nr2recps-x24.c",
827 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr2fma-x4.c",
828 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr2fma-x8.c",
829 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr2fma-x12.c",
830 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr2fma-x16.c",
831 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr2fma-x20.c",
832 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr2fma-x24.c",
833 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr1recps1fma-x4.c",
834 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr1recps1fma-x8.c",
835 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr1recps1fma-x12.c",
836 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr1recps1fma-x16.c",
837 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr1recps1fma-x20.c",
838 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr1recps1fma-x24.c",
839 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr2recps-x4.c",
840 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr2recps-x8.c",
841 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr2recps-x12.c",
842 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr2recps-x16.c",
843 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr2recps-x20.c",
844 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-nr2recps-x24.c",
845 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr2fma-x4.c",
846 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr2fma-x8.c",
847 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr2fma-x12.c",
848 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr2fma-x16.c",
849 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr2fma-x20.c",
850 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr2fma-x24.c",
851 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr1recps1fma-x4.c",
852 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr1recps1fma-x8.c",
853 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr1recps1fma-x12.c",
854 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr1recps1fma-x16.c",
855 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr1recps1fma-x20.c",
856 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr1recps1fma-x24.c",
857 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr2recps-x4.c",
858 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr2recps-x8.c",
859 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr2recps-x12.c",
860 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr2recps-x16.c",
861 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr2recps-x20.c",
862 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-nr2recps-x24.c",
Marat Dukhan99936602020-04-11 16:47:01 -0700863 "src/f32-vmulcaddc/gen/c4-minmax-neonfma-2x.c",
864 "src/f32-vmulcaddc/gen/c8-minmax-neonfma-2x.c",
Marat Dukhan797a8fe2019-11-14 20:21:57 -0800865 "src/math/exp-neonfma-lut64-p2.c",
866 "src/math/exp-neonfma-p5.c",
Marat Dukhan189ae802019-11-26 11:28:44 -0800867 "src/math/expminus-neonfma-lut2048-p1.c",
868 "src/math/expminus-neonfma-lut64-p2.c",
Marat Dukhan346a9e52019-11-15 09:06:30 -0800869 "src/math/expminus-neonfma-p5.c",
Marat Dukhan77221d32020-01-06 10:04:39 -0800870 "src/math/sigmoid-neonfma-rr1-lut2048-p1-nr1recps1fma.c",
871 "src/math/sigmoid-neonfma-rr1-lut2048-p1-nr2fma.c",
872 "src/math/sigmoid-neonfma-rr1-lut2048-p1-nr2recps.c",
873 "src/math/sigmoid-neonfma-rr1-lut64-p2-nr1recps1fma.c",
874 "src/math/sigmoid-neonfma-rr1-lut64-p2-nr2fma.c",
875 "src/math/sigmoid-neonfma-rr1-lut64-p2-nr2recps.c",
876 "src/math/sigmoid-neonfma-rr1-p5-nr1recps1fma.c",
877 "src/math/sigmoid-neonfma-rr1-p5-nr2fma.c",
878 "src/math/sigmoid-neonfma-rr1-p5-nr2recps.c",
879 "src/math/sigmoid-neonfma-rr2-lut2048-p1-nr1recps1fma.c",
880 "src/math/sigmoid-neonfma-rr2-lut2048-p1-nr2fma.c",
881 "src/math/sigmoid-neonfma-rr2-lut2048-p1-nr2recps.c",
882 "src/math/sigmoid-neonfma-rr2-lut64-p2-nr1recps1fma.c",
883 "src/math/sigmoid-neonfma-rr2-lut64-p2-nr2fma.c",
884 "src/math/sigmoid-neonfma-rr2-lut64-p2-nr2recps.c",
885 "src/math/sigmoid-neonfma-rr2-p5-nr1recps1fma.c",
886 "src/math/sigmoid-neonfma-rr2-p5-nr2fma.c",
887 "src/math/sigmoid-neonfma-rr2-p5-nr2recps.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -0700888]
889
890AARCH64_NEONFMA_UKERNELS = [
Marat Dukhan91cd2b72020-04-09 23:57:31 -0700891 "src/f32-vbinary/gen/vdiv-minmax-neon-x4.c",
892 "src/f32-vbinary/gen/vdiv-minmax-neon-x8.c",
893 "src/f32-vbinary/gen/vdivc-minmax-neon-x4.c",
894 "src/f32-vbinary/gen/vdivc-minmax-neon-x8.c",
895 "src/f32-vbinary/gen/vrdivc-minmax-neon-x4.c",
896 "src/f32-vbinary/gen/vrdivc-minmax-neon-x8.c",
Marat Dukhan1c587112020-04-08 20:04:28 -0700897 "src/f32-gemm/gen/1x8-minmax-neonfma-lane-ld64.c",
898 "src/f32-gemm/gen/4x2-minmax-neonfma-lane-ld64.c",
899 "src/f32-gemm/gen/4x8-minmax-neonfma-lane-ld128.c",
900 "src/f32-gemm/gen/4x8-minmax-neonfma-lane-ld64.c",
901 "src/f32-gemm/gen/5x8-minmax-neonfma-lane-ld64.c",
902 "src/f32-gemm/gen/6x8-minmax-neonfma-lane-ld64.c",
903 "src/f32-gemm/gen/6x8-minmax-neonfma-lane-ld128.c",
904 "src/f32-gemm/gen-inc/1x8inc-minmax-neonfma-lane-ld64.c",
905 "src/f32-gemm/gen-inc/4x8inc-minmax-neonfma-lane-ld128.c",
906 "src/f32-gemm/gen-inc/4x8inc-minmax-neonfma-lane-ld64.c",
907 "src/f32-gemm/gen-inc/5x8inc-minmax-neonfma-lane-ld64.c",
908 "src/f32-gemm/gen-inc/6x8inc-minmax-neonfma-lane-ld64.c",
909 "src/f32-gemm/gen-inc/6x8inc-minmax-neonfma-lane-ld128.c",
910 "src/f32-igemm/gen/1x8-minmax-neonfma-lane-ld64.c",
911 "src/f32-igemm/gen/4x2-minmax-neonfma-lane-ld64.c",
912 "src/f32-igemm/gen/4x4-minmax-neonfma-lane-ld64.c",
913 "src/f32-igemm/gen/4x8-minmax-neonfma-lane-ld128.c",
914 "src/f32-igemm/gen/4x8-minmax-neonfma-lane-ld64.c",
915 "src/f32-igemm/gen/6x8-minmax-neonfma-lane-ld64.c",
916 "src/f32-igemm/gen/6x8-minmax-neonfma-lane-ld128.c",
Marat Dukhan56b10cd2020-05-18 09:35:49 -0700917 "src/f32-conv-hwc/gen/3x3s2p0p1c3x4-neonfma-2x1.c",
918 "src/f32-conv-hwc/gen/3x3s2p0p1c3x8-neonfma-2x1.c",
Marat Dukhance7a3f82020-05-17 21:46:44 -0700919 "src/f32-conv-hwc/gen/3x3s2p0p1c3x4-neonfma-2x2.c",
920 "src/f32-conv-hwc/gen/3x3s2p0p1c3x8-neonfma-2x2.c",
Marat Dukhan56b10cd2020-05-18 09:35:49 -0700921 "src/f32-conv-hwc/gen/3x3s2p1c3x4-neonfma-2x1.c",
922 "src/f32-conv-hwc/gen/3x3s2p1c3x8-neonfma-2x1.c",
Marat Dukhance7a3f82020-05-17 21:46:44 -0700923 "src/f32-conv-hwc/gen/3x3s2p1c3x4-neonfma-2x2.c",
924 "src/f32-conv-hwc/gen/3x3s2p1c3x8-neonfma-2x2.c",
Marat Dukhan1f29b802020-05-15 23:46:39 -0700925 "src/f32-conv-hwc2chw/3x3s2p1c3x4-neonfma-2x2.c",
926 "src/f32-dwconv-chw/3x3p1-neonfma.c",
927 "src/f32-dwconv-chw/5x5p2-neonfma.c",
928 "src/f32-dwconv-chw/3x3s2p1-neonfma.c",
929 "src/f32-dwconv-chw/5x5s2p2-neonfma.c",
Marat Dukhan4a24a582020-01-06 13:30:00 -0800930 "src/f32-sigmoid/gen/neonfma-rr1-p5-div-x4.c",
931 "src/f32-sigmoid/gen/neonfma-rr1-p5-div-x8.c",
932 "src/f32-sigmoid/gen/neonfma-rr1-p5-div-x12.c",
933 "src/f32-sigmoid/gen/neonfma-rr1-p5-div-x16.c",
934 "src/f32-sigmoid/gen/neonfma-rr1-p5-div-x20.c",
935 "src/f32-sigmoid/gen/neonfma-rr1-p5-div-x24.c",
936 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-div-x4.c",
937 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-div-x8.c",
938 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-div-x12.c",
939 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-div-x16.c",
940 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-div-x20.c",
941 "src/f32-sigmoid/gen/neonfma-rr1-lut64-p2-div-x24.c",
942 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-div-x4.c",
943 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-div-x8.c",
944 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-div-x12.c",
945 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-div-x16.c",
946 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-div-x20.c",
947 "src/f32-sigmoid/gen/neonfma-rr1-lut2048-p1-div-x24.c",
Marat Dukhan355ab432020-04-09 19:01:52 -0700948 "src/f32-spmm/gen/12x1-minmax-neonfma.c",
949 "src/f32-spmm/gen/12x2-minmax-neonfma.c",
950 "src/f32-spmm/gen/12x4-minmax-neonfma.c",
951 "src/f32-spmm/gen/16x1-minmax-neonfma-pipelined.c",
952 "src/f32-spmm/gen/16x1-minmax-neonfma-unroll2.c",
953 "src/f32-spmm/gen/16x1-minmax-neonfma.c",
954 "src/f32-spmm/gen/16x2-minmax-neonfma.c",
955 "src/f32-spmm/gen/16x4-minmax-neonfma.c",
956 "src/f32-spmm/gen/4x1-minmax-neonfma-pipelined.c",
957 "src/f32-spmm/gen/4x1-minmax-neonfma-unroll2.c",
958 "src/f32-spmm/gen/4x1-minmax-neonfma.c",
959 "src/f32-spmm/gen/4x2-minmax-neonfma.c",
960 "src/f32-spmm/gen/4x4-minmax-neonfma.c",
961 "src/f32-spmm/gen/8x1-minmax-neonfma-pipelined.c",
962 "src/f32-spmm/gen/8x1-minmax-neonfma-unroll2.c",
963 "src/f32-spmm/gen/8x1-minmax-neonfma.c",
964 "src/f32-spmm/gen/8x2-minmax-neonfma.c",
965 "src/f32-spmm/gen/8x4-minmax-neonfma.c",
Marat Dukhan77221d32020-01-06 10:04:39 -0800966 "src/math/sigmoid-neonfma-rr1-lut2048-p1-div.c",
967 "src/math/sigmoid-neonfma-rr1-lut64-p2-div.c",
968 "src/math/sigmoid-neonfma-rr1-p5-div.c",
969 "src/math/sigmoid-neonfma-rr2-lut2048-p1-div.c",
970 "src/math/sigmoid-neonfma-rr2-lut64-p2-div.c",
971 "src/math/sigmoid-neonfma-rr2-p5-div.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -0700972]
973
Marat Dukhan8853b822020-05-07 12:19:01 -0700974NEONV8_UKERNELS = [
975 "src/math/roundne-neonv8.c",
Marat Dukhanc9852ba2020-05-13 17:21:29 -0700976 "src/math/roundd-neonv8.c",
977 "src/math/roundu-neonv8.c",
Marat Dukhan2dbb9442020-05-12 20:43:43 -0700978 "src/math/roundz-neonv8.c",
Marat Dukhan8853b822020-05-07 12:19:01 -0700979]
980
Marat Dukhan08c4a432019-10-03 09:29:21 -0700981AARCH64_NEONFP16ARITH_UKERNELS = [
Frank Barchardb1966592020-05-12 13:47:06 -0700982 "src/f16-clamp/gen/neonfp16arith-x16.c",
983 "src/f16-clamp/gen/neonfp16arith-x8.c",
984 "src/f16-hswish/gen/neonfp16arith-x16.c",
985 "src/f16-hswish/gen/neonfp16arith-x8.c",
986 "src/f16-prelu/gen/neonfp16arith-2x16.c",
987 "src/f16-prelu/gen/neonfp16arith-2x8.c",
Frank Barchardd793f6c2020-05-08 13:37:43 -0700988 "src/f16-vbinary/gen/vaddc-minmax-neonfp16arith-x16.c",
989 "src/f16-vbinary/gen/vaddc-minmax-neonfp16arith-x8.c",
990 "src/f16-vbinary/gen/vadd-minmax-neonfp16arith-x16.c",
991 "src/f16-vbinary/gen/vadd-minmax-neonfp16arith-x8.c",
992 "src/f16-vbinary/gen/vdivc-minmax-neonfp16arith-x16.c",
993 "src/f16-vbinary/gen/vdivc-minmax-neonfp16arith-x8.c",
994 "src/f16-vbinary/gen/vdiv-minmax-neonfp16arith-x16.c",
995 "src/f16-vbinary/gen/vdiv-minmax-neonfp16arith-x8.c",
996 "src/f16-vbinary/gen/vmaxc-neonfp16arith-x16.c",
997 "src/f16-vbinary/gen/vmaxc-neonfp16arith-x8.c",
998 "src/f16-vbinary/gen/vmax-neonfp16arith-x16.c",
999 "src/f16-vbinary/gen/vmax-neonfp16arith-x8.c",
1000 "src/f16-vbinary/gen/vminc-neonfp16arith-x16.c",
1001 "src/f16-vbinary/gen/vminc-neonfp16arith-x8.c",
1002 "src/f16-vbinary/gen/vmin-neonfp16arith-x16.c",
1003 "src/f16-vbinary/gen/vmin-neonfp16arith-x8.c",
1004 "src/f16-vbinary/gen/vmulc-minmax-neonfp16arith-x16.c",
1005 "src/f16-vbinary/gen/vmulc-minmax-neonfp16arith-x8.c",
1006 "src/f16-vbinary/gen/vmul-minmax-neonfp16arith-x16.c",
1007 "src/f16-vbinary/gen/vmul-minmax-neonfp16arith-x8.c",
1008 "src/f16-vbinary/gen/vrdivc-minmax-neonfp16arith-x16.c",
1009 "src/f16-vbinary/gen/vrdivc-minmax-neonfp16arith-x8.c",
1010 "src/f16-vbinary/gen/vrsubc-minmax-neonfp16arith-x16.c",
1011 "src/f16-vbinary/gen/vrsubc-minmax-neonfp16arith-x8.c",
1012 "src/f16-vbinary/gen/vsubc-minmax-neonfp16arith-x16.c",
1013 "src/f16-vbinary/gen/vsubc-minmax-neonfp16arith-x8.c",
1014 "src/f16-vbinary/gen/vsub-minmax-neonfp16arith-x16.c",
1015 "src/f16-vbinary/gen/vsub-minmax-neonfp16arith-x8.c",
Frank Barchard83ea9392020-04-20 20:15:14 -07001016 "src/f16-gemm/gen/1x8-minmax-neonfp16arith-ld64.c",
1017 "src/f16-gemm/gen/4x8-minmax-neonfp16arith-ld64.c",
1018 "src/f16-gemm/gen/6x8-minmax-neonfp16arith-ld64.c",
1019 "src/f16-gemm/gen/8x8-minmax-neonfp16arith-ld64.c",
1020 "src/f16-gemm/gen-inc/1x8inc-minmax-neonfp16arith-ld64.c",
1021 "src/f16-gemm/gen-inc/4x8inc-minmax-neonfp16arith-ld64.c",
1022 "src/f16-gemm/gen-inc/6x8inc-minmax-neonfp16arith-ld64.c",
1023 "src/f16-gemm/gen-inc/8x8inc-minmax-neonfp16arith-ld64.c",
Frank Barchardb0e4fae2020-05-04 15:27:51 -07001024 "src/f16-igemm/gen/1x8-minmax-neonfp16arith-ld64.c",
1025 "src/f16-igemm/gen/4x8-minmax-neonfp16arith-ld64.c",
1026 "src/f16-igemm/gen/6x8-minmax-neonfp16arith-ld64.c",
1027 "src/f16-igemm/gen/8x8-minmax-neonfp16arith-ld64.c",
Frank Barchard3f9f99f2020-05-06 01:12:04 -07001028 "src/f16-gemm/gen/1x16-minmax-neonfp16arith-ld64.c",
1029 "src/f16-gemm/gen/4x16-minmax-neonfp16arith-ld64.c",
1030 "src/f16-gemm/gen/6x16-minmax-neonfp16arith-ld64.c",
1031 "src/f16-gemm/gen/8x16-minmax-neonfp16arith-ld64.c",
1032 "src/f16-gemm/gen-inc/1x16inc-minmax-neonfp16arith-ld64.c",
1033 "src/f16-gemm/gen-inc/4x16inc-minmax-neonfp16arith-ld64.c",
1034 "src/f16-gemm/gen-inc/6x16inc-minmax-neonfp16arith-ld64.c",
1035 "src/f16-gemm/gen-inc/8x16inc-minmax-neonfp16arith-ld64.c",
1036 "src/f16-igemm/gen/1x16-minmax-neonfp16arith-ld64.c",
1037 "src/f16-igemm/gen/4x16-minmax-neonfp16arith-ld64.c",
1038 "src/f16-igemm/gen/6x16-minmax-neonfp16arith-ld64.c",
1039 "src/f16-igemm/gen/8x16-minmax-neonfp16arith-ld64.c",
Marat Dukhan355ab432020-04-09 19:01:52 -07001040 "src/f16-spmm/gen/8x1-minmax-neonfp16arith.c",
1041 "src/f16-spmm/gen/8x1-minmax-neonfp16arith-unroll2.c",
1042 "src/f16-spmm/gen/16x1-minmax-neonfp16arith.c",
1043 "src/f16-spmm/gen/16x1-minmax-neonfp16arith-unroll2.c",
1044 "src/f16-spmm/gen/24x1-minmax-neonfp16arith.c",
1045 "src/f16-spmm/gen/24x1-minmax-neonfp16arith-unroll2.c",
1046 "src/f16-spmm/gen/32x1-minmax-neonfp16arith.c",
1047 "src/f16-spmm/gen/32x1-minmax-neonfp16arith-unroll2.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001048]
1049
1050SSE_UKERNELS = [
Marat Dukhan99936602020-04-11 16:47:01 -07001051 "src/f32-avgpool/9p8x-minmax-sse-c4.c",
1052 "src/f32-avgpool/9x-minmax-sse-c4.c",
Marat Dukhan5c5fa962020-03-10 18:38:33 -07001053 "src/f32-clamp/gen/sse-x4.c",
1054 "src/f32-clamp/gen/sse-x8.c",
Marat Dukhan1f29b802020-05-15 23:46:39 -07001055 "src/f32-dwconv-chw/3x3p1-sse.c",
1056 "src/f32-dwconv-chw/3x3s2p1-sse.c",
Marat Dukhan1c587112020-04-08 20:04:28 -07001057 "src/f32-dwconv/gen/up4x25-minmax-sse-acc2.c",
1058 "src/f32-dwconv/gen/up4x25-minmax-sse.c",
1059 "src/f32-dwconv/gen/up4x4-minmax-sse-acc2.c",
1060 "src/f32-dwconv/gen/up4x4-minmax-sse.c",
1061 "src/f32-dwconv/gen/up4x9-minmax-sse-acc2.c",
1062 "src/f32-dwconv/gen/up4x9-minmax-sse.c",
1063 "src/f32-dwconv/gen/up8x25-minmax-sse-acc2.c",
1064 "src/f32-dwconv/gen/up8x25-minmax-sse.c",
1065 "src/f32-dwconv/gen/up8x4-minmax-sse-acc2.c",
1066 "src/f32-dwconv/gen/up8x4-minmax-sse.c",
1067 "src/f32-dwconv/gen/up8x9-minmax-sse-acc2.c",
1068 "src/f32-dwconv/gen/up8x9-minmax-sse.c",
Marat Dukhan1f29b802020-05-15 23:46:39 -07001069 "src/f32-gavgpool-cw/sse-x4.c",
Marat Dukhan99936602020-04-11 16:47:01 -07001070 "src/f32-gavgpool/7p7x-minmax-sse-c4.c",
1071 "src/f32-gavgpool/7x-minmax-sse-c4.c",
Marat Dukhan1c587112020-04-08 20:04:28 -07001072 "src/f32-gemm/gen/1x8-minmax-sse-dup.c",
1073 "src/f32-gemm/gen/1x8-minmax-sse-load1.c",
1074 "src/f32-gemm/gen/1x8s4-minmax-sse.c",
1075 "src/f32-gemm/gen/4x2c4-minmax-sse.c",
1076 "src/f32-gemm/gen/4x8-minmax-sse-dup.c",
1077 "src/f32-gemm/gen/4x8-minmax-sse-load1.c",
1078 "src/f32-gemm/gen/4x8s4-minmax-sse.c",
1079 "src/f32-gemm/gen-inc/1x8inc-minmax-sse-dup.c",
1080 "src/f32-gemm/gen-inc/1x8inc-minmax-sse-load1.c",
1081 "src/f32-gemm/gen-inc/1x8s4inc-minmax-sse.c",
1082 "src/f32-gemm/gen-inc/4x8inc-minmax-sse-dup.c",
1083 "src/f32-gemm/gen-inc/4x8inc-minmax-sse-load1.c",
1084 "src/f32-gemm/gen-inc/4x8s4inc-minmax-sse.c",
Marat Dukhan662faa02019-12-09 22:48:16 -08001085 "src/f32-hswish/gen/sse-x4.c",
1086 "src/f32-hswish/gen/sse-x8.c",
Marat Dukhan660fd192020-03-10 04:55:30 -07001087 "src/f32-ibilinear/gen/sse-c4.c",
1088 "src/f32-ibilinear/gen/sse-c8.c",
Marat Dukhan1c587112020-04-08 20:04:28 -07001089 "src/f32-igemm/gen/1x8-minmax-sse-dup.c",
1090 "src/f32-igemm/gen/1x8-minmax-sse-load1.c",
1091 "src/f32-igemm/gen/1x8s4-minmax-sse.c",
1092 "src/f32-igemm/gen/4x2c4-minmax-sse.c",
1093 "src/f32-igemm/gen/4x8-minmax-sse-dup.c",
1094 "src/f32-igemm/gen/4x8-minmax-sse-load1.c",
1095 "src/f32-igemm/gen/4x8s4-minmax-sse.c",
Marat Dukhan99936602020-04-11 16:47:01 -07001096 "src/f32-maxpool/9p8x-minmax-sse-c4.c",
1097 "src/f32-pavgpool/9p8x-minmax-sse-c4.c",
1098 "src/f32-pavgpool/9x-minmax-sse-c4.c",
Marat Dukhan1c587112020-04-08 20:04:28 -07001099 "src/f32-ppmm/gen/4x8-minmax-sse.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001100 "src/f32-rmax/sse.c",
Marat Dukhan355ab432020-04-09 19:01:52 -07001101 "src/f32-spmm/gen/4x1-minmax-sse.c",
1102 "src/f32-spmm/gen/8x1-minmax-sse.c",
Marat Dukhan91cd2b72020-04-09 23:57:31 -07001103 "src/f32-vbinary/gen/vadd-minmax-sse-x4.c",
1104 "src/f32-vbinary/gen/vadd-minmax-sse-x8.c",
1105 "src/f32-vbinary/gen/vaddc-minmax-sse-x4.c",
1106 "src/f32-vbinary/gen/vaddc-minmax-sse-x8.c",
1107 "src/f32-vbinary/gen/vdiv-minmax-sse-x4.c",
1108 "src/f32-vbinary/gen/vdiv-minmax-sse-x8.c",
1109 "src/f32-vbinary/gen/vdivc-minmax-sse-x4.c",
1110 "src/f32-vbinary/gen/vdivc-minmax-sse-x8.c",
Marat Dukhan403b7d42019-12-05 12:49:11 -08001111 "src/f32-vbinary/gen/vmax-sse-x4.c",
1112 "src/f32-vbinary/gen/vmax-sse-x8.c",
1113 "src/f32-vbinary/gen/vmaxc-sse-x4.c",
1114 "src/f32-vbinary/gen/vmaxc-sse-x8.c",
1115 "src/f32-vbinary/gen/vmin-sse-x4.c",
1116 "src/f32-vbinary/gen/vmin-sse-x8.c",
1117 "src/f32-vbinary/gen/vminc-sse-x4.c",
1118 "src/f32-vbinary/gen/vminc-sse-x8.c",
Marat Dukhan91cd2b72020-04-09 23:57:31 -07001119 "src/f32-vbinary/gen/vmul-minmax-sse-x4.c",
1120 "src/f32-vbinary/gen/vmul-minmax-sse-x8.c",
1121 "src/f32-vbinary/gen/vmulc-minmax-sse-x4.c",
1122 "src/f32-vbinary/gen/vmulc-minmax-sse-x8.c",
1123 "src/f32-vbinary/gen/vrdivc-minmax-sse-x4.c",
1124 "src/f32-vbinary/gen/vrdivc-minmax-sse-x8.c",
1125 "src/f32-vbinary/gen/vrsubc-minmax-sse-x4.c",
1126 "src/f32-vbinary/gen/vrsubc-minmax-sse-x8.c",
1127 "src/f32-vbinary/gen/vsub-minmax-sse-x4.c",
1128 "src/f32-vbinary/gen/vsub-minmax-sse-x8.c",
1129 "src/f32-vbinary/gen/vsubc-minmax-sse-x4.c",
1130 "src/f32-vbinary/gen/vsubc-minmax-sse-x8.c",
Marat Dukhan99936602020-04-11 16:47:01 -07001131 "src/f32-vmulcaddc/gen/c4-minmax-sse-2x.c",
1132 "src/f32-vmulcaddc/gen/c8-minmax-sse-2x.c",
Marat Dukhan3bb3bfc2020-05-19 17:42:46 -07001133 "src/x32-fill/sse.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001134 "src/x32-packx/x4-sse.c",
Marat Dukhan63523d42020-05-22 17:07:33 -07001135 "src/x32-pad/sse.c",
Marat Dukhan075088a2020-05-12 19:42:12 -07001136 "src/math/roundne-sse-addsub.c",
Marat Dukhanc9852ba2020-05-13 17:21:29 -07001137 "src/math/roundd-sse-addsub.c",
1138 "src/math/roundu-sse-addsub.c",
Marat Dukhan2dbb9442020-05-12 20:43:43 -07001139 "src/math/roundz-sse-addsub.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001140]
1141
1142SSE2_UKERNELS = [
Marat Dukhan329da642019-11-19 21:44:39 -08001143 "src/f32-argmaxpool/9p8x-sse2-c4.c",
1144 "src/f32-argmaxpool/4x-sse2-c4.c",
1145 "src/f32-argmaxpool/9x-sse2-c4.c",
Marat Dukhan40a672f2019-11-25 03:08:22 -08001146 "src/f32-prelu/gen/sse2-2x4.c",
1147 "src/f32-prelu/gen/sse2-2x8.c",
Marat Dukhanb39689d2020-01-24 13:32:20 -08001148 "src/f32-raddstoreexpminusmax/gen/sse2-p5-x4.c",
1149 "src/f32-raddstoreexpminusmax/gen/sse2-p5-x8.c",
1150 "src/f32-raddstoreexpminusmax/gen/sse2-p5-x8-acc2.c",
1151 "src/f32-raddstoreexpminusmax/gen/sse2-p5-x12.c",
1152 "src/f32-raddstoreexpminusmax/gen/sse2-p5-x12-acc2.c",
1153 "src/f32-raddstoreexpminusmax/gen/sse2-p5-x12-acc3.c",
1154 "src/f32-raddstoreexpminusmax/gen/sse2-p5-x16.c",
1155 "src/f32-raddstoreexpminusmax/gen/sse2-p5-x16-acc2.c",
1156 "src/f32-raddstoreexpminusmax/gen/sse2-p5-x16-acc4.c",
1157 "src/f32-raddstoreexpminusmax/gen/sse2-p5-x20.c",
1158 "src/f32-raddstoreexpminusmax/gen/sse2-p5-x20-acc2.c",
1159 "src/f32-raddstoreexpminusmax/gen/sse2-p5-x20-acc5.c",
Marat Dukhan8d3c07e2020-01-02 01:20:59 -08001160 "src/f32-sigmoid/gen/sse2-p5-div-x4.c",
Marat Dukhan40a672f2019-11-25 03:08:22 -08001161 "src/f32-sigmoid/gen/sse2-p5-div-x8.c",
Marat Dukhan8d3c07e2020-01-02 01:20:59 -08001162 "src/f32-sigmoid/gen/sse2-p5-div-x12.c",
Marat Dukhan40a672f2019-11-25 03:08:22 -08001163 "src/f32-sigmoid/gen/sse2-p5-div-x16.c",
Marat Dukhan8d3c07e2020-01-02 01:20:59 -08001164 "src/f32-sigmoid/gen/sse2-p5-div-x20.c",
1165 "src/f32-sigmoid/gen/sse2-p5-div-x24.c",
Marat Dukhan99936602020-04-11 16:47:01 -07001166 "src/q8-avgpool/9p8x-minmax-sse2-c8.c",
1167 "src/q8-avgpool/9x-minmax-sse2-c8.c",
Marat Dukhande06f492020-04-09 00:19:31 -07001168 "src/q8-igemm/4x4c2-minmax-sse2.c",
1169 "src/q8-dwconv/up8x9-minmax-sse2.c",
Marat Dukhan99936602020-04-11 16:47:01 -07001170 "src/q8-gavgpool/7p7x-minmax-sse2-c8.c",
1171 "src/q8-gavgpool/7x-minmax-sse2-c8.c",
Marat Dukhande06f492020-04-09 00:19:31 -07001172 "src/q8-gemm/2x4c8-minmax-sse2.c",
1173 "src/q8-gemm/4x4c2-minmax-sse2.c",
Marat Dukhan99936602020-04-11 16:47:01 -07001174 "src/q8-vadd/minmax-sse2.c",
Marat Dukhan5c5fa962020-03-10 18:38:33 -07001175 "src/u8-clamp/sse2-x64.c",
Marat Dukhan99936602020-04-11 16:47:01 -07001176 "src/u8-maxpool/9p8x-minmax-sse2-c16.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001177 "src/u8-rmax/sse2.c",
Marat Dukhan57dccd82020-04-14 00:53:10 -07001178 "src/x32-unpool/sse2.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001179 "src/x32-zip/x2-sse2.c",
1180 "src/x32-zip/x3-sse2.c",
1181 "src/x32-zip/x4-sse2.c",
1182 "src/x32-zip/xm-sse2.c",
1183 "src/x8-zip/x2-sse2.c",
1184 "src/x8-zip/x3-sse2.c",
1185 "src/x8-zip/x4-sse2.c",
1186 "src/x8-zip/xm-sse2.c",
Marat Dukhanffd68402019-11-15 15:19:11 -08001187 "src/math/exp-sse2-p5.c",
1188 "src/math/expminus-sse2-p5.c",
Marat Dukhan075088a2020-05-12 19:42:12 -07001189 "src/math/roundne-sse2-cvt.c",
Marat Dukhanc9852ba2020-05-13 17:21:29 -07001190 "src/math/roundd-sse2-cvt.c",
1191 "src/math/roundu-sse2-cvt.c",
Marat Dukhan2dbb9442020-05-12 20:43:43 -07001192 "src/math/roundz-sse2-cvt.c",
Marat Dukhan80bafd22019-11-18 10:16:01 -08001193 "src/math/sigmoid-sse2-p5-div.c",
Marat Dukhanfe7acb62020-03-09 19:30:05 -07001194 "src/requantization/precise-sse2.c",
1195 "src/requantization/fp32-sse2.c",
1196 "src/requantization/q31-sse2.c",
1197 "src/requantization/gemmlowp-sse2.c",
1198]
1199
1200SSSE3_UKERNELS = [
1201 "src/requantization/precise-ssse3.c",
1202 "src/requantization/q31-ssse3.c",
1203 "src/requantization/gemmlowp-ssse3.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001204]
1205
Marat Dukhan69c3f2c2019-11-06 12:30:01 -08001206SSE41_UKERNELS = [
Marat Dukhan40a672f2019-11-25 03:08:22 -08001207 "src/f32-prelu/gen/sse41-2x4.c",
1208 "src/f32-prelu/gen/sse41-2x8.c",
Marat Dukhan8d3c07e2020-01-02 01:20:59 -08001209 "src/f32-sigmoid/gen/sse41-p5-div-x4.c",
Marat Dukhan40a672f2019-11-25 03:08:22 -08001210 "src/f32-sigmoid/gen/sse41-p5-div-x8.c",
Marat Dukhan8d3c07e2020-01-02 01:20:59 -08001211 "src/f32-sigmoid/gen/sse41-p5-div-x12.c",
Marat Dukhan40a672f2019-11-25 03:08:22 -08001212 "src/f32-sigmoid/gen/sse41-p5-div-x16.c",
Marat Dukhan8d3c07e2020-01-02 01:20:59 -08001213 "src/f32-sigmoid/gen/sse41-p5-div-x20.c",
1214 "src/f32-sigmoid/gen/sse41-p5-div-x24.c",
Marat Dukhan8853b822020-05-07 12:19:01 -07001215 "src/math/roundne-sse41.c",
Marat Dukhanc9852ba2020-05-13 17:21:29 -07001216 "src/math/roundd-sse41.c",
1217 "src/math/roundu-sse41.c",
Marat Dukhan2dbb9442020-05-12 20:43:43 -07001218 "src/math/roundz-sse41.c",
Marat Dukhanfe7acb62020-03-09 19:30:05 -07001219 "src/requantization/precise-sse4.c",
1220 "src/requantization/q31-sse4.c",
1221 "src/requantization/gemmlowp-sse4.c",
Marat Dukhan69c3f2c2019-11-06 12:30:01 -08001222]
1223
Marat Dukhan08c4a432019-10-03 09:29:21 -07001224AVX_UKERNELS = [
Marat Dukhan5c5fa962020-03-10 18:38:33 -07001225 "src/f32-clamp/gen/avx-x8.c",
1226 "src/f32-clamp/gen/avx-x16.c",
Marat Dukhan1c587112020-04-08 20:04:28 -07001227 "src/f32-dwconv/gen/up16x4-minmax-avx-acc2.c",
1228 "src/f32-dwconv/gen/up16x4-minmax-avx.c",
1229 "src/f32-dwconv/gen/up8x4-minmax-avx-acc2.c",
1230 "src/f32-dwconv/gen/up8x4-minmax-avx.c",
1231 "src/f32-dwconv/gen/up16x9-minmax-avx-acc2.c",
1232 "src/f32-dwconv/gen/up16x9-minmax-avx.c",
1233 "src/f32-dwconv/gen/up8x9-minmax-avx-acc2.c",
1234 "src/f32-dwconv/gen/up8x9-minmax-avx.c",
1235 "src/f32-dwconv/gen/up16x25-minmax-avx-acc2.c",
1236 "src/f32-dwconv/gen/up16x25-minmax-avx.c",
1237 "src/f32-dwconv/gen/up8x25-minmax-avx-acc2.c",
1238 "src/f32-dwconv/gen/up8x25-minmax-avx.c",
1239 "src/f32-gemm/gen/1x8-minmax-avx-broadcast.c",
1240 "src/f32-gemm/gen/4x8-minmax-avx-broadcast.c",
1241 "src/f32-gemm/gen/5x8-minmax-avx-broadcast.c",
1242 "src/f32-gemm/gen/6x8-minmax-avx-broadcast.c",
1243 "src/f32-gemm/gen/7x8-minmax-avx-broadcast.c",
1244 "src/f32-gemm/gen/1x16-minmax-avx-broadcast.c",
1245 "src/f32-gemm/gen/3x16-minmax-avx-broadcast.c",
1246 "src/f32-gemm/gen/4x16-minmax-avx-broadcast.c",
1247 "src/f32-gemm/gen/5x16-minmax-avx-broadcast.c",
1248 "src/f32-gemm/gen-inc/1x8inc-minmax-avx-broadcast.c",
1249 "src/f32-gemm/gen-inc/4x8inc-minmax-avx-broadcast.c",
1250 "src/f32-gemm/gen-inc/5x8inc-minmax-avx-broadcast.c",
1251 "src/f32-gemm/gen-inc/6x8inc-minmax-avx-broadcast.c",
1252 "src/f32-gemm/gen-inc/7x8inc-minmax-avx-broadcast.c",
1253 "src/f32-gemm/gen-inc/1x16inc-minmax-avx-broadcast.c",
1254 "src/f32-gemm/gen-inc/3x16inc-minmax-avx-broadcast.c",
1255 "src/f32-gemm/gen-inc/4x16inc-minmax-avx-broadcast.c",
1256 "src/f32-gemm/gen-inc/5x16inc-minmax-avx-broadcast.c",
Marat Dukhan662faa02019-12-09 22:48:16 -08001257 "src/f32-hswish/gen/avx-x8.c",
1258 "src/f32-hswish/gen/avx-x16.c",
Marat Dukhan1c587112020-04-08 20:04:28 -07001259 "src/f32-igemm/gen/1x8-minmax-avx-broadcast.c",
1260 "src/f32-igemm/gen/4x8-minmax-avx-broadcast.c",
1261 "src/f32-igemm/gen/5x8-minmax-avx-broadcast.c",
1262 "src/f32-igemm/gen/6x8-minmax-avx-broadcast.c",
1263 "src/f32-igemm/gen/7x8-minmax-avx-broadcast.c",
1264 "src/f32-igemm/gen/1x16-minmax-avx-broadcast.c",
1265 "src/f32-igemm/gen/3x16-minmax-avx-broadcast.c",
1266 "src/f32-igemm/gen/4x16-minmax-avx-broadcast.c",
1267 "src/f32-igemm/gen/5x16-minmax-avx-broadcast.c",
Marat Dukhan90eca0a2020-03-11 00:52:23 -07001268 "src/f32-prelu/gen/avx-2x8.c",
1269 "src/f32-prelu/gen/avx-2x16.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001270 "src/f32-rmax/avx.c",
Marat Dukhan91cd2b72020-04-09 23:57:31 -07001271 "src/f32-vbinary/gen/vadd-minmax-avx-x8.c",
1272 "src/f32-vbinary/gen/vadd-minmax-avx-x16.c",
1273 "src/f32-vbinary/gen/vaddc-minmax-avx-x8.c",
1274 "src/f32-vbinary/gen/vaddc-minmax-avx-x16.c",
1275 "src/f32-vbinary/gen/vdiv-minmax-avx-x8.c",
1276 "src/f32-vbinary/gen/vdiv-minmax-avx-x16.c",
1277 "src/f32-vbinary/gen/vdivc-minmax-avx-x8.c",
1278 "src/f32-vbinary/gen/vdivc-minmax-avx-x16.c",
Marat Dukhan9a88efe2019-12-10 15:54:24 -08001279 "src/f32-vbinary/gen/vmax-avx-x8.c",
1280 "src/f32-vbinary/gen/vmax-avx-x16.c",
1281 "src/f32-vbinary/gen/vmaxc-avx-x8.c",
1282 "src/f32-vbinary/gen/vmaxc-avx-x16.c",
1283 "src/f32-vbinary/gen/vmin-avx-x8.c",
1284 "src/f32-vbinary/gen/vmin-avx-x16.c",
1285 "src/f32-vbinary/gen/vminc-avx-x8.c",
1286 "src/f32-vbinary/gen/vminc-avx-x16.c",
Marat Dukhan91cd2b72020-04-09 23:57:31 -07001287 "src/f32-vbinary/gen/vmul-minmax-avx-x8.c",
1288 "src/f32-vbinary/gen/vmul-minmax-avx-x16.c",
1289 "src/f32-vbinary/gen/vmulc-minmax-avx-x8.c",
1290 "src/f32-vbinary/gen/vmulc-minmax-avx-x16.c",
1291 "src/f32-vbinary/gen/vrdivc-minmax-avx-x8.c",
1292 "src/f32-vbinary/gen/vrdivc-minmax-avx-x16.c",
1293 "src/f32-vbinary/gen/vrsubc-minmax-avx-x8.c",
1294 "src/f32-vbinary/gen/vrsubc-minmax-avx-x16.c",
1295 "src/f32-vbinary/gen/vsub-minmax-avx-x8.c",
1296 "src/f32-vbinary/gen/vsub-minmax-avx-x16.c",
1297 "src/f32-vbinary/gen/vsubc-minmax-avx-x8.c",
1298 "src/f32-vbinary/gen/vsubc-minmax-avx-x16.c",
Marat Dukhan05ac8e32019-10-21 15:39:33 -07001299 "src/f32-vscale/avx-unroll32.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001300]
1301
Marat Dukhanfda12b82019-11-21 12:27:59 -08001302FMA3_UKERNELS = [
Marat Dukhan1c587112020-04-08 20:04:28 -07001303 "src/f32-dwconv/gen/up16x4-minmax-fma3-acc2.c",
1304 "src/f32-dwconv/gen/up16x4-minmax-fma3.c",
1305 "src/f32-dwconv/gen/up8x4-minmax-fma3-acc2.c",
1306 "src/f32-dwconv/gen/up8x4-minmax-fma3.c",
1307 "src/f32-dwconv/gen/up16x9-minmax-fma3-acc2.c",
1308 "src/f32-dwconv/gen/up16x9-minmax-fma3.c",
1309 "src/f32-dwconv/gen/up8x9-minmax-fma3-acc2.c",
1310 "src/f32-dwconv/gen/up8x9-minmax-fma3.c",
1311 "src/f32-dwconv/gen/up16x25-minmax-fma3-acc2.c",
1312 "src/f32-dwconv/gen/up16x25-minmax-fma3.c",
1313 "src/f32-dwconv/gen/up8x25-minmax-fma3-acc2.c",
1314 "src/f32-dwconv/gen/up8x25-minmax-fma3.c",
1315 "src/f32-gemm/gen/1x8-minmax-fma3-broadcast.c",
1316 "src/f32-gemm/gen/4x8-minmax-fma3-broadcast.c",
1317 "src/f32-gemm/gen/5x8-minmax-fma3-broadcast.c",
1318 "src/f32-gemm/gen/6x8-minmax-fma3-broadcast.c",
1319 "src/f32-gemm/gen/7x8-minmax-fma3-broadcast.c",
1320 "src/f32-gemm/gen/8x8-minmax-fma3-broadcast.c",
1321 "src/f32-gemm/gen/1x16-minmax-fma3-broadcast.c",
1322 "src/f32-gemm/gen/3x16-minmax-fma3-broadcast.c",
1323 "src/f32-gemm/gen/4x16-minmax-fma3-broadcast.c",
1324 "src/f32-gemm/gen/5x16-minmax-fma3-broadcast.c",
1325 "src/f32-gemm/gen/1x16s4-minmax-fma3-broadcast.c",
1326 "src/f32-gemm/gen/3x16s4-minmax-fma3-broadcast.c",
1327 "src/f32-gemm/gen/4x16s4-minmax-fma3-broadcast.c",
1328 "src/f32-gemm/gen/5x16s4-minmax-fma3-broadcast.c",
1329 "src/f32-gemm/gen-inc/1x8inc-minmax-fma3-broadcast.c",
1330 "src/f32-gemm/gen-inc/4x8inc-minmax-fma3-broadcast.c",
1331 "src/f32-gemm/gen-inc/5x8inc-minmax-fma3-broadcast.c",
1332 "src/f32-gemm/gen-inc/6x8inc-minmax-fma3-broadcast.c",
1333 "src/f32-gemm/gen-inc/7x8inc-minmax-fma3-broadcast.c",
1334 "src/f32-gemm/gen-inc/8x8inc-minmax-fma3-broadcast.c",
1335 "src/f32-gemm/gen-inc/1x16inc-minmax-fma3-broadcast.c",
1336 "src/f32-gemm/gen-inc/3x16inc-minmax-fma3-broadcast.c",
1337 "src/f32-gemm/gen-inc/4x16inc-minmax-fma3-broadcast.c",
1338 "src/f32-gemm/gen-inc/5x16inc-minmax-fma3-broadcast.c",
1339 "src/f32-gemm/gen-inc/1x16s4inc-minmax-fma3-broadcast.c",
1340 "src/f32-gemm/gen-inc/3x16s4inc-minmax-fma3-broadcast.c",
1341 "src/f32-gemm/gen-inc/4x16s4inc-minmax-fma3-broadcast.c",
1342 "src/f32-gemm/gen-inc/5x16s4inc-minmax-fma3-broadcast.c",
Marat Dukhan662faa02019-12-09 22:48:16 -08001343 "src/f32-hswish/gen/fma3-x8.c",
1344 "src/f32-hswish/gen/fma3-x16.c",
Marat Dukhan1c587112020-04-08 20:04:28 -07001345 "src/f32-igemm/gen/1x8-minmax-fma3-broadcast.c",
1346 "src/f32-igemm/gen/4x8-minmax-fma3-broadcast.c",
1347 "src/f32-igemm/gen/5x8-minmax-fma3-broadcast.c",
1348 "src/f32-igemm/gen/6x8-minmax-fma3-broadcast.c",
1349 "src/f32-igemm/gen/7x8-minmax-fma3-broadcast.c",
1350 "src/f32-igemm/gen/8x8-minmax-fma3-broadcast.c",
1351 "src/f32-igemm/gen/1x16-minmax-fma3-broadcast.c",
1352 "src/f32-igemm/gen/3x16-minmax-fma3-broadcast.c",
1353 "src/f32-igemm/gen/4x16-minmax-fma3-broadcast.c",
1354 "src/f32-igemm/gen/5x16-minmax-fma3-broadcast.c",
1355 "src/f32-igemm/gen/1x16s4-minmax-fma3-broadcast.c",
1356 "src/f32-igemm/gen/3x16s4-minmax-fma3-broadcast.c",
1357 "src/f32-igemm/gen/4x16s4-minmax-fma3-broadcast.c",
1358 "src/f32-igemm/gen/5x16s4-minmax-fma3-broadcast.c",
Marat Dukhanfda12b82019-11-21 12:27:59 -08001359]
1360
Marat Dukhan6adff4e2019-10-14 18:32:07 -07001361AVX2_UKERNELS = [
Marat Dukhan4c4eb002019-12-08 21:27:49 -08001362 "src/f32-raddexpminusmax/gen/avx2-p5-x64.c",
1363 "src/f32-raddexpminusmax/gen/avx2-p5-x64-acc2.c",
1364 "src/f32-raddexpminusmax/gen/avx2-p5-x64-acc4.c",
1365 "src/f32-raddexpminusmax/gen/avx2-p5-x72.c",
1366 "src/f32-raddexpminusmax/gen/avx2-p5-x72-acc3.c",
1367 "src/f32-raddexpminusmax/gen/avx2-p5-x80.c",
1368 "src/f32-raddexpminusmax/gen/avx2-p5-x80-acc2.c",
1369 "src/f32-raddexpminusmax/gen/avx2-p5-x80-acc5.c",
1370 "src/f32-raddexpminusmax/gen/avx2-p5-x96.c",
1371 "src/f32-raddexpminusmax/gen/avx2-p5-x96-acc2.c",
1372 "src/f32-raddexpminusmax/gen/avx2-p5-x96-acc3.c",
1373 "src/f32-raddexpminusmax/gen/avx2-p5-x96-acc6.c",
1374 "src/f32-raddextexp/gen/avx2-p5-x64.c",
1375 "src/f32-raddextexp/gen/avx2-p5-x64-acc2.c",
1376 "src/f32-raddextexp/gen/avx2-p5-x64-acc4.c",
1377 "src/f32-raddextexp/gen/avx2-p5-x72.c",
1378 "src/f32-raddextexp/gen/avx2-p5-x72-acc3.c",
1379 "src/f32-raddextexp/gen/avx2-p5-x80.c",
1380 "src/f32-raddextexp/gen/avx2-p5-x80-acc2.c",
1381 "src/f32-raddextexp/gen/avx2-p5-x80-acc5.c",
1382 "src/f32-raddextexp/gen/avx2-p5-x96.c",
1383 "src/f32-raddextexp/gen/avx2-p5-x96-acc2.c",
1384 "src/f32-raddextexp/gen/avx2-p5-x96-acc3.c",
1385 "src/f32-raddextexp/gen/avx2-p5-x96-acc6.c",
1386 "src/f32-raddstoreexpminusmax/gen/avx2-p5-x64.c",
1387 "src/f32-raddstoreexpminusmax/gen/avx2-p5-x64-acc2.c",
1388 "src/f32-raddstoreexpminusmax/gen/avx2-p5-x64-acc4.c",
1389 "src/f32-raddstoreexpminusmax/gen/avx2-p5-x72.c",
1390 "src/f32-raddstoreexpminusmax/gen/avx2-p5-x72-acc3.c",
1391 "src/f32-raddstoreexpminusmax/gen/avx2-p5-x80.c",
1392 "src/f32-raddstoreexpminusmax/gen/avx2-p5-x80-acc2.c",
1393 "src/f32-raddstoreexpminusmax/gen/avx2-p5-x80-acc5.c",
1394 "src/f32-raddstoreexpminusmax/gen/avx2-p5-x96.c",
1395 "src/f32-raddstoreexpminusmax/gen/avx2-p5-x96-acc2.c",
1396 "src/f32-raddstoreexpminusmax/gen/avx2-p5-x96-acc3.c",
1397 "src/f32-raddstoreexpminusmax/gen/avx2-p5-x96-acc6.c",
Marat Dukhanfa0a4322020-01-06 16:14:29 -08001398 "src/f32-sigmoid/gen/avx2-rr1-p5-div-x8.c",
1399 "src/f32-sigmoid/gen/avx2-rr1-p5-div-x16.c",
1400 "src/f32-sigmoid/gen/avx2-rr1-p5-div-x24.c",
1401 "src/f32-sigmoid/gen/avx2-rr1-p5-div-x32.c",
1402 "src/f32-sigmoid/gen/avx2-rr1-p5-div-x40.c",
1403 "src/f32-sigmoid/gen/avx2-rr1-p5-div-x48.c",
1404 "src/f32-sigmoid/gen/avx2-rr1-p5-div-x56.c",
1405 "src/f32-sigmoid/gen/avx2-rr1-p5-div-x64.c",
1406 "src/f32-sigmoid/gen/avx2-rr1-p5-div-x72.c",
1407 "src/f32-sigmoid/gen/avx2-rr1-p5-div-x80.c",
1408 "src/f32-sigmoid/gen/avx2-rr1-p5-nr1fma-x8.c",
1409 "src/f32-sigmoid/gen/avx2-rr1-p5-nr1fma-x16.c",
1410 "src/f32-sigmoid/gen/avx2-rr1-p5-nr1fma-x24.c",
1411 "src/f32-sigmoid/gen/avx2-rr1-p5-nr1fma-x32.c",
1412 "src/f32-sigmoid/gen/avx2-rr1-p5-nr1fma-x40.c",
1413 "src/f32-sigmoid/gen/avx2-rr1-p5-nr1fma-x48.c",
1414 "src/f32-sigmoid/gen/avx2-rr1-p5-nr1fma-x56.c",
1415 "src/f32-sigmoid/gen/avx2-rr1-p5-nr1fma-x64.c",
1416 "src/f32-sigmoid/gen/avx2-rr1-p5-nr1fma-x72.c",
1417 "src/f32-sigmoid/gen/avx2-rr1-p5-nr1fma-x80.c",
1418 "src/f32-sigmoid/gen/avx2-rr1-p5-nr2fma-x8.c",
1419 "src/f32-sigmoid/gen/avx2-rr1-p5-nr2fma-x16.c",
1420 "src/f32-sigmoid/gen/avx2-rr1-p5-nr2fma-x24.c",
1421 "src/f32-sigmoid/gen/avx2-rr1-p5-nr2fma-x32.c",
1422 "src/f32-sigmoid/gen/avx2-rr1-p5-nr2fma-x40.c",
1423 "src/f32-sigmoid/gen/avx2-rr1-p5-nr2fma-x48.c",
1424 "src/f32-sigmoid/gen/avx2-rr1-p5-nr2fma-x56.c",
1425 "src/f32-sigmoid/gen/avx2-rr1-p5-nr2fma-x64.c",
1426 "src/f32-sigmoid/gen/avx2-rr1-p5-nr2fma-x72.c",
1427 "src/f32-sigmoid/gen/avx2-rr1-p5-nr2fma-x80.c",
Marat Dukhan4c4eb002019-12-08 21:27:49 -08001428 "src/f32-vscaleexpminusmax/gen/avx2-p5-x8.c",
1429 "src/f32-vscaleexpminusmax/gen/avx2-p5-x16.c",
1430 "src/f32-vscaleexpminusmax/gen/avx2-p5-x24.c",
1431 "src/f32-vscaleexpminusmax/gen/avx2-p5-x32.c",
1432 "src/f32-vscaleexpminusmax/gen/avx2-p5-x40.c",
1433 "src/f32-vscaleexpminusmax/gen/avx2-p5-x48.c",
1434 "src/f32-vscaleexpminusmax/gen/avx2-p5-x56.c",
1435 "src/f32-vscaleexpminusmax/gen/avx2-p5-x64.c",
1436 "src/f32-vscaleexpminusmax/gen/avx2-p5-x72.c",
1437 "src/f32-vscaleexpminusmax/gen/avx2-p5-x80.c",
1438 "src/f32-vscaleexpminusmax/gen/avx2-p5-x88.c",
1439 "src/f32-vscaleexpminusmax/gen/avx2-p5-x96.c",
1440 "src/f32-vscaleextexp/gen/avx2-p5-x8.c",
1441 "src/f32-vscaleextexp/gen/avx2-p5-x16.c",
1442 "src/f32-vscaleextexp/gen/avx2-p5-x24.c",
1443 "src/f32-vscaleextexp/gen/avx2-p5-x32.c",
1444 "src/f32-vscaleextexp/gen/avx2-p5-x40.c",
1445 "src/f32-vscaleextexp/gen/avx2-p5-x48.c",
1446 "src/f32-vscaleextexp/gen/avx2-p5-x56.c",
1447 "src/f32-vscaleextexp/gen/avx2-p5-x64.c",
1448 "src/f32-vscaleextexp/gen/avx2-p5-x72.c",
1449 "src/f32-vscaleextexp/gen/avx2-p5-x80.c",
1450 "src/f32-vscaleextexp/gen/avx2-p5-x88.c",
1451 "src/f32-vscaleextexp/gen/avx2-p5-x96.c",
Marat Dukhan6adff4e2019-10-14 18:32:07 -07001452 "src/math/exp-avx2-p5.c",
1453 "src/math/exp-avx2-perm-p3.c",
1454 "src/math/exp-avx2-perm-p4.c",
Marat Dukhan515c9772019-10-17 18:07:57 -07001455 "src/math/expminus-avx2-p5.c",
Marat Dukhan98ba4412019-10-23 02:14:28 -07001456 "src/math/extexp-avx2-p5.c",
Marat Dukhan72416002020-01-05 21:53:19 -08001457 "src/math/sigmoid-avx2-rr2-p5-div.c",
1458 "src/math/sigmoid-avx2-rr1-p5-div.c",
1459 "src/math/sigmoid-avx2-rr2-p5-nr2fma.c",
1460 "src/math/sigmoid-avx2-rr1-p5-nr2fma.c",
1461 "src/math/sigmoid-avx2-rr2-p5-nr1fma.c",
1462 "src/math/sigmoid-avx2-rr1-p5-nr1fma.c",
Marat Dukhan6adff4e2019-10-14 18:32:07 -07001463]
1464
Marat Dukhan08c4a432019-10-03 09:29:21 -07001465AVX512F_UKERNELS = [
Marat Dukhan5c5fa962020-03-10 18:38:33 -07001466 "src/f32-clamp/gen/avx512f-x16.c",
1467 "src/f32-clamp/gen/avx512f-x32.c",
Marat Dukhan1c587112020-04-08 20:04:28 -07001468 "src/f32-dwconv/gen/up32x4-minmax-avx512f-acc2.c",
1469 "src/f32-dwconv/gen/up32x4-minmax-avx512f.c",
1470 "src/f32-dwconv/gen/up16x4-minmax-avx512f-acc2.c",
1471 "src/f32-dwconv/gen/up16x4-minmax-avx512f.c",
1472 "src/f32-dwconv/gen/up32x9-minmax-avx512f-acc2.c",
1473 "src/f32-dwconv/gen/up32x9-minmax-avx512f.c",
1474 "src/f32-dwconv/gen/up16x9-minmax-avx512f-acc2.c",
1475 "src/f32-dwconv/gen/up16x9-minmax-avx512f.c",
1476 "src/f32-dwconv/gen/up32x25-minmax-avx512f-acc2.c",
1477 "src/f32-dwconv/gen/up32x25-minmax-avx512f.c",
1478 "src/f32-dwconv/gen/up16x25-minmax-avx512f-acc2.c",
1479 "src/f32-dwconv/gen/up16x25-minmax-avx512f.c",
1480 "src/f32-gemm/gen/1x16-minmax-avx512f-broadcast.c",
1481 "src/f32-gemm/gen/4x16-minmax-avx512f-broadcast.c",
1482 "src/f32-gemm/gen/5x16-minmax-avx512f-broadcast.c",
1483 "src/f32-gemm/gen/6x16-minmax-avx512f-broadcast.c",
1484 "src/f32-gemm/gen/7x16-minmax-avx512f-broadcast.c",
1485 "src/f32-gemm/gen/8x16-minmax-avx512f-broadcast.c",
1486 "src/f32-gemm/gen-inc/1x16inc-minmax-avx512f-broadcast.c",
1487 "src/f32-gemm/gen-inc/4x16inc-minmax-avx512f-broadcast.c",
1488 "src/f32-gemm/gen-inc/5x16inc-minmax-avx512f-broadcast.c",
1489 "src/f32-gemm/gen-inc/6x16inc-minmax-avx512f-broadcast.c",
1490 "src/f32-gemm/gen-inc/7x16inc-minmax-avx512f-broadcast.c",
1491 "src/f32-gemm/gen-inc/8x16inc-minmax-avx512f-broadcast.c",
Marat Dukhan662faa02019-12-09 22:48:16 -08001492 "src/f32-hswish/gen/avx512f-x16.c",
1493 "src/f32-hswish/gen/avx512f-x32.c",
Marat Dukhan1c587112020-04-08 20:04:28 -07001494 "src/f32-igemm/gen/1x16-minmax-avx512f-broadcast.c",
1495 "src/f32-igemm/gen/4x16-minmax-avx512f-broadcast.c",
1496 "src/f32-igemm/gen/5x16-minmax-avx512f-broadcast.c",
1497 "src/f32-igemm/gen/6x16-minmax-avx512f-broadcast.c",
1498 "src/f32-igemm/gen/7x16-minmax-avx512f-broadcast.c",
1499 "src/f32-igemm/gen/8x16-minmax-avx512f-broadcast.c",
Marat Dukhan90eca0a2020-03-11 00:52:23 -07001500 "src/f32-prelu/gen/avx512f-2x16.c",
1501 "src/f32-prelu/gen/avx512f-2x32.c",
Marat Dukhan4c4eb002019-12-08 21:27:49 -08001502 "src/f32-raddexpminusmax/gen/avx512f-p5-scalef-x128.c",
1503 "src/f32-raddexpminusmax/gen/avx512f-p5-scalef-x128-acc2.c",
1504 "src/f32-raddexpminusmax/gen/avx512f-p5-scalef-x128-acc4.c",
1505 "src/f32-raddexpminusmax/gen/avx512f-p5-scalef-x144.c",
1506 "src/f32-raddexpminusmax/gen/avx512f-p5-scalef-x144-acc3.c",
1507 "src/f32-raddexpminusmax/gen/avx512f-p5-scalef-x160.c",
1508 "src/f32-raddexpminusmax/gen/avx512f-p5-scalef-x160-acc2.c",
1509 "src/f32-raddexpminusmax/gen/avx512f-p5-scalef-x160-acc5.c",
1510 "src/f32-raddexpminusmax/gen/avx512f-p5-scalef-x192.c",
1511 "src/f32-raddexpminusmax/gen/avx512f-p5-scalef-x192-acc2.c",
1512 "src/f32-raddexpminusmax/gen/avx512f-p5-scalef-x192-acc3.c",
1513 "src/f32-raddexpminusmax/gen/avx512f-p5-scalef-x192-acc6.c",
1514 "src/f32-raddextexp/gen/avx512f-p5-scalef-x128.c",
1515 "src/f32-raddextexp/gen/avx512f-p5-scalef-x128-acc2.c",
1516 "src/f32-raddextexp/gen/avx512f-p5-scalef-x128-acc4.c",
1517 "src/f32-raddextexp/gen/avx512f-p5-scalef-x144.c",
1518 "src/f32-raddextexp/gen/avx512f-p5-scalef-x144-acc3.c",
1519 "src/f32-raddextexp/gen/avx512f-p5-scalef-x160.c",
1520 "src/f32-raddextexp/gen/avx512f-p5-scalef-x160-acc2.c",
1521 "src/f32-raddextexp/gen/avx512f-p5-scalef-x160-acc5.c",
1522 "src/f32-raddextexp/gen/avx512f-p5-scalef-x192.c",
1523 "src/f32-raddextexp/gen/avx512f-p5-scalef-x192-acc2.c",
1524 "src/f32-raddextexp/gen/avx512f-p5-scalef-x192-acc3.c",
1525 "src/f32-raddextexp/gen/avx512f-p5-scalef-x192-acc6.c",
1526 "src/f32-raddstoreexpminusmax/gen/avx512f-p5-scalef-x128.c",
1527 "src/f32-raddstoreexpminusmax/gen/avx512f-p5-scalef-x128-acc2.c",
1528 "src/f32-raddstoreexpminusmax/gen/avx512f-p5-scalef-x128-acc4.c",
1529 "src/f32-raddstoreexpminusmax/gen/avx512f-p5-scalef-x144.c",
1530 "src/f32-raddstoreexpminusmax/gen/avx512f-p5-scalef-x144-acc3.c",
1531 "src/f32-raddstoreexpminusmax/gen/avx512f-p5-scalef-x160.c",
1532 "src/f32-raddstoreexpminusmax/gen/avx512f-p5-scalef-x160-acc2.c",
1533 "src/f32-raddstoreexpminusmax/gen/avx512f-p5-scalef-x160-acc5.c",
1534 "src/f32-raddstoreexpminusmax/gen/avx512f-p5-scalef-x192.c",
1535 "src/f32-raddstoreexpminusmax/gen/avx512f-p5-scalef-x192-acc2.c",
1536 "src/f32-raddstoreexpminusmax/gen/avx512f-p5-scalef-x192-acc3.c",
1537 "src/f32-raddstoreexpminusmax/gen/avx512f-p5-scalef-x192-acc6.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001538 "src/f32-rmax/avx512f.c",
Marat Dukhan91cd2b72020-04-09 23:57:31 -07001539 "src/f32-vbinary/gen/vadd-minmax-avx512f-x16.c",
1540 "src/f32-vbinary/gen/vadd-minmax-avx512f-x32.c",
1541 "src/f32-vbinary/gen/vaddc-minmax-avx512f-x16.c",
1542 "src/f32-vbinary/gen/vaddc-minmax-avx512f-x32.c",
1543 "src/f32-vbinary/gen/vdiv-minmax-avx512f-x16.c",
1544 "src/f32-vbinary/gen/vdiv-minmax-avx512f-x32.c",
1545 "src/f32-vbinary/gen/vdivc-minmax-avx512f-x16.c",
1546 "src/f32-vbinary/gen/vdivc-minmax-avx512f-x32.c",
Marat Dukhan9a88efe2019-12-10 15:54:24 -08001547 "src/f32-vbinary/gen/vmax-avx512f-x16.c",
1548 "src/f32-vbinary/gen/vmax-avx512f-x32.c",
1549 "src/f32-vbinary/gen/vmaxc-avx512f-x16.c",
1550 "src/f32-vbinary/gen/vmaxc-avx512f-x32.c",
1551 "src/f32-vbinary/gen/vmin-avx512f-x16.c",
1552 "src/f32-vbinary/gen/vmin-avx512f-x32.c",
1553 "src/f32-vbinary/gen/vminc-avx512f-x16.c",
1554 "src/f32-vbinary/gen/vminc-avx512f-x32.c",
Marat Dukhan91cd2b72020-04-09 23:57:31 -07001555 "src/f32-vbinary/gen/vmul-minmax-avx512f-x16.c",
1556 "src/f32-vbinary/gen/vmul-minmax-avx512f-x32.c",
1557 "src/f32-vbinary/gen/vmulc-minmax-avx512f-x16.c",
1558 "src/f32-vbinary/gen/vmulc-minmax-avx512f-x32.c",
1559 "src/f32-vbinary/gen/vrdivc-minmax-avx512f-x16.c",
1560 "src/f32-vbinary/gen/vrdivc-minmax-avx512f-x32.c",
1561 "src/f32-vbinary/gen/vrsubc-minmax-avx512f-x16.c",
1562 "src/f32-vbinary/gen/vrsubc-minmax-avx512f-x32.c",
1563 "src/f32-vbinary/gen/vsub-minmax-avx512f-x16.c",
1564 "src/f32-vbinary/gen/vsub-minmax-avx512f-x32.c",
1565 "src/f32-vbinary/gen/vsubc-minmax-avx512f-x16.c",
1566 "src/f32-vbinary/gen/vsubc-minmax-avx512f-x32.c",
Marat Dukhan05ac8e32019-10-21 15:39:33 -07001567 "src/f32-vscale/avx512f-unroll64.c",
Marat Dukhan4c4eb002019-12-08 21:27:49 -08001568 "src/f32-vscaleexpminusmax/gen/avx512f-p5-scalef-x16.c",
1569 "src/f32-vscaleexpminusmax/gen/avx512f-p5-scalef-x32.c",
1570 "src/f32-vscaleexpminusmax/gen/avx512f-p5-scalef-x48.c",
1571 "src/f32-vscaleexpminusmax/gen/avx512f-p5-scalef-x64.c",
1572 "src/f32-vscaleexpminusmax/gen/avx512f-p5-scalef-x80.c",
1573 "src/f32-vscaleexpminusmax/gen/avx512f-p5-scalef-x96.c",
1574 "src/f32-vscaleexpminusmax/gen/avx512f-p5-scalef-x112.c",
1575 "src/f32-vscaleexpminusmax/gen/avx512f-p5-scalef-x128.c",
1576 "src/f32-vscaleexpminusmax/gen/avx512f-p5-scalef-x144.c",
1577 "src/f32-vscaleexpminusmax/gen/avx512f-p5-scalef-x160.c",
1578 "src/f32-vscaleexpminusmax/gen/avx512f-p5-scalef-x176.c",
1579 "src/f32-vscaleexpminusmax/gen/avx512f-p5-scalef-x192.c",
1580 "src/f32-vscaleextexp/gen/avx512f-p5-scalef-x16.c",
1581 "src/f32-vscaleextexp/gen/avx512f-p5-scalef-x32.c",
1582 "src/f32-vscaleextexp/gen/avx512f-p5-scalef-x48.c",
1583 "src/f32-vscaleextexp/gen/avx512f-p5-scalef-x64.c",
1584 "src/f32-vscaleextexp/gen/avx512f-p5-scalef-x80.c",
1585 "src/f32-vscaleextexp/gen/avx512f-p5-scalef-x96.c",
1586 "src/f32-vscaleextexp/gen/avx512f-p5-scalef-x112.c",
1587 "src/f32-vscaleextexp/gen/avx512f-p5-scalef-x128.c",
1588 "src/f32-vscaleextexp/gen/avx512f-p5-scalef-x144.c",
1589 "src/f32-vscaleextexp/gen/avx512f-p5-scalef-x160.c",
1590 "src/f32-vscaleextexp/gen/avx512f-p5-scalef-x176.c",
1591 "src/f32-vscaleextexp/gen/avx512f-p5-scalef-x192.c",
Marat Dukhan6adff4e2019-10-14 18:32:07 -07001592 "src/math/exp-avx512f-p5-scalef.c",
1593 "src/math/exp-avx512f-p5.c",
1594 "src/math/exp-avx512f-perm-p3.c",
Marat Dukhanfeb49232019-10-28 11:03:31 -07001595 "src/math/exp-avx512f-perm2-p2.c",
Marat Dukhan98ba4412019-10-23 02:14:28 -07001596 "src/math/extexp-avx512f-p5.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001597]
1598
1599AARCH32_ASM_UKERNELS = [
Marat Dukhan32f93812020-05-17 20:31:21 -07001600 "src/f32-gemm/4x4-aarch32-vfp-ld64.S",
Marat Dukhan3b98f6b2020-05-17 10:09:22 -07001601 "src/f32-gemm/4x4-minmax-aarch32-vfp-ld64.S",
Marat Dukhan1c587112020-04-08 20:04:28 -07001602 "src/f32-gemm/4x8-minmax-aarch32-neon-cortex-a53.S",
1603 "src/f32-gemm/4x8-minmax-aarch32-neon-cortex-a55.S",
1604 "src/f32-gemm/gen/4x8-minmax-aarch32-neon-cortex-a75.S",
1605 "src/f32-gemm/gen/4x8-minmax-aarch32-neon-pld-cortex-a75.S",
1606 "src/f32-gemm/4x8-minmax-aarch32-neon-ld64.S",
1607 "src/f32-igemm/4x8-minmax-aarch32-neon-ld64.S",
1608 "src/f32-igemm/gen/4x8-minmax-aarch32-neon-cortex-a75.S",
1609 "src/f32-igemm/gen/4x8-minmax-aarch32-neon-pld-cortex-a75.S",
1610 "src/f32-igemm/4x8-minmax-aarch32-neon-cortex-a53.S",
1611 "src/f32-igemm/4x8-minmax-aarch32-neon-cortex-a55.S",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001612]
1613
1614AARCH64_ASM_UKERNELS = [
Frank Barchard36b76b62020-04-10 12:39:17 -07001615 "src/f16-gemm/gen/1x16-minmax-aarch64-neonfp16arith-ld32.S",
Frank Barchard683f5592020-04-10 00:48:26 -07001616 "src/f16-gemm/gen/4x16-minmax-aarch64-neonfp16arith-ld32.S",
1617 "src/f16-gemm/gen/6x16-minmax-aarch64-neonfp16arith-ld32.S",
Frank Barchard36b76b62020-04-10 12:39:17 -07001618 "src/f16-gemm/gen-inc/1x16inc-minmax-aarch64-neonfp16arith-ld32.S",
Frank Barchard683f5592020-04-10 00:48:26 -07001619 "src/f16-gemm/gen-inc/4x16inc-minmax-aarch64-neonfp16arith-ld32.S",
1620 "src/f16-gemm/gen-inc/6x16inc-minmax-aarch64-neonfp16arith-ld32.S",
Frank Barchardbddfbcd2020-04-15 12:32:41 -07001621 "src/f16-gemm/gen/1x8-minmax-aarch64-neonfp16arith-ld64.S",
1622 "src/f16-gemm/gen/4x8-minmax-aarch64-neonfp16arith-ld64.S",
1623 "src/f16-gemm/gen/6x8-minmax-aarch64-neonfp16arith-ld64.S",
Frank Barchard3b8e5662020-04-20 12:12:53 -07001624 "src/f16-gemm/gen/8x8-minmax-aarch64-neonfp16arith-ld64.S",
Frank Barchardbddfbcd2020-04-15 12:32:41 -07001625 "src/f16-gemm/gen-inc/1x8inc-minmax-aarch64-neonfp16arith-ld64.S",
1626 "src/f16-gemm/gen-inc/4x8inc-minmax-aarch64-neonfp16arith-ld64.S",
1627 "src/f16-gemm/gen-inc/6x8inc-minmax-aarch64-neonfp16arith-ld64.S",
Frank Barchard3b8e5662020-04-20 12:12:53 -07001628 "src/f16-gemm/gen-inc/8x8inc-minmax-aarch64-neonfp16arith-ld64.S",
Marat Dukhan1c587112020-04-08 20:04:28 -07001629 "src/f32-dwconv/up4x9-minmax-aarch64-neonfma-cortex-a55.S",
1630 "src/f32-dwconv/up4x9-minmax-aarch64-neonfma.S",
Frank Barchard3cb54f92020-04-10 10:46:08 -07001631 "src/f32-gemm/gen/1x8-minmax-aarch64-neonfma-ld64.S",
Marat Dukhan1c587112020-04-08 20:04:28 -07001632 "src/f32-gemm/gen/1x12-minmax-aarch64-neonfma-cortex-a53.S",
1633 "src/f32-gemm/gen/1x8-minmax-aarch64-neonfma-cortex-a53.S",
1634 "src/f32-gemm/gen/1x8-minmax-aarch64-neonfma-cortex-a57.S",
1635 "src/f32-gemm/gen/1x8-minmax-aarch64-neonfma-cortex-a75.S",
1636 "src/f32-gemm/gen/4x12-minmax-aarch64-neonfma-cortex-a53.S",
1637 "src/f32-gemm/gen/4x8-minmax-aarch64-neonfma-cortex-a53.S",
1638 "src/f32-gemm/gen/4x8-minmax-aarch64-neonfma-cortex-a55.S",
1639 "src/f32-gemm/gen/4x8-minmax-aarch64-neonfma-cortex-a57.S",
1640 "src/f32-gemm/gen/4x8-minmax-aarch64-neonfma-cortex-a75.S",
1641 "src/f32-gemm/gen/4x8-minmax-aarch64-neonfma-ld128.S",
1642 "src/f32-gemm/gen/4x8-minmax-aarch64-neonfma-ld64.S",
1643 "src/f32-gemm/gen/5x8-minmax-aarch64-neonfma-cortex-a57.S",
1644 "src/f32-gemm/gen/5x8-minmax-aarch64-neonfma-cortex-a75.S",
1645 "src/f32-gemm/gen/6x8-minmax-aarch64-neonfma-cortex-a53.S",
1646 "src/f32-gemm/gen/6x8-minmax-aarch64-neonfma-cortex-a55.S",
1647 "src/f32-gemm/gen/6x8-minmax-aarch64-neonfma-cortex-a73.S",
1648 "src/f32-gemm/gen/6x8-minmax-aarch64-neonfma-cortex-a57.S",
1649 "src/f32-gemm/gen/6x8-minmax-aarch64-neonfma-cortex-a75.S",
1650 "src/f32-gemm/gen/6x8-minmax-aarch64-neonfma-ios.S",
1651 "src/f32-gemm/gen/6x8-minmax-aarch64-neonfma-ld128.S",
1652 "src/f32-gemm/gen/6x8-minmax-aarch64-neonfma-ld64.S",
Frank Barchard3cb54f92020-04-10 10:46:08 -07001653 "src/f32-gemm/gen-inc/1x8inc-minmax-aarch64-neonfma-ld64.S",
Marat Dukhan1c587112020-04-08 20:04:28 -07001654 "src/f32-gemm/gen-inc/1x12inc-minmax-aarch64-neonfma-cortex-a53.S",
1655 "src/f32-gemm/gen-inc/1x8inc-minmax-aarch64-neonfma-cortex-a53.S",
1656 "src/f32-gemm/gen-inc/1x8inc-minmax-aarch64-neonfma-cortex-a57.S",
1657 "src/f32-gemm/gen-inc/1x8inc-minmax-aarch64-neonfma-cortex-a75.S",
1658 "src/f32-gemm/gen-inc/4x12inc-minmax-aarch64-neonfma-cortex-a53.S",
1659 "src/f32-gemm/gen-inc/4x8inc-minmax-aarch64-neonfma-cortex-a53.S",
1660 "src/f32-gemm/gen-inc/4x8inc-minmax-aarch64-neonfma-cortex-a55.S",
1661 "src/f32-gemm/gen-inc/4x8inc-minmax-aarch64-neonfma-cortex-a57.S",
1662 "src/f32-gemm/gen-inc/4x8inc-minmax-aarch64-neonfma-cortex-a75.S",
1663 "src/f32-gemm/gen-inc/4x8inc-minmax-aarch64-neonfma-ld128.S",
1664 "src/f32-gemm/gen-inc/4x8inc-minmax-aarch64-neonfma-ld64.S",
1665 "src/f32-gemm/gen-inc/5x8inc-minmax-aarch64-neonfma-cortex-a57.S",
1666 "src/f32-gemm/gen-inc/5x8inc-minmax-aarch64-neonfma-cortex-a75.S",
1667 "src/f32-gemm/gen-inc/6x8inc-minmax-aarch64-neonfma-cortex-a53.S",
1668 "src/f32-gemm/gen-inc/6x8inc-minmax-aarch64-neonfma-cortex-a55.S",
1669 "src/f32-gemm/gen-inc/6x8inc-minmax-aarch64-neonfma-cortex-a73.S",
1670 "src/f32-gemm/gen-inc/6x8inc-minmax-aarch64-neonfma-cortex-a57.S",
1671 "src/f32-gemm/gen-inc/6x8inc-minmax-aarch64-neonfma-cortex-a75.S",
1672 "src/f32-gemm/gen-inc/6x8inc-minmax-aarch64-neonfma-ios.S",
1673 "src/f32-gemm/gen-inc/6x8inc-minmax-aarch64-neonfma-ld128.S",
1674 "src/f32-gemm/gen-inc/6x8inc-minmax-aarch64-neonfma-ld64.S",
1675 "src/f32-igemm/1x12-minmax-aarch64-neonfma-cortex-a53.S",
1676 "src/f32-igemm/1x8-minmax-aarch64-neonfma-cortex-a53.S",
1677 "src/f32-igemm/gen/1x8-minmax-aarch64-neonfma-cortex-a57.S",
1678 "src/f32-igemm/gen/1x8-minmax-aarch64-neonfma-cortex-a75.S",
1679 "src/f32-igemm/4x12-minmax-aarch64-neonfma-cortex-a53.S",
1680 "src/f32-igemm/4x8-minmax-aarch64-neonfma-cortex-a53.S",
1681 "src/f32-igemm/4x8-minmax-aarch64-neonfma-cortex-a55.S",
1682 "src/f32-igemm/gen/4x8-minmax-aarch64-neonfma-cortex-a57.S",
1683 "src/f32-igemm/gen/4x8-minmax-aarch64-neonfma-cortex-a75.S",
1684 "src/f32-igemm/gen/5x8-minmax-aarch64-neonfma-cortex-a57.S",
1685 "src/f32-igemm/gen/5x8-minmax-aarch64-neonfma-cortex-a75.S",
1686 "src/f32-igemm/6x8-minmax-aarch64-neonfma-cortex-a53.S",
1687 "src/f32-igemm/6x8-minmax-aarch64-neonfma-cortex-a55.S",
1688 "src/f32-igemm/6x8-minmax-aarch64-neonfma-cortex-a73.S",
1689 "src/f32-igemm/gen/6x8-minmax-aarch64-neonfma-cortex-a57.S",
1690 "src/f32-igemm/gen/6x8-minmax-aarch64-neonfma-cortex-a75.S",
1691 "src/f32-igemm/gen/6x8-minmax-aarch64-neonfma-ios.S",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001692]
1693
Marat Dukhan1b354632020-03-23 12:50:22 -07001694INTERNAL_MICROKERNEL_HDRS = [
Marat Dukhanfe7acb62020-03-09 19:30:05 -07001695 "src/requantization/gemmlowp-requantization.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001696 "src/xnnpack/argmaxpool.h",
1697 "src/xnnpack/avgpool.h",
1698 "src/xnnpack/clamp.h",
1699 "src/xnnpack/common.h",
1700 "src/xnnpack/conv.h",
1701 "src/xnnpack/dwconv.h",
1702 "src/xnnpack/gavgpool.h",
1703 "src/xnnpack/gemm.h",
Marat Dukhan3bb3bfc2020-05-19 17:42:46 -07001704 "src/xnnpack/fill.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001705 "src/xnnpack/hswish.h",
Marat Dukhan660fd192020-03-10 04:55:30 -07001706 "src/xnnpack/ibilinear.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001707 "src/xnnpack/igemm.h",
Marat Dukhancfb31342019-12-05 10:42:57 -08001708 "src/xnnpack/intrinsics-polyfill.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001709 "src/xnnpack/lut.h",
1710 "src/xnnpack/math.h",
1711 "src/xnnpack/maxpool.h",
Marat Dukhan04f03be2019-11-19 12:36:47 -08001712 "src/xnnpack/memory.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001713 "src/xnnpack/packx.h",
1714 "src/xnnpack/pad.h",
1715 "src/xnnpack/params.h",
1716 "src/xnnpack/pavgpool.h",
1717 "src/xnnpack/ppmm.h",
1718 "src/xnnpack/prelu.h",
Marat Dukhan97579532019-10-18 16:40:39 -07001719 "src/xnnpack/raddexpminusmax.h",
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001720 "src/xnnpack/raddextexp.h",
Marat Dukhan97579532019-10-18 16:40:39 -07001721 "src/xnnpack/raddstoreexpminusmax.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001722 "src/xnnpack/rmax.h",
1723 "src/xnnpack/scalar-utils.h",
1724 "src/xnnpack/spmm.h",
1725 "src/xnnpack/unpool.h",
1726 "src/xnnpack/vadd.h",
Marat Dukhan1e782c42019-11-21 17:02:40 -08001727 "src/xnnpack/vbinary.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001728 "src/xnnpack/vmulcaddc.h",
Marat Dukhan05ac8e32019-10-21 15:39:33 -07001729 "src/xnnpack/vscale.h",
Marat Dukhan97579532019-10-18 16:40:39 -07001730 "src/xnnpack/vscaleexpminusmax.h",
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07001731 "src/xnnpack/vscaleextexp.h",
Marat Dukhan1e782c42019-11-21 17:02:40 -08001732 "src/xnnpack/vunary.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001733 "src/xnnpack/zip.h",
Marat Dukhan1b354632020-03-23 12:50:22 -07001734]
1735
1736INTERNAL_HDRS = INTERNAL_MICROKERNEL_HDRS + [
Marat Dukhan08c4a432019-10-03 09:29:21 -07001737 "include/xnnpack.h",
1738 "src/xnnpack/allocator.h",
1739 "src/xnnpack/compute.h",
1740 "src/xnnpack/im2col.h",
1741 "src/xnnpack/indirection.h",
Marat Dukhan6adff4e2019-10-14 18:32:07 -07001742 "src/xnnpack/math-stubs.h",
Chao Mei6ddfc602020-05-13 22:29:36 -07001743 "src/xnnpack/memory-planner.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001744 "src/xnnpack/operator.h",
1745 "src/xnnpack/pack.h",
Marat Dukhaneeaa7bd2019-10-25 17:31:25 -07001746 "src/xnnpack/params-init.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001747 "src/xnnpack/requantization-stubs.h",
Marat Dukhan6adff4e2019-10-14 18:32:07 -07001748 "src/xnnpack/requantization.h",
Marat Dukhan1d75a542020-02-03 12:23:01 -08001749 "src/xnnpack/subgraph.h",
Marat Dukhan6adff4e2019-10-14 18:32:07 -07001750]
1751
Marat Dukhan1b354632020-03-23 12:50:22 -07001752ACCURACY_EVAL_HDRS = INTERNAL_MICROKERNEL_HDRS + [
Marat Dukhan6adff4e2019-10-14 18:32:07 -07001753 "src/xnnpack/math-stubs.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001754]
1755
Marat Dukhan1b354632020-03-23 12:50:22 -07001756MICROKERNEL_BENCHMARK_HDRS = INTERNAL_MICROKERNEL_HDRS + [
Marat Dukhaneeaa7bd2019-10-25 17:31:25 -07001757 "src/xnnpack/params-init.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001758 "include/xnnpack.h",
1759]
1760
Marat Dukhan1b354632020-03-23 12:50:22 -07001761MICROKERNEL_TEST_HDRS = INTERNAL_MICROKERNEL_HDRS + [
Marat Dukhan08c4a432019-10-03 09:29:21 -07001762 "src/xnnpack/isa-checks.h",
Marat Dukhaneeaa7bd2019-10-25 17:31:25 -07001763 "src/xnnpack/params-init.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001764 "src/xnnpack/requantization.h",
1765 "include/xnnpack.h",
1766]
1767
1768OPERATOR_TEST_PARAMS_HDRS = [
1769 "src/xnnpack/params.h",
1770 "src/xnnpack/common.h",
1771]
1772
1773WEIGHTS_PACK_HDRS = [
1774 "src/xnnpack/pack.h",
1775 "src/xnnpack/operator.h",
1776 "src/xnnpack/compute.h",
1777]
1778
Marat Dukhanc8e00eb2019-10-04 14:55:26 -07001779LOGGING_COPTS = select({
1780 # No logging in optimized mode
1781 ":optimized_build": ["-DXNN_LOG_LEVEL=0"],
1782 # Full logging in debug mode
1783 ":debug_build": ["-DXNN_LOG_LEVEL=5"],
1784 # Error-only logging in default (fastbuild) mode
1785 "//conditions:default": ["-DXNN_LOG_LEVEL=2"],
1786})
1787
1788LOGGING_HDRS = [
1789 "src/xnnpack/log.h",
1790]
1791
Marat Dukhan08c4a432019-10-03 09:29:21 -07001792xnnpack_cc_library(
Marat Dukhan3a77ea72019-12-23 12:10:24 -08001793 name = "tables",
1794 srcs = TABLE_SRCS,
1795 hdrs = INTERNAL_HDRS,
Marat Dukhan10a38082020-04-17 03:58:35 -07001796 gcc_copts = xnnpack_gcc_std_copts(),
1797 msvc_copts = xnnpack_msvc_std_copts(),
Marat Dukhan3a77ea72019-12-23 12:10:24 -08001798)
1799
1800xnnpack_cc_library(
Marat Dukhan08c4a432019-10-03 09:29:21 -07001801 name = "scalar_ukernels",
1802 srcs = SCALAR_UKERNELS,
1803 hdrs = INTERNAL_HDRS,
1804 aarch32_copts = ["-marm"],
Marat Dukhan10a38082020-04-17 03:58:35 -07001805 gcc_copts = xnnpack_gcc_std_copts(),
1806 msvc_copts = xnnpack_msvc_std_copts(),
Marat Dukhan08c4a432019-10-03 09:29:21 -07001807 deps = [
Marat Dukhan3a77ea72019-12-23 12:10:24 -08001808 ":tables",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001809 "@FP16",
1810 "@FXdiv",
Marat Dukhan04f03be2019-11-19 12:36:47 -08001811 "@pthreadpool",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001812 ],
1813)
1814
1815xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07001816 name = "scalar_ukernels_test_mode",
1817 srcs = SCALAR_UKERNELS,
1818 hdrs = INTERNAL_HDRS,
1819 aarch32_copts = ["-marm"],
1820 copts = [
1821 "-UNDEBUG",
1822 "-DXNN_TEST_MODE=1",
1823 ],
1824 gcc_copts = xnnpack_gcc_std_copts(),
1825 msvc_copts = xnnpack_msvc_std_copts(),
1826 deps = [
1827 ":tables",
1828 "@FP16",
1829 "@FXdiv",
1830 "@pthreadpool",
1831 ],
1832)
1833
1834xnnpack_cc_library(
Marat Dukhan436ebe62019-12-04 15:10:12 -08001835 name = "wasm_ukernels",
1836 hdrs = INTERNAL_HDRS,
Marat Dukhan10a38082020-04-17 03:58:35 -07001837 gcc_copts = xnnpack_gcc_std_copts(),
1838 msvc_copts = xnnpack_msvc_std_copts(),
Marat Dukhan436ebe62019-12-04 15:10:12 -08001839 wasm_srcs = WASM_UKERNELS,
1840 deps = [
Marat Dukhan3a77ea72019-12-23 12:10:24 -08001841 ":tables",
Marat Dukhan436ebe62019-12-04 15:10:12 -08001842 "@FP16",
1843 "@FXdiv",
1844 "@pthreadpool",
1845 ],
1846)
1847
1848xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07001849 name = "wasm_ukernels_test_mode",
1850 hdrs = INTERNAL_HDRS,
1851 copts = [
1852 "-UNDEBUG",
1853 "-DXNN_TEST_MODE=1",
1854 ],
1855 gcc_copts = xnnpack_gcc_std_copts(),
1856 msvc_copts = xnnpack_msvc_std_copts(),
1857 wasm_srcs = WASM_UKERNELS,
1858 deps = [
1859 ":tables",
1860 "@FP16",
1861 "@FXdiv",
1862 "@pthreadpool",
1863 ],
1864)
1865
1866xnnpack_cc_library(
Marat Dukhan8d3c07e2020-01-02 01:20:59 -08001867 name = "psimd_fastmath_ukernels",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001868 hdrs = INTERNAL_HDRS,
1869 aarch32_copts = [
1870 "-marm",
1871 "-mfpu=neon",
1872 ],
Marat Dukhan10a38082020-04-17 03:58:35 -07001873 gcc_copts = xnnpack_gcc_std_copts(),
1874 gcc_x86_copts = ["-msse2"],
1875 msvc_copts = xnnpack_msvc_std_copts(),
Marat Dukhan08c4a432019-10-03 09:29:21 -07001876 optimized_copts = [
1877 "-O3",
1878 "-ffast-math",
1879 ],
Marat Dukhan500b8892020-04-15 17:09:50 -07001880 psimd_srcs = PSIMD_FASTMATH_UKERNELS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07001881 deps = [
Marat Dukhan3a77ea72019-12-23 12:10:24 -08001882 ":tables",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001883 "@FP16",
1884 "@psimd",
Marat Dukhan04f03be2019-11-19 12:36:47 -08001885 "@pthreadpool",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001886 ],
1887)
1888
1889xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07001890 name = "psimd_fastmath_ukernels_test_mode",
1891 hdrs = INTERNAL_HDRS,
1892 aarch32_copts = [
1893 "-marm",
1894 "-mfpu=neon",
1895 ],
1896 copts = [
1897 "-UNDEBUG",
1898 "-DXNN_TEST_MODE=1",
1899 ],
1900 gcc_copts = xnnpack_gcc_std_copts(),
1901 gcc_x86_copts = ["-msse2"],
1902 msvc_copts = xnnpack_msvc_std_copts(),
1903 optimized_copts = [
1904 "-O3",
1905 "-ffast-math",
1906 ],
1907 psimd_srcs = PSIMD_FASTMATH_UKERNELS,
1908 deps = [
1909 ":tables",
1910 "@FP16",
1911 "@psimd",
1912 "@pthreadpool",
1913 ],
1914)
1915
1916xnnpack_cc_library(
Marat Dukhan8d3c07e2020-01-02 01:20:59 -08001917 name = "psimd_accmath_ukernels",
Marat Dukhan8d3c07e2020-01-02 01:20:59 -08001918 hdrs = INTERNAL_HDRS,
1919 aarch32_copts = [
1920 "-marm",
1921 "-mfpu=neon",
1922 ],
Marat Dukhan10a38082020-04-17 03:58:35 -07001923 gcc_copts = xnnpack_gcc_std_copts(),
1924 gcc_x86_copts = ["-msse2"],
1925 msvc_copts = xnnpack_msvc_std_copts(),
Marat Dukhan8d3c07e2020-01-02 01:20:59 -08001926 optimized_copts = [
1927 "-O3",
1928 ],
Marat Dukhan500b8892020-04-15 17:09:50 -07001929 psimd_srcs = PSIMD_ACCMATH_UKERNELS,
Marat Dukhan8d3c07e2020-01-02 01:20:59 -08001930 deps = [
1931 ":tables",
1932 "@FP16",
1933 "@psimd",
1934 "@pthreadpool",
1935 ],
1936)
1937
1938xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07001939 name = "psimd_accmath_ukernels_test_mode",
1940 hdrs = INTERNAL_HDRS,
1941 aarch32_copts = [
1942 "-marm",
1943 "-mfpu=neon",
1944 ],
1945 copts = [
1946 "-UNDEBUG",
1947 "-DXNN_TEST_MODE=1",
1948 ],
1949 gcc_copts = xnnpack_gcc_std_copts(),
1950 gcc_x86_copts = ["-msse2"],
1951 msvc_copts = xnnpack_msvc_std_copts(),
1952 optimized_copts = [
1953 "-O3",
1954 ],
1955 psimd_srcs = PSIMD_ACCMATH_UKERNELS,
1956 deps = [
1957 ":tables",
1958 "@FP16",
1959 "@psimd",
1960 "@pthreadpool",
1961 ],
1962)
1963
1964xnnpack_cc_library(
Marat Dukhan08c4a432019-10-03 09:29:21 -07001965 name = "neon_ukernels",
1966 hdrs = INTERNAL_HDRS,
1967 aarch32_copts = [
1968 "-marm",
Marat Dukhan8853b822020-05-07 12:19:01 -07001969 "-march=armv7-a",
Marat Dukhan08c4a432019-10-03 09:29:21 -07001970 "-mfpu=neon",
1971 ],
1972 aarch32_srcs = NEON_UKERNELS,
1973 aarch64_srcs = NEON_UKERNELS,
Marat Dukhan10a38082020-04-17 03:58:35 -07001974 gcc_copts = xnnpack_gcc_std_copts(),
1975 msvc_copts = xnnpack_msvc_std_copts(),
Marat Dukhan04f03be2019-11-19 12:36:47 -08001976 deps = [
Marat Dukhan3a77ea72019-12-23 12:10:24 -08001977 ":tables",
Marat Dukhan04f03be2019-11-19 12:36:47 -08001978 "@FP16",
1979 "@pthreadpool",
1980 ],
Marat Dukhan08c4a432019-10-03 09:29:21 -07001981)
1982
1983xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07001984 name = "neon_ukernels_test_mode",
1985 hdrs = INTERNAL_HDRS,
1986 aarch32_copts = [
1987 "-marm",
1988 "-march=armv7-a",
1989 "-mfpu=neon",
1990 ],
1991 aarch32_srcs = NEON_UKERNELS,
1992 aarch64_srcs = NEON_UKERNELS,
1993 copts = [
1994 "-UNDEBUG",
1995 "-DXNN_TEST_MODE=1",
1996 ],
1997 gcc_copts = xnnpack_gcc_std_copts(),
1998 msvc_copts = xnnpack_msvc_std_copts(),
1999 deps = [
2000 ":tables",
2001 "@FP16",
2002 "@pthreadpool",
2003 ],
2004)
2005
2006xnnpack_cc_library(
Marat Dukhan08c4a432019-10-03 09:29:21 -07002007 name = "neonfma_ukernels",
2008 hdrs = INTERNAL_HDRS,
2009 aarch32_copts = [
2010 "-marm",
Marat Dukhan8853b822020-05-07 12:19:01 -07002011 "-march=armv7-a",
Marat Dukhan08c4a432019-10-03 09:29:21 -07002012 "-mfpu=neon-vfpv4",
2013 ],
2014 aarch32_srcs = NEONFMA_UKERNELS,
2015 aarch64_srcs = NEONFMA_UKERNELS + AARCH64_NEONFMA_UKERNELS,
Marat Dukhan10a38082020-04-17 03:58:35 -07002016 gcc_copts = xnnpack_gcc_std_copts(),
2017 msvc_copts = xnnpack_msvc_std_copts(),
Marat Dukhan04f03be2019-11-19 12:36:47 -08002018 deps = [
Marat Dukhan3a77ea72019-12-23 12:10:24 -08002019 ":tables",
Marat Dukhan04f03be2019-11-19 12:36:47 -08002020 "@FP16",
2021 "@pthreadpool",
2022 ],
Marat Dukhan08c4a432019-10-03 09:29:21 -07002023)
2024
2025xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07002026 name = "neonfma_ukernels_test_mode",
2027 hdrs = INTERNAL_HDRS,
2028 aarch32_copts = [
2029 "-marm",
2030 "-march=armv7-a",
2031 "-mfpu=neon-vfpv4",
2032 ],
2033 aarch32_srcs = NEONFMA_UKERNELS,
2034 aarch64_srcs = NEONFMA_UKERNELS + AARCH64_NEONFMA_UKERNELS,
2035 copts = [
2036 "-UNDEBUG",
2037 "-DXNN_TEST_MODE=1",
2038 ],
2039 gcc_copts = xnnpack_gcc_std_copts(),
2040 msvc_copts = xnnpack_msvc_std_copts(),
2041 deps = [
2042 ":tables",
2043 "@FP16",
2044 "@pthreadpool",
2045 ],
2046)
2047
2048xnnpack_cc_library(
Marat Dukhan8853b822020-05-07 12:19:01 -07002049 name = "neonv8_ukernels",
2050 hdrs = INTERNAL_HDRS,
2051 aarch32_copts = [
2052 "-marm",
2053 "-march=armv8-a",
2054 "-mfpu=neon-fp-armv8",
2055 ],
2056 aarch32_srcs = NEONV8_UKERNELS,
2057 aarch64_srcs = NEONV8_UKERNELS,
2058 gcc_copts = xnnpack_gcc_std_copts(),
2059 msvc_copts = xnnpack_msvc_std_copts(),
2060 deps = [
2061 ":tables",
2062 "@FP16",
2063 "@pthreadpool",
2064 ],
2065)
2066
2067xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07002068 name = "neonv8_ukernels_test_mode",
2069 hdrs = INTERNAL_HDRS,
2070 aarch32_copts = [
2071 "-marm",
2072 "-march=armv8-a",
2073 "-mfpu=neon-fp-armv8",
2074 ],
2075 aarch32_srcs = NEONV8_UKERNELS,
2076 aarch64_srcs = NEONV8_UKERNELS,
2077 copts = [
2078 "-UNDEBUG",
2079 "-DXNN_TEST_MODE=1",
2080 ],
2081 gcc_copts = xnnpack_gcc_std_copts(),
2082 msvc_copts = xnnpack_msvc_std_copts(),
2083 deps = [
2084 ":tables",
2085 "@FP16",
2086 "@pthreadpool",
2087 ],
2088)
2089
2090xnnpack_cc_library(
Marat Dukhan08c4a432019-10-03 09:29:21 -07002091 name = "neonfp16arith_ukernels",
2092 hdrs = INTERNAL_HDRS,
2093 aarch64_copts = ["-march=armv8.2-a+fp16"],
2094 aarch64_srcs = AARCH64_NEONFP16ARITH_UKERNELS,
Marat Dukhan10a38082020-04-17 03:58:35 -07002095 gcc_copts = xnnpack_gcc_std_copts(),
2096 msvc_copts = xnnpack_msvc_std_copts(),
Marat Dukhan04f03be2019-11-19 12:36:47 -08002097 deps = [
Marat Dukhan3a77ea72019-12-23 12:10:24 -08002098 ":tables",
Marat Dukhan04f03be2019-11-19 12:36:47 -08002099 "@FP16",
2100 "@pthreadpool",
2101 ],
Marat Dukhan08c4a432019-10-03 09:29:21 -07002102)
2103
2104xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07002105 name = "neonfp16arith_ukernels_test_mode",
2106 hdrs = INTERNAL_HDRS,
2107 aarch64_copts = ["-march=armv8.2-a+fp16"],
2108 aarch64_srcs = AARCH64_NEONFP16ARITH_UKERNELS,
2109 copts = [
2110 "-UNDEBUG",
2111 "-DXNN_TEST_MODE=1",
2112 ],
2113 gcc_copts = xnnpack_gcc_std_copts(),
2114 msvc_copts = xnnpack_msvc_std_copts(),
2115 deps = [
2116 ":tables",
2117 "@FP16",
2118 "@pthreadpool",
2119 ],
2120)
2121
2122xnnpack_cc_library(
Marat Dukhan08c4a432019-10-03 09:29:21 -07002123 name = "sse2_ukernels",
2124 hdrs = INTERNAL_HDRS,
Marat Dukhan10a38082020-04-17 03:58:35 -07002125 gcc_copts = xnnpack_gcc_std_copts(),
2126 gcc_x86_copts = ["-msse2"],
2127 msvc_copts = xnnpack_msvc_std_copts(),
2128 msvc_x86_32_copts = ["/arch:SSE2"],
Marat Dukhan08c4a432019-10-03 09:29:21 -07002129 x86_srcs = SSE_UKERNELS + SSE2_UKERNELS,
Marat Dukhan04f03be2019-11-19 12:36:47 -08002130 deps = [
Marat Dukhan3a77ea72019-12-23 12:10:24 -08002131 ":tables",
Marat Dukhan04f03be2019-11-19 12:36:47 -08002132 "@FP16",
2133 "@pthreadpool",
2134 ],
Marat Dukhan08c4a432019-10-03 09:29:21 -07002135)
2136
2137xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07002138 name = "sse2_ukernels_test_mode",
2139 hdrs = INTERNAL_HDRS,
2140 copts = [
2141 "-UNDEBUG",
2142 "-DXNN_TEST_MODE=1",
2143 ],
2144 gcc_copts = xnnpack_gcc_std_copts(),
2145 gcc_x86_copts = ["-msse2"],
2146 msvc_copts = xnnpack_msvc_std_copts(),
2147 msvc_x86_32_copts = ["/arch:SSE2"],
2148 x86_srcs = SSE_UKERNELS + SSE2_UKERNELS,
2149 deps = [
2150 ":tables",
2151 "@FP16",
2152 "@pthreadpool",
2153 ],
2154)
2155
2156xnnpack_cc_library(
Marat Dukhanfe7acb62020-03-09 19:30:05 -07002157 name = "ssse3_ukernels",
2158 hdrs = INTERNAL_HDRS,
Marat Dukhan10a38082020-04-17 03:58:35 -07002159 gcc_copts = xnnpack_gcc_std_copts(),
2160 gcc_x86_copts = ["-mssse3"],
2161 msvc_copts = xnnpack_msvc_std_copts(),
2162 msvc_x86_32_copts = ["/arch:SSE2"],
Marat Dukhanfe7acb62020-03-09 19:30:05 -07002163 x86_srcs = SSSE3_UKERNELS,
2164 deps = [
2165 ":tables",
2166 "@FP16",
2167 "@pthreadpool",
2168 ],
2169)
2170
2171xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07002172 name = "ssse3_ukernels_test_mode",
2173 hdrs = INTERNAL_HDRS,
2174 copts = [
2175 "-UNDEBUG",
2176 "-DXNN_TEST_MODE=1",
2177 ],
2178 gcc_copts = xnnpack_gcc_std_copts(),
2179 gcc_x86_copts = ["-mssse3"],
2180 msvc_copts = xnnpack_msvc_std_copts(),
2181 msvc_x86_32_copts = ["/arch:SSE2"],
2182 x86_srcs = SSSE3_UKERNELS,
2183 deps = [
2184 ":tables",
2185 "@FP16",
2186 "@pthreadpool",
2187 ],
2188)
2189
2190xnnpack_cc_library(
Marat Dukhan69c3f2c2019-11-06 12:30:01 -08002191 name = "sse41_ukernels",
2192 hdrs = INTERNAL_HDRS,
Marat Dukhan10a38082020-04-17 03:58:35 -07002193 gcc_copts = xnnpack_gcc_std_copts(),
2194 gcc_x86_copts = ["-msse4.1"],
2195 msvc_copts = xnnpack_msvc_std_copts(),
2196 msvc_x86_32_copts = ["/arch:SSE2"],
Marat Dukhan69c3f2c2019-11-06 12:30:01 -08002197 x86_srcs = SSE41_UKERNELS,
Marat Dukhan04f03be2019-11-19 12:36:47 -08002198 deps = [
Marat Dukhan3a77ea72019-12-23 12:10:24 -08002199 ":tables",
Marat Dukhan04f03be2019-11-19 12:36:47 -08002200 "@FP16",
2201 "@pthreadpool",
2202 ],
Marat Dukhan69c3f2c2019-11-06 12:30:01 -08002203)
2204
2205xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07002206 name = "sse41_ukernels_test_mode",
2207 hdrs = INTERNAL_HDRS,
2208 copts = [
2209 "-UNDEBUG",
2210 "-DXNN_TEST_MODE=1",
2211 ],
2212 gcc_copts = xnnpack_gcc_std_copts(),
2213 gcc_x86_copts = ["-msse4.1"],
2214 msvc_copts = xnnpack_msvc_std_copts(),
2215 msvc_x86_32_copts = ["/arch:SSE2"],
2216 x86_srcs = SSE41_UKERNELS,
2217 deps = [
2218 ":tables",
2219 "@FP16",
2220 "@pthreadpool",
2221 ],
2222)
2223
2224xnnpack_cc_library(
Marat Dukhan08c4a432019-10-03 09:29:21 -07002225 name = "avx_ukernels",
2226 hdrs = INTERNAL_HDRS,
Marat Dukhan10a38082020-04-17 03:58:35 -07002227 gcc_copts = xnnpack_gcc_std_copts(),
2228 gcc_x86_copts = ["-mavx"],
2229 msvc_copts = xnnpack_msvc_std_copts(),
2230 msvc_x86_32_copts = ["/arch:AVX"],
2231 msvc_x86_64_copts = ["/arch:AVX"],
Marat Dukhan08c4a432019-10-03 09:29:21 -07002232 x86_srcs = AVX_UKERNELS,
Marat Dukhan04f03be2019-11-19 12:36:47 -08002233 deps = [
Marat Dukhan3a77ea72019-12-23 12:10:24 -08002234 ":tables",
Marat Dukhan04f03be2019-11-19 12:36:47 -08002235 "@FP16",
2236 "@pthreadpool",
2237 ],
Marat Dukhan08c4a432019-10-03 09:29:21 -07002238)
2239
2240xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07002241 name = "avx_ukernels_test_mode",
2242 hdrs = INTERNAL_HDRS,
2243 copts = [
2244 "-UNDEBUG",
2245 "-DXNN_TEST_MODE=1",
2246 ],
2247 gcc_copts = xnnpack_gcc_std_copts(),
2248 gcc_x86_copts = ["-mavx"],
2249 msvc_copts = xnnpack_msvc_std_copts(),
2250 msvc_x86_32_copts = ["/arch:AVX"],
2251 msvc_x86_64_copts = ["/arch:AVX"],
2252 x86_srcs = AVX_UKERNELS,
2253 deps = [
2254 ":tables",
2255 "@FP16",
2256 "@pthreadpool",
2257 ],
2258)
2259
2260xnnpack_cc_library(
Marat Dukhanfda12b82019-11-21 12:27:59 -08002261 name = "fma3_ukernels",
2262 hdrs = INTERNAL_HDRS,
Marat Dukhan10a38082020-04-17 03:58:35 -07002263 gcc_copts = xnnpack_gcc_std_copts(),
2264 gcc_x86_copts = ["-mfma"],
2265 msvc_copts = xnnpack_msvc_std_copts(),
2266 msvc_x86_32_copts = ["/arch:AVX"],
2267 msvc_x86_64_copts = ["/arch:AVX"],
Marat Dukhanfda12b82019-11-21 12:27:59 -08002268 x86_srcs = FMA3_UKERNELS,
2269 deps = [
Marat Dukhan3a77ea72019-12-23 12:10:24 -08002270 ":tables",
Marat Dukhanfda12b82019-11-21 12:27:59 -08002271 "@FP16",
2272 "@pthreadpool",
2273 ],
2274)
2275
2276xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07002277 name = "fma3_ukernels_test_mode",
2278 hdrs = INTERNAL_HDRS,
2279 copts = [
2280 "-UNDEBUG",
2281 "-DXNN_TEST_MODE=1",
2282 ],
2283 gcc_copts = xnnpack_gcc_std_copts(),
2284 gcc_x86_copts = ["-mfma"],
2285 msvc_copts = xnnpack_msvc_std_copts(),
2286 msvc_x86_32_copts = ["/arch:AVX"],
2287 msvc_x86_64_copts = ["/arch:AVX"],
2288 x86_srcs = FMA3_UKERNELS,
2289 deps = [
2290 ":tables",
2291 "@FP16",
2292 "@pthreadpool",
2293 ],
2294)
2295
2296xnnpack_cc_library(
Marat Dukhan6adff4e2019-10-14 18:32:07 -07002297 name = "avx2_ukernels",
2298 hdrs = INTERNAL_HDRS,
Marat Dukhan10a38082020-04-17 03:58:35 -07002299 gcc_copts = xnnpack_gcc_std_copts(),
2300 gcc_x86_copts = [
Marat Dukhan6adff4e2019-10-14 18:32:07 -07002301 "-mfma",
2302 "-mavx2",
2303 ],
Marat Dukhan10a38082020-04-17 03:58:35 -07002304 msvc_copts = xnnpack_msvc_std_copts(),
2305 msvc_x86_32_copts = ["/arch:AVX2"],
2306 msvc_x86_64_copts = ["/arch:AVX2"],
Marat Dukhan6adff4e2019-10-14 18:32:07 -07002307 x86_srcs = AVX2_UKERNELS,
Marat Dukhan04f03be2019-11-19 12:36:47 -08002308 deps = [
Marat Dukhan3a77ea72019-12-23 12:10:24 -08002309 ":tables",
Marat Dukhan04f03be2019-11-19 12:36:47 -08002310 "@FP16",
2311 "@pthreadpool",
2312 ],
Marat Dukhan6adff4e2019-10-14 18:32:07 -07002313)
2314
2315xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07002316 name = "avx2_ukernels_test_mode",
2317 hdrs = INTERNAL_HDRS,
2318 copts = [
2319 "-UNDEBUG",
2320 "-DXNN_TEST_MODE=1",
2321 ],
2322 gcc_copts = xnnpack_gcc_std_copts(),
2323 gcc_x86_copts = [
2324 "-mfma",
2325 "-mavx2",
2326 ],
2327 msvc_copts = xnnpack_msvc_std_copts(),
2328 msvc_x86_32_copts = ["/arch:AVX2"],
2329 msvc_x86_64_copts = ["/arch:AVX2"],
2330 x86_srcs = AVX2_UKERNELS,
2331 deps = [
2332 ":tables",
2333 "@FP16",
2334 "@pthreadpool",
2335 ],
2336)
2337
2338xnnpack_cc_library(
Marat Dukhan08c4a432019-10-03 09:29:21 -07002339 name = "avx512f_ukernels",
2340 hdrs = INTERNAL_HDRS,
Marat Dukhan10a38082020-04-17 03:58:35 -07002341 gcc_copts = xnnpack_gcc_std_copts(),
2342 gcc_x86_copts = ["-mavx512f"],
2343 mingw_copts = ["-fno-asynchronous-unwind-tables"],
2344 msvc_copts = xnnpack_msvc_std_copts(),
2345 msvc_x86_32_copts = ["/arch:AVX512"],
2346 msvc_x86_64_copts = ["/arch:AVX512"],
2347 msys_copts = ["-fno-asynchronous-unwind-tables"],
Marat Dukhan08c4a432019-10-03 09:29:21 -07002348 x86_srcs = AVX512F_UKERNELS,
Marat Dukhan04f03be2019-11-19 12:36:47 -08002349 deps = [
Marat Dukhan3a77ea72019-12-23 12:10:24 -08002350 ":tables",
Marat Dukhan04f03be2019-11-19 12:36:47 -08002351 "@FP16",
2352 "@pthreadpool",
2353 ],
Marat Dukhan08c4a432019-10-03 09:29:21 -07002354)
2355
2356xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07002357 name = "avx512f_ukernels_test_mode",
2358 hdrs = INTERNAL_HDRS,
2359 copts = [
2360 "-UNDEBUG",
2361 "-DXNN_TEST_MODE=1",
2362 ],
2363 gcc_copts = xnnpack_gcc_std_copts(),
2364 gcc_x86_copts = ["-mavx512f"],
2365 mingw_copts = ["-fno-asynchronous-unwind-tables"],
2366 msvc_copts = xnnpack_msvc_std_copts(),
2367 msvc_x86_32_copts = ["/arch:AVX512"],
2368 msvc_x86_64_copts = ["/arch:AVX512"],
2369 msys_copts = ["-fno-asynchronous-unwind-tables"],
2370 x86_srcs = AVX512F_UKERNELS,
2371 deps = [
2372 ":tables",
2373 "@FP16",
2374 "@pthreadpool",
2375 ],
2376)
2377
2378xnnpack_cc_library(
Marat Dukhan08c4a432019-10-03 09:29:21 -07002379 name = "asm_ukernels",
2380 hdrs = ["src/xnnpack/assembly.h"],
2381 aarch32_srcs = AARCH32_ASM_UKERNELS,
Frank Barchard683f5592020-04-10 00:48:26 -07002382 aarch64_copts = ["-march=armv8.2-a+fp16"],
Marat Dukhan08c4a432019-10-03 09:29:21 -07002383 aarch64_srcs = AARCH64_ASM_UKERNELS,
2384)
2385
2386xnnpack_aggregate_library(
2387 name = "ukernels",
2388 aarch32_deps = [
Marat Dukhan08c4a432019-10-03 09:29:21 -07002389 ":neon_ukernels",
2390 ":neonfma_ukernels",
Marat Dukhan8853b822020-05-07 12:19:01 -07002391 ":neonv8_ukernels",
Marat Dukhan08c4a432019-10-03 09:29:21 -07002392 ":asm_ukernels",
2393 ],
2394 aarch64_deps = [
Marat Dukhan08c4a432019-10-03 09:29:21 -07002395 ":neon_ukernels",
2396 ":neonfma_ukernels",
Marat Dukhan8853b822020-05-07 12:19:01 -07002397 ":neonv8_ukernels",
Marat Dukhan08c4a432019-10-03 09:29:21 -07002398 ":neonfp16arith_ukernels",
2399 ":asm_ukernels",
2400 ],
Marat Dukhan33fcf782020-05-24 14:27:15 -07002401 generic_deps = [
2402 ":scalar_ukernels",
2403 ],
Marat Dukhan500b8892020-04-15 17:09:50 -07002404 psimd_deps = [
Marat Dukhan8d3c07e2020-01-02 01:20:59 -08002405 ":psimd_fastmath_ukernels",
2406 ":psimd_accmath_ukernels",
Marat Dukhan08c4a432019-10-03 09:29:21 -07002407 ],
Marat Dukhan33fcf782020-05-24 14:27:15 -07002408 wasm_deps = [
2409 ":wasm_ukernels",
2410 ],
2411 wasmsimd_deps = [
2412 ":wasm_ukernels",
2413 ],
Marat Dukhan08c4a432019-10-03 09:29:21 -07002414 x86_deps = [
Marat Dukhan08c4a432019-10-03 09:29:21 -07002415 ":sse2_ukernels",
Marat Dukhanfe7acb62020-03-09 19:30:05 -07002416 ":ssse3_ukernels",
Marat Dukhan69c3f2c2019-11-06 12:30:01 -08002417 ":sse41_ukernels",
Marat Dukhan08c4a432019-10-03 09:29:21 -07002418 ":avx_ukernels",
Marat Dukhanfda12b82019-11-21 12:27:59 -08002419 ":fma3_ukernels",
Marat Dukhan6adff4e2019-10-14 18:32:07 -07002420 ":avx2_ukernels",
Marat Dukhan08c4a432019-10-03 09:29:21 -07002421 ":avx512f_ukernels",
2422 ],
2423)
2424
Marat Dukhan33fcf782020-05-24 14:27:15 -07002425xnnpack_aggregate_library(
2426 name = "ukernels_test_mode",
2427 aarch32_deps = [
2428 ":neon_ukernels_test_mode",
2429 ":neonfma_ukernels_test_mode",
2430 ":neonv8_ukernels_test_mode",
2431 ":asm_ukernels",
2432 ],
2433 aarch64_deps = [
2434 ":neon_ukernels_test_mode",
2435 ":neonfma_ukernels_test_mode",
2436 ":neonv8_ukernels_test_mode",
2437 ":neonfp16arith_ukernels_test_mode",
2438 ":asm_ukernels",
2439 ],
2440 generic_deps = [
2441 ":scalar_ukernels_test_mode",
2442 ],
2443 psimd_deps = [
2444 ":psimd_fastmath_ukernels_test_mode",
2445 ":psimd_accmath_ukernels_test_mode",
2446 ],
2447 wasm_deps = [
2448 ":wasm_ukernels_test_mode",
2449 ],
2450 wasmsimd_deps = [
2451 ":wasm_ukernels_test_mode",
2452 ],
2453 x86_deps = [
2454 ":sse2_ukernels_test_mode",
2455 ":ssse3_ukernels_test_mode",
2456 ":sse41_ukernels_test_mode",
2457 ":avx_ukernels_test_mode",
2458 ":fma3_ukernels_test_mode",
2459 ":avx2_ukernels_test_mode",
2460 ":avx512f_ukernels_test_mode",
2461 ],
2462)
2463
Marat Dukhan08c4a432019-10-03 09:29:21 -07002464xnnpack_cc_library(
2465 name = "im2col",
2466 srcs = ["src/im2col.c"],
2467 hdrs = [
2468 "src/xnnpack/common.h",
2469 "src/xnnpack/im2col.h",
2470 ],
Marat Dukhan10a38082020-04-17 03:58:35 -07002471 gcc_copts = xnnpack_gcc_std_copts(),
2472 msvc_copts = xnnpack_msvc_std_copts(),
Marat Dukhan08c4a432019-10-03 09:29:21 -07002473)
2474
2475xnnpack_cc_library(
2476 name = "indirection",
2477 srcs = ["src/indirection.c"],
2478 hdrs = INTERNAL_HDRS,
Marat Dukhan10a38082020-04-17 03:58:35 -07002479 gcc_copts = xnnpack_gcc_std_copts(),
2480 msvc_copts = xnnpack_msvc_std_copts(),
Marat Dukhan08c4a432019-10-03 09:29:21 -07002481 deps = [
2482 "@FP16",
2483 "@FXdiv",
2484 "@pthreadpool",
2485 ],
2486)
2487
2488xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07002489 name = "indirection_test_mode",
2490 srcs = ["src/indirection.c"],
2491 hdrs = INTERNAL_HDRS,
2492 copts = [
2493 "-UNDEBUG",
2494 "-DXNN_TEST_MODE=1",
2495 ],
2496 gcc_copts = xnnpack_gcc_std_copts(),
2497 msvc_copts = xnnpack_msvc_std_copts(),
2498 deps = [
2499 "@FP16",
2500 "@FXdiv",
2501 "@pthreadpool",
2502 ],
2503)
2504
2505xnnpack_cc_library(
Marat Dukhan08c4a432019-10-03 09:29:21 -07002506 name = "operator_run",
2507 srcs = ["src/operator-run.c"],
Marat Dukhanc8e00eb2019-10-04 14:55:26 -07002508 hdrs = INTERNAL_HDRS + LOGGING_HDRS,
Marat Dukhan10a38082020-04-17 03:58:35 -07002509 copts = LOGGING_COPTS + select({
Marat Dukhan05702cf2020-03-26 15:41:33 -07002510 ":xnn_enable_hmp_explicit_false": ["-DXNN_MAX_UARCH_TYPES=1"],
2511 "//conditions:default": [],
2512 }),
Marat Dukhan10a38082020-04-17 03:58:35 -07002513 gcc_copts = xnnpack_gcc_std_copts(),
2514 msvc_copts = xnnpack_msvc_std_copts(),
Marat Dukhan08c4a432019-10-03 09:29:21 -07002515 deps = [
2516 "@FP16",
2517 "@FXdiv",
2518 "@clog",
2519 "@pthreadpool",
2520 ],
2521)
2522
Chao Mei6ddfc602020-05-13 22:29:36 -07002523xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07002524 name = "operator_run_test_mode",
2525 srcs = ["src/operator-run.c"],
2526 hdrs = INTERNAL_HDRS + LOGGING_HDRS,
2527 copts = LOGGING_COPTS + [
2528 "-UNDEBUG",
2529 "-DXNN_TEST_MODE=1",
2530 ] + select({
2531 ":xnn_enable_hmp_explicit_false": ["-DXNN_MAX_UARCH_TYPES=1"],
2532 "//conditions:default": [],
2533 }),
2534 gcc_copts = xnnpack_gcc_std_copts(),
2535 msvc_copts = xnnpack_msvc_std_copts(),
2536 deps = [
2537 "@FP16",
2538 "@FXdiv",
2539 "@clog",
2540 "@pthreadpool",
2541 ],
2542)
2543
2544xnnpack_cc_library(
Chao Mei6ddfc602020-05-13 22:29:36 -07002545 name = "memory_planner",
2546 srcs = ["src/memory-planner.c"],
2547 hdrs = INTERNAL_HDRS,
2548 defines = select({
2549 ":xnn_enable_memopt_explicit_true": ["XNN_ENABLE_MEMOPT=1"],
2550 ":xnn_enable_memopt_explicit_false": ["XNN_ENABLE_MEMOPT=0"],
2551 "//conditions:default": ["XNN_ENABLE_MEMOPT=1"],
2552 }),
2553 gcc_copts = xnnpack_gcc_std_copts(),
2554 msvc_copts = xnnpack_msvc_std_copts(),
2555 deps = [
2556 "@pthreadpool",
2557 ],
2558)
2559
Marat Dukhan33fcf782020-05-24 14:27:15 -07002560xnnpack_cc_library(
2561 name = "memory_planner_test_mode",
2562 srcs = ["src/memory-planner.c"],
2563 hdrs = INTERNAL_HDRS,
2564 copts = [
2565 "-UNDEBUG",
2566 "-DXNN_TEST_MODE=1",
2567 ],
2568 defines = select({
2569 ":xnn_enable_memopt_explicit_true": ["XNN_ENABLE_MEMOPT=1"],
2570 ":xnn_enable_memopt_explicit_false": ["XNN_ENABLE_MEMOPT=0"],
2571 "//conditions:default": ["XNN_ENABLE_MEMOPT=1"],
2572 }),
2573 gcc_copts = xnnpack_gcc_std_copts(),
2574 msvc_copts = xnnpack_msvc_std_copts(),
2575 deps = [
2576 "@pthreadpool",
2577 ],
2578)
2579
Marat Dukhan08c4a432019-10-03 09:29:21 -07002580cc_library(
2581 name = "enable_assembly",
2582 defines = select({
2583 ":xnn_enable_assembly_explicit_true": ["XNN_ENABLE_ASSEMBLY=1"],
2584 ":xnn_enable_assembly_explicit_false": ["XNN_ENABLE_ASSEMBLY=0"],
Frank Barchard810171d2019-10-10 10:34:51 -07002585 "//conditions:default": ["XNN_ENABLE_ASSEMBLY=1"],
Marat Dukhan08c4a432019-10-03 09:29:21 -07002586 }),
2587)
2588
Marat Dukhancf056b22019-10-07 10:26:29 -07002589xnnpack_cc_library(
Marat Dukhan08c4a432019-10-03 09:29:21 -07002590 name = "operators",
2591 srcs = OPERATOR_SRCS + [
Marat Dukhan04f03be2019-11-19 12:36:47 -08002592 "src/memory.c",
Marat Dukhan08c4a432019-10-03 09:29:21 -07002593 "src/operator-delete.c",
Marat Dukhancf056b22019-10-07 10:26:29 -07002594 ],
2595 hdrs = INTERNAL_HDRS + LOGGING_HDRS,
Marat Dukhan10a38082020-04-17 03:58:35 -07002596 copts = LOGGING_COPTS + [
Marat Dukhan08c4a432019-10-03 09:29:21 -07002597 "-Isrc",
2598 "-Iinclude",
2599 ] + select({
2600 ":debug_build": [],
2601 "//conditions:default": xnnpack_min_size_copts(),
Marat Dukhan05702cf2020-03-26 15:41:33 -07002602 }) + select({
2603 ":xnn_enable_hmp_explicit_false": ["-DXNN_MAX_UARCH_TYPES=1"],
2604 "//conditions:default": [],
Marat Dukhan08c4a432019-10-03 09:29:21 -07002605 }),
Marat Dukhan10a38082020-04-17 03:58:35 -07002606 gcc_copts = xnnpack_gcc_std_copts(),
2607 msvc_copts = xnnpack_msvc_std_copts(),
Marat Dukhan08c4a432019-10-03 09:29:21 -07002608 deps = [
Marat Dukhan08c4a432019-10-03 09:29:21 -07002609 ":indirection",
Marat Dukhan08c4a432019-10-03 09:29:21 -07002610 "@FP16",
2611 "@FXdiv",
2612 "@clog",
Marat Dukhan08c4a432019-10-03 09:29:21 -07002613 "@pthreadpool",
Marat Dukhan8fe54e42019-10-10 14:12:59 -07002614 ],
2615)
2616
Marat Dukhan10a38082020-04-17 03:58:35 -07002617xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07002618 name = "operators_test_mode",
2619 srcs = OPERATOR_SRCS + [
2620 "src/memory.c",
2621 "src/operator-delete.c",
2622 ],
2623 hdrs = INTERNAL_HDRS + LOGGING_HDRS,
2624 copts = LOGGING_COPTS + [
2625 "-Isrc",
2626 "-Iinclude",
2627 "-UNDEBUG",
2628 "-DXNN_TEST_MODE=1",
2629 ] + select({
2630 ":debug_build": [],
2631 "//conditions:default": xnnpack_min_size_copts(),
2632 }) + select({
2633 ":xnn_enable_hmp_explicit_false": ["-DXNN_MAX_UARCH_TYPES=1"],
2634 "//conditions:default": [],
2635 }),
2636 gcc_copts = xnnpack_gcc_std_copts(),
2637 msvc_copts = xnnpack_msvc_std_copts(),
2638 deps = [
2639 ":indirection_test_mode",
2640 "@FP16",
2641 "@FXdiv",
2642 "@clog",
2643 "@pthreadpool",
2644 ],
2645)
2646
2647xnnpack_cc_library(
Marat Dukhan8fe54e42019-10-10 14:12:59 -07002648 name = "XNNPACK",
2649 srcs = [
2650 "src/init.c",
Marat Dukhanccfdbd12020-02-03 14:27:45 -08002651 "src/runtime.c",
2652 "src/subgraph.c",
2653 "src/tensor.c",
Marat Dukhan8fe54e42019-10-10 14:12:59 -07002654 ],
Marat Dukhan10a38082020-04-17 03:58:35 -07002655 hdrs = ["include/xnnpack.h"],
2656 copts = LOGGING_COPTS + [
Marat Dukhan8fe54e42019-10-10 14:12:59 -07002657 "-Isrc",
2658 "-Iinclude",
2659 ] + select({
2660 ":debug_build": [],
2661 "//conditions:default": xnnpack_min_size_copts(),
Marat Dukhan05702cf2020-03-26 15:41:33 -07002662 }) + select({
2663 ":xnn_enable_hmp_explicit_false": ["-DXNN_MAX_UARCH_TYPES=1"],
2664 "//conditions:default": [],
Marat Dukhan8fe54e42019-10-10 14:12:59 -07002665 }),
Marat Dukhan10a38082020-04-17 03:58:35 -07002666 gcc_copts = xnnpack_gcc_std_copts(),
Marat Dukhan8fe54e42019-10-10 14:12:59 -07002667 includes = ["include"],
Marat Dukhan10a38082020-04-17 03:58:35 -07002668 msvc_copts = xnnpack_msvc_std_copts(),
Marat Dukhan8fe54e42019-10-10 14:12:59 -07002669 visibility = xnnpack_visibility(),
2670 deps = [
2671 ":enable_assembly",
Chao Mei6ddfc602020-05-13 22:29:36 -07002672 ":memory_planner",
Marat Dukhan8fe54e42019-10-10 14:12:59 -07002673 ":ukernels",
2674 ":operator_run",
2675 ":operators",
2676 "@clog",
Marat Dukhanab2946c2020-05-21 20:04:13 -07002677 "@FP16",
Marat Dukhan8fe54e42019-10-10 14:12:59 -07002678 "@pthreadpool",
Marat Dukhand343c222019-10-07 09:22:14 -07002679 ] + select({
2680 ":emscripten": [],
2681 "//conditions:default": ["@cpuinfo"],
2682 }),
Marat Dukhan08c4a432019-10-03 09:29:21 -07002683)
2684
Marat Dukhan10a38082020-04-17 03:58:35 -07002685xnnpack_cc_library(
Marat Dukhan33fcf782020-05-24 14:27:15 -07002686 name = "XNNPACK_test_mode",
2687 srcs = [
2688 "src/init.c",
2689 "src/runtime.c",
2690 "src/subgraph.c",
2691 "src/tensor.c",
2692 ],
2693 hdrs = ["include/xnnpack.h"],
2694 copts = LOGGING_COPTS + [
2695 "-Isrc",
2696 "-Iinclude",
2697 "-UNDEBUG",
2698 "-DXNN_TEST_MODE=1",
2699 ] + select({
2700 ":debug_build": [],
2701 "//conditions:default": xnnpack_min_size_copts(),
2702 }) + select({
2703 ":xnn_enable_hmp_explicit_false": ["-DXNN_MAX_UARCH_TYPES=1"],
2704 "//conditions:default": [],
2705 }),
2706 gcc_copts = xnnpack_gcc_std_copts(),
2707 includes = ["include"],
2708 msvc_copts = xnnpack_msvc_std_copts(),
2709 visibility = xnnpack_visibility(),
2710 deps = [
2711 ":enable_assembly",
2712 ":memory_planner_test_mode",
2713 ":ukernels_test_mode",
2714 ":operator_run_test_mode",
2715 ":operators_test_mode",
2716 "@clog",
2717 "@FP16",
2718 "@pthreadpool",
2719 ] + select({
2720 ":emscripten": [],
2721 "//conditions:default": ["@cpuinfo"],
2722 }),
2723)
2724
2725xnnpack_cc_library(
Marat Dukhan8fe54e42019-10-10 14:12:59 -07002726 name = "xnnpack_operators_nhwc_f32",
2727 srcs = [
2728 "src/init.c",
2729 ],
Marat Dukhan10a38082020-04-17 03:58:35 -07002730 hdrs = ["include/xnnpack.h"],
2731 copts = LOGGING_COPTS + [
Marat Dukhan8fe54e42019-10-10 14:12:59 -07002732 "-Isrc",
2733 "-Iinclude",
2734 ] + select({
2735 ":debug_build": [],
2736 "//conditions:default": xnnpack_min_size_copts(),
Marat Dukhan05702cf2020-03-26 15:41:33 -07002737 }) + select({
2738 ":xnn_enable_hmp_explicit_false": ["-DXNN_MAX_UARCH_TYPES=1"],
2739 "//conditions:default": [],
Marat Dukhan8fe54e42019-10-10 14:12:59 -07002740 }),
2741 defines = [
2742 "XNN_NO_Q8_OPERATORS",
2743 "XNN_NO_U8_OPERATORS",
2744 "XNN_NO_X8_OPERATORS",
Marat Dukhanefc47b82019-11-18 09:25:38 -08002745 "XNN_NO_NCHW_OPERATORS",
Marat Dukhan8fe54e42019-10-10 14:12:59 -07002746 ],
Marat Dukhan10a38082020-04-17 03:58:35 -07002747 gcc_copts = xnnpack_gcc_std_copts(),
Marat Dukhan08c4a432019-10-03 09:29:21 -07002748 includes = ["include"],
Marat Dukhan10a38082020-04-17 03:58:35 -07002749 msvc_copts = xnnpack_msvc_std_copts(),
Marat Dukhan08c4a432019-10-03 09:29:21 -07002750 visibility = xnnpack_visibility(),
2751 deps = [
Marat Dukhan8fe54e42019-10-10 14:12:59 -07002752 ":enable_assembly",
2753 ":ukernels",
Marat Dukhan08c4a432019-10-03 09:29:21 -07002754 ":operator_run",
2755 ":operators",
Marat Dukhan8fe54e42019-10-10 14:12:59 -07002756 "@clog",
Marat Dukhan08c4a432019-10-03 09:29:21 -07002757 "@pthreadpool",
Marat Dukhan8fe54e42019-10-10 14:12:59 -07002758 ] + select({
2759 ":emscripten": [],
2760 "//conditions:default": ["@cpuinfo"],
2761 }),
Marat Dukhan08c4a432019-10-03 09:29:21 -07002762)
2763
Marat Dukhancf056b22019-10-07 10:26:29 -07002764xnnpack_cc_library(
Marat Dukhan08c4a432019-10-03 09:29:21 -07002765 name = "bench_utils",
2766 srcs = ["bench/utils.cc"],
2767 hdrs = ["bench/utils.h"],
Marat Dukhanbad48fe2019-11-04 10:35:22 -08002768 deps = [
2769 "@com_google_benchmark//:benchmark",
2770 "@cpuinfo",
2771 ],
Marat Dukhan08c4a432019-10-03 09:29:21 -07002772)
2773
Frank Barchard7e955972019-10-11 10:34:25 -07002774######################### Benchmarks for micro-kernels #########################
Marat Dukhan08c4a432019-10-03 09:29:21 -07002775
2776xnnpack_benchmark(
2777 name = "q8_gemm_bench",
2778 srcs = [
2779 "bench/gemm.h",
2780 "bench/q8-gemm.cc",
2781 "src/xnnpack/AlignedAllocator.h",
2782 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_BENCHMARK_HDRS,
Marat Dukhan10a38082020-04-17 03:58:35 -07002783 copts = xnnpack_optional_ruy_copts() + xnnpack_optional_gemmlowp_copts(),
Marat Dukhan08c4a432019-10-03 09:29:21 -07002784 deps = MICROKERNEL_BENCHMARK_DEPS + xnnpack_optional_ruy_deps() + xnnpack_optional_gemmlowp_deps(),
2785)
2786
2787xnnpack_benchmark(
Frank Barchard40d20fe2020-05-05 00:37:45 -07002788 name = "f16_igemm_bench",
2789 srcs = [
2790 "bench/f16-igemm.cc",
2791 "bench/conv.h",
2792 "bench/google/conv.h",
2793 "src/xnnpack/AlignedAllocator.h",
2794 ] + MICROKERNEL_BENCHMARK_HDRS,
2795 deps = MICROKERNEL_BENCHMARK_DEPS + [":indirection"],
2796)
2797
2798xnnpack_benchmark(
Marat Dukhan08c4a432019-10-03 09:29:21 -07002799 name = "f16_gemm_bench",
2800 srcs = [
2801 "bench/f16-gemm.cc",
2802 "bench/gemm.h",
2803 "src/xnnpack/AlignedAllocator.h",
2804 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_BENCHMARK_HDRS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07002805 deps = MICROKERNEL_BENCHMARK_DEPS,
2806)
2807
2808xnnpack_benchmark(
Marat Dukhanbdb56f52020-02-05 21:42:49 -08002809 name = "f16_spmm_bench",
2810 srcs = [
2811 "bench/f16-spmm.cc",
2812 "bench/gemm.h",
Marat Dukhanbdb56f52020-02-05 21:42:49 -08002813 "src/xnnpack/AlignedAllocator.h",
2814 ] + MICROKERNEL_BENCHMARK_HDRS,
Marat Dukhanbdb56f52020-02-05 21:42:49 -08002815 deps = MICROKERNEL_BENCHMARK_DEPS,
2816)
2817
2818xnnpack_benchmark(
Marat Dukhan08c4a432019-10-03 09:29:21 -07002819 name = "f32_igemm_bench",
2820 srcs = [
2821 "bench/f32-igemm.cc",
2822 "bench/conv.h",
2823 "src/xnnpack/AlignedAllocator.h",
2824 ] + MICROKERNEL_BENCHMARK_HDRS,
Frank Barchard7e955972019-10-11 10:34:25 -07002825 deps = MICROKERNEL_BENCHMARK_DEPS + [":indirection"],
Marat Dukhan08c4a432019-10-03 09:29:21 -07002826)
2827
2828xnnpack_benchmark(
2829 name = "f32_conv_hwc_bench",
2830 srcs = [
2831 "bench/f32-conv-hwc.cc",
2832 "bench/dconv.h",
2833 "src/xnnpack/AlignedAllocator.h",
2834 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_BENCHMARK_HDRS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07002835 deps = MICROKERNEL_BENCHMARK_DEPS,
2836)
2837
2838xnnpack_benchmark(
Marat Dukhan1f29b802020-05-15 23:46:39 -07002839 name = "f32_conv_hwc2chw_bench",
Erich Elsen563df5f2019-10-23 08:02:21 -07002840 srcs = [
Marat Dukhan1f29b802020-05-15 23:46:39 -07002841 "bench/f32-conv-hwc2chw.cc",
Erich Elsen563df5f2019-10-23 08:02:21 -07002842 "bench/dconv.h",
2843 "src/xnnpack/AlignedAllocator.h",
2844 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_BENCHMARK_HDRS,
Erich Elsen563df5f2019-10-23 08:02:21 -07002845 deps = MICROKERNEL_BENCHMARK_DEPS,
2846)
2847
2848xnnpack_benchmark(
Marat Dukhan08c4a432019-10-03 09:29:21 -07002849 name = "f32_dwconv_bench",
2850 srcs = [
2851 "bench/f32-dwconv.cc",
2852 "bench/dwconv.h",
2853 "src/xnnpack/AlignedAllocator.h",
2854 ] + MICROKERNEL_BENCHMARK_HDRS,
2855 deps = MICROKERNEL_BENCHMARK_DEPS + [":indirection"],
2856)
2857
2858xnnpack_benchmark(
Marat Dukhan1f29b802020-05-15 23:46:39 -07002859 name = "f32_dwconv_chw_bench",
Marat Dukhan08c4a432019-10-03 09:29:21 -07002860 srcs = [
Marat Dukhan1f29b802020-05-15 23:46:39 -07002861 "bench/f32-dwconv-chw.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07002862 "bench/dwconv.h",
2863 "src/xnnpack/AlignedAllocator.h",
2864 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_BENCHMARK_HDRS,
2865 deps = MICROKERNEL_BENCHMARK_DEPS + [":indirection"],
2866)
2867
2868xnnpack_benchmark(
2869 name = "f32_gemm_bench",
2870 srcs = [
2871 "bench/f32-gemm.cc",
2872 "bench/gemm.h",
2873 "src/xnnpack/AlignedAllocator.h",
2874 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_BENCHMARK_HDRS,
Marat Dukhan10a38082020-04-17 03:58:35 -07002875 copts = xnnpack_optional_ruy_copts(),
Frank Barchard7e955972019-10-11 10:34:25 -07002876 deps = MICROKERNEL_BENCHMARK_DEPS + xnnpack_optional_ruy_deps(),
Marat Dukhan08c4a432019-10-03 09:29:21 -07002877)
2878
2879xnnpack_benchmark(
Marat Dukhan4c4eb002019-12-08 21:27:49 -08002880 name = "f32_raddexpminusmax_bench",
2881 srcs = [
2882 "bench/f32-raddexpminusmax.cc",
2883 "src/xnnpack/AlignedAllocator.h",
2884 ] + MICROKERNEL_BENCHMARK_HDRS,
2885 deps = MICROKERNEL_BENCHMARK_DEPS,
2886)
2887
2888xnnpack_benchmark(
2889 name = "f32_raddextexp_bench",
2890 srcs = [
2891 "bench/f32-raddextexp.cc",
2892 "src/xnnpack/AlignedAllocator.h",
2893 ] + MICROKERNEL_BENCHMARK_HDRS,
2894 deps = MICROKERNEL_BENCHMARK_DEPS,
2895)
2896
2897xnnpack_benchmark(
2898 name = "f32_raddstoreexpminusmax_bench",
2899 srcs = [
2900 "bench/f32-raddstoreexpminusmax.cc",
2901 "src/xnnpack/AlignedAllocator.h",
2902 ] + MICROKERNEL_BENCHMARK_HDRS,
2903 deps = MICROKERNEL_BENCHMARK_DEPS,
2904)
2905
2906xnnpack_benchmark(
Marat Dukhan08c4a432019-10-03 09:29:21 -07002907 name = "f32_rmax_bench",
2908 srcs = [
2909 "bench/f32-rmax.cc",
2910 "src/xnnpack/AlignedAllocator.h",
2911 ] + MICROKERNEL_BENCHMARK_HDRS,
2912 deps = MICROKERNEL_BENCHMARK_DEPS,
2913)
2914
2915xnnpack_benchmark(
Marat Dukhan14bec502019-11-18 11:35:31 -08002916 name = "f32_sigmoid_bench",
2917 srcs = [
2918 "bench/f32-sigmoid.cc",
2919 "src/xnnpack/AlignedAllocator.h",
2920 ] + MICROKERNEL_BENCHMARK_HDRS,
Marat Dukhan14bec502019-11-18 11:35:31 -08002921 deps = MICROKERNEL_BENCHMARK_DEPS,
2922)
2923
2924xnnpack_benchmark(
Marat Dukhan08c4a432019-10-03 09:29:21 -07002925 name = "f32_spmm_bench",
2926 srcs = [
2927 "bench/f32-spmm.cc",
2928 "bench/gemm.h",
2929 "src/xnnpack/AlignedAllocator.h",
2930 ] + MICROKERNEL_BENCHMARK_HDRS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07002931 deps = MICROKERNEL_BENCHMARK_DEPS,
2932)
2933
2934xnnpack_benchmark(
Marat Dukhanfd8e6892020-01-27 15:25:25 -08002935 name = "f32_softmax_bench",
Marat Dukhan4a4a7fa2019-10-21 13:46:14 -07002936 srcs = [
Marat Dukhanfd8e6892020-01-27 15:25:25 -08002937 "bench/f32-softmax.cc",
Marat Dukhan4a4a7fa2019-10-21 13:46:14 -07002938 ] + MICROKERNEL_BENCHMARK_HDRS,
Marat Dukhan10a38082020-04-17 03:58:35 -07002939 copts = xnnpack_optional_dnnl_copts(),
Marat Dukhan8d3c6932020-03-06 20:27:27 -08002940 deps = MICROKERNEL_BENCHMARK_DEPS + xnnpack_optional_dnnl_deps(),
Marat Dukhan4a4a7fa2019-10-21 13:46:14 -07002941)
2942
2943xnnpack_benchmark(
Marat Dukhan4c4eb002019-12-08 21:27:49 -08002944 name = "f32_vscaleexpminusmax_bench",
2945 srcs = [
2946 "bench/f32-vscaleexpminusmax.cc",
2947 "src/xnnpack/AlignedAllocator.h",
2948 ] + MICROKERNEL_BENCHMARK_HDRS,
2949 deps = MICROKERNEL_BENCHMARK_DEPS,
2950)
2951
2952xnnpack_benchmark(
2953 name = "f32_vscaleextexp_bench",
2954 srcs = [
2955 "bench/f32-vscaleextexp.cc",
2956 "src/xnnpack/AlignedAllocator.h",
2957 ] + MICROKERNEL_BENCHMARK_HDRS,
2958 deps = MICROKERNEL_BENCHMARK_DEPS,
2959)
2960
2961xnnpack_benchmark(
Marat Dukhan08c4a432019-10-03 09:29:21 -07002962 name = "f32_im2col_gemm_bench",
2963 srcs = [
2964 "bench/f32-im2col-gemm.cc",
2965 "bench/conv.h",
2966 "src/xnnpack/AlignedAllocator.h",
2967 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_BENCHMARK_HDRS,
2968 deps = MICROKERNEL_BENCHMARK_DEPS + [":im2col"],
2969)
2970
Marat Dukhanfe7acb62020-03-09 19:30:05 -07002971xnnpack_benchmark(
2972 name = "requantization_bench",
2973 srcs = [
2974 "bench/requantization.cc",
2975 "src/xnnpack/requantization-stubs.h",
2976 "src/xnnpack/AlignedAllocator.h",
2977 ] + MICROKERNEL_BENCHMARK_HDRS,
2978 deps = MICROKERNEL_BENCHMARK_DEPS,
2979)
2980
Marat Dukhanffbf96a2020-05-14 02:59:08 -07002981xnnpack_benchmark(
2982 name = "rounding_bench",
2983 srcs = [
2984 "bench/rounding.cc",
2985 "src/xnnpack/math-stubs.h",
2986 "src/xnnpack/AlignedAllocator.h",
2987 ] + MICROKERNEL_BENCHMARK_HDRS,
2988 deps = MICROKERNEL_BENCHMARK_DEPS,
2989)
2990
Marat Dukhan08c4a432019-10-03 09:29:21 -07002991########################### Benchmarks for operators ###########################
2992
2993xnnpack_benchmark(
2994 name = "add_bench",
2995 srcs = ["bench/add.cc"],
Marat Dukhan1b354632020-03-23 12:50:22 -07002996 deps = OPERATOR_BENCHMARK_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07002997)
2998
2999xnnpack_benchmark(
3000 name = "average_pooling_bench",
3001 srcs = ["bench/average-pooling.cc"],
Marat Dukhan7a16d8b2020-03-11 04:22:44 -07003002 copts = xnnpack_optional_tflite_copts(),
Marat Dukhan8ea0b072020-04-23 16:12:18 -07003003 tags = ["nowin32"],
Marat Dukhan1b354632020-03-23 12:50:22 -07003004 deps = OPERATOR_BENCHMARK_DEPS + xnnpack_optional_tflite_deps(),
Marat Dukhan08c4a432019-10-03 09:29:21 -07003005)
3006
3007xnnpack_benchmark(
3008 name = "channel_shuffle_bench",
3009 srcs = ["bench/channel-shuffle.cc"],
Marat Dukhan1b354632020-03-23 12:50:22 -07003010 deps = OPERATOR_BENCHMARK_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07003011)
3012
3013xnnpack_benchmark(
3014 name = "convolution_bench",
3015 srcs = ["bench/convolution.cc"],
3016 copts = xnnpack_optional_tflite_copts() + xnnpack_optional_armcl_copts(),
Marat Dukhan8ea0b072020-04-23 16:12:18 -07003017 tags = ["nowin32"],
Marat Dukhan1b354632020-03-23 12:50:22 -07003018 deps = OPERATOR_BENCHMARK_DEPS + xnnpack_optional_tflite_deps() + xnnpack_optional_armcl_deps(),
Marat Dukhan08c4a432019-10-03 09:29:21 -07003019)
3020
3021xnnpack_benchmark(
3022 name = "deconvolution_bench",
3023 srcs = ["bench/deconvolution.cc"],
3024 copts = xnnpack_optional_tflite_copts(),
Marat Dukhan8ea0b072020-04-23 16:12:18 -07003025 tags = ["nowin32"],
Marat Dukhan1b354632020-03-23 12:50:22 -07003026 deps = OPERATOR_BENCHMARK_DEPS + xnnpack_optional_tflite_deps(),
Marat Dukhan08c4a432019-10-03 09:29:21 -07003027)
3028
3029xnnpack_benchmark(
3030 name = "global_average_pooling_bench",
3031 srcs = ["bench/global-average-pooling.cc"],
Marat Dukhan1b354632020-03-23 12:50:22 -07003032 deps = OPERATOR_BENCHMARK_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07003033)
3034
3035xnnpack_benchmark(
3036 name = "max_pooling_bench",
3037 srcs = ["bench/max-pooling.cc"],
Marat Dukhan1b354632020-03-23 12:50:22 -07003038 deps = OPERATOR_BENCHMARK_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07003039)
3040
3041xnnpack_benchmark(
3042 name = "sigmoid_bench",
3043 srcs = ["bench/sigmoid.cc"],
Marat Dukhanc3b9e862019-11-17 13:18:54 -08003044 copts = xnnpack_optional_tflite_copts(),
Marat Dukhanca2ba702020-04-24 01:31:47 -07003045 tags = ["nowin32"],
Marat Dukhan1b354632020-03-23 12:50:22 -07003046 deps = OPERATOR_BENCHMARK_DEPS + xnnpack_optional_tflite_deps(),
Marat Dukhan08c4a432019-10-03 09:29:21 -07003047)
3048
3049xnnpack_benchmark(
Marat Dukhan95b22432019-10-30 16:30:14 -07003050 name = "prelu_bench",
3051 srcs = ["bench/prelu.cc"],
3052 copts = xnnpack_optional_tflite_copts(),
Marat Dukhan8ea0b072020-04-23 16:12:18 -07003053 tags = ["nowin32"],
Marat Dukhan1b354632020-03-23 12:50:22 -07003054 deps = OPERATOR_BENCHMARK_DEPS + xnnpack_optional_tflite_deps(),
Marat Dukhan95b22432019-10-30 16:30:14 -07003055)
3056
3057xnnpack_benchmark(
Marat Dukhanfd8e6892020-01-27 15:25:25 -08003058 name = "softmax_bench",
3059 srcs = ["bench/softmax.cc"],
Marat Dukhan9c0db962020-01-28 12:30:14 -08003060 copts = xnnpack_optional_tflite_copts(),
Marat Dukhanca2ba702020-04-24 01:31:47 -07003061 tags = ["nowin32"],
Marat Dukhan1b354632020-03-23 12:50:22 -07003062 deps = OPERATOR_BENCHMARK_DEPS + xnnpack_optional_tflite_deps(),
Marat Dukhan08c4a432019-10-03 09:29:21 -07003063)
3064
Marat Dukhanc068bb62019-10-04 13:24:39 -07003065############################# End-to-end benchmarks ############################
3066
3067cc_library(
3068 name = "mobilenet_v1",
3069 srcs = ["models/mobilenet-v1.cc"],
3070 hdrs = ["models/models.h"],
Marat Dukhana84e40b2019-12-11 15:38:03 -08003071 copts = xnnpack_std_cxxopts(),
Marat Dukhanc068bb62019-10-04 13:24:39 -07003072 linkstatic = True,
3073 deps = [
3074 ":XNNPACK",
3075 "@pthreadpool",
3076 ],
3077)
3078
3079cc_library(
3080 name = "mobilenet_v2",
3081 srcs = ["models/mobilenet-v2.cc"],
3082 hdrs = ["models/models.h"],
Marat Dukhana84e40b2019-12-11 15:38:03 -08003083 copts = xnnpack_std_cxxopts(),
Marat Dukhanc068bb62019-10-04 13:24:39 -07003084 linkstatic = True,
3085 deps = [
3086 ":XNNPACK",
3087 "@pthreadpool",
3088 ],
3089)
3090
Marat Dukhanc08cdf52019-12-09 09:17:51 -08003091cc_library(
3092 name = "mobilenet_v3_large",
3093 srcs = ["models/mobilenet-v3-large.cc"],
3094 hdrs = ["models/models.h"],
Marat Dukhana84e40b2019-12-11 15:38:03 -08003095 copts = xnnpack_std_cxxopts(),
Marat Dukhanc08cdf52019-12-09 09:17:51 -08003096 linkstatic = True,
3097 deps = [
3098 ":XNNPACK",
3099 "@pthreadpool",
3100 ],
3101)
3102
3103cc_library(
3104 name = "mobilenet_v3_small",
3105 srcs = ["models/mobilenet-v3-small.cc"],
3106 hdrs = ["models/models.h"],
Marat Dukhana84e40b2019-12-11 15:38:03 -08003107 copts = xnnpack_std_cxxopts(),
Marat Dukhanc08cdf52019-12-09 09:17:51 -08003108 linkstatic = True,
3109 deps = [
3110 ":XNNPACK",
3111 "@pthreadpool",
3112 ],
3113)
3114
Marat Dukhanc068bb62019-10-04 13:24:39 -07003115xnnpack_benchmark(
Marat Dukhanef4416e2019-10-31 13:44:40 -07003116 name = "f32_dwconv_e2e_bench",
Marat Dukhanc08cdf52019-12-09 09:17:51 -08003117 srcs = [
3118 "bench/f32-dwconv-e2e.cc",
3119 "bench/end2end.h",
3120 ] + MICROKERNEL_BENCHMARK_HDRS,
Marat Dukhanef4416e2019-10-31 13:44:40 -07003121 deps = MICROKERNEL_BENCHMARK_DEPS + [
3122 ":XNNPACK",
3123 ":mobilenet_v1",
3124 ":mobilenet_v2",
Marat Dukhanc08cdf52019-12-09 09:17:51 -08003125 ":mobilenet_v3_large",
3126 ":mobilenet_v3_small",
Marat Dukhanef4416e2019-10-31 13:44:40 -07003127 ],
3128)
3129
3130xnnpack_benchmark(
Marat Dukhan5f18d262019-10-31 10:24:14 -07003131 name = "f32_gemm_e2e_bench",
Marat Dukhanc08cdf52019-12-09 09:17:51 -08003132 srcs = [
3133 "bench/f32-gemm-e2e.cc",
3134 "bench/end2end.h",
3135 ] + MICROKERNEL_BENCHMARK_HDRS,
Marat Dukhan5f18d262019-10-31 10:24:14 -07003136 deps = MICROKERNEL_BENCHMARK_DEPS + [
3137 ":XNNPACK",
3138 ":mobilenet_v1",
3139 ":mobilenet_v2",
Marat Dukhanc08cdf52019-12-09 09:17:51 -08003140 ":mobilenet_v3_large",
3141 ":mobilenet_v3_small",
Marat Dukhan5f18d262019-10-31 10:24:14 -07003142 ],
3143)
3144
3145xnnpack_benchmark(
Marat Dukhanc068bb62019-10-04 13:24:39 -07003146 name = "end2end_bench",
3147 srcs = ["bench/end2end.cc"],
3148 deps = [
3149 ":XNNPACK",
Frank Barchardc712fa42019-10-31 14:00:21 -07003150 ":bench_utils",
Marat Dukhanc068bb62019-10-04 13:24:39 -07003151 ":mobilenet_v1",
3152 ":mobilenet_v2",
Marat Dukhanc08cdf52019-12-09 09:17:51 -08003153 ":mobilenet_v3_large",
3154 ":mobilenet_v3_small",
Marat Dukhanc068bb62019-10-04 13:24:39 -07003155 "@pthreadpool",
3156 ],
3157)
3158
Marat Dukhan6adff4e2019-10-14 18:32:07 -07003159#################### Accuracy evaluation for math functions ####################
3160
3161xnnpack_benchmark(
3162 name = "f32_exp_eval",
3163 srcs = [
3164 "eval/f32-exp.cc",
3165 "src/xnnpack/AlignedAllocator.h",
3166 ] + ACCURACY_EVAL_HDRS,
3167 deps = ACCURACY_EVAL_DEPS,
3168)
3169
Marat Dukhan515c9772019-10-17 18:07:57 -07003170xnnpack_benchmark(
3171 name = "f32_expminus_eval",
3172 srcs = [
3173 "eval/f32-expminus.cc",
3174 "src/xnnpack/AlignedAllocator.h",
3175 ] + ACCURACY_EVAL_HDRS,
3176 deps = ACCURACY_EVAL_DEPS,
3177)
3178
Marat Dukhan98ba4412019-10-23 02:14:28 -07003179xnnpack_benchmark(
3180 name = "f32_extexp_eval",
3181 srcs = [
3182 "eval/f32-extexp.cc",
3183 "src/xnnpack/AlignedAllocator.h",
3184 ] + ACCURACY_EVAL_HDRS,
3185 deps = ACCURACY_EVAL_DEPS,
3186)
3187
Marat Dukhan8853b822020-05-07 12:19:01 -07003188xnnpack_unit_test(
3189 name = "f32_roundne_eval",
3190 srcs = [
3191 "eval/f32-roundne.cc",
3192 "src/xnnpack/AlignedAllocator.h",
3193 "src/xnnpack/math-stubs.h",
3194 ] + MICROKERNEL_TEST_HDRS,
Marat Dukhan22eed3d2020-05-11 20:13:37 -07003195 automatic = False,
Marat Dukhan8853b822020-05-07 12:19:01 -07003196 deps = MICROKERNEL_TEST_DEPS,
3197)
3198
Marat Dukhan2dbb9442020-05-12 20:43:43 -07003199xnnpack_unit_test(
Marat Dukhanc9852ba2020-05-13 17:21:29 -07003200 name = "f32_roundd_eval",
3201 srcs = [
3202 "eval/f32-roundd.cc",
3203 "src/xnnpack/AlignedAllocator.h",
3204 "src/xnnpack/math-stubs.h",
3205 ] + MICROKERNEL_TEST_HDRS,
3206 automatic = False,
3207 deps = MICROKERNEL_TEST_DEPS,
3208)
3209
3210xnnpack_unit_test(
3211 name = "f32_roundu_eval",
3212 srcs = [
3213 "eval/f32-roundu.cc",
3214 "src/xnnpack/AlignedAllocator.h",
3215 "src/xnnpack/math-stubs.h",
3216 ] + MICROKERNEL_TEST_HDRS,
3217 automatic = False,
3218 deps = MICROKERNEL_TEST_DEPS,
3219)
3220
3221xnnpack_unit_test(
Marat Dukhan2dbb9442020-05-12 20:43:43 -07003222 name = "f32_roundz_eval",
3223 srcs = [
3224 "eval/f32-roundz.cc",
3225 "src/xnnpack/AlignedAllocator.h",
3226 "src/xnnpack/math-stubs.h",
3227 ] + MICROKERNEL_TEST_HDRS,
Marat Dukhanc9852ba2020-05-13 17:21:29 -07003228 automatic = False,
Marat Dukhan2dbb9442020-05-12 20:43:43 -07003229 deps = MICROKERNEL_TEST_DEPS,
3230)
3231
Marat Dukhan346a9e52019-11-15 09:06:30 -08003232xnnpack_benchmark(
3233 name = "f32_sigmoid_eval",
3234 srcs = [
3235 "eval/f32-sigmoid.cc",
3236 "src/xnnpack/AlignedAllocator.h",
3237 ] + ACCURACY_EVAL_HDRS,
3238 deps = ACCURACY_EVAL_DEPS,
3239)
3240
Marat Dukhan08c4a432019-10-03 09:29:21 -07003241######################### Unit tests for micro-kernels #########################
3242
3243xnnpack_unit_test(
Marat Dukhande06f492020-04-09 00:19:31 -07003244 name = "f16_gemm_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003245 srcs = [
Marat Dukhande06f492020-04-09 00:19:31 -07003246 "test/f16-gemm-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003247 "test/gemm-microkernel-tester.h",
3248 "src/xnnpack/AlignedAllocator.h",
3249 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3250 deps = MICROKERNEL_TEST_DEPS,
3251)
3252
3253xnnpack_unit_test(
Marat Dukhan355ab432020-04-09 19:01:52 -07003254 name = "f16_spmm_minmax_test",
Marat Dukhanbdb56f52020-02-05 21:42:49 -08003255 srcs = [
Marat Dukhan355ab432020-04-09 19:01:52 -07003256 "test/f16-spmm-minmax.cc",
Marat Dukhanbdb56f52020-02-05 21:42:49 -08003257 "test/spmm-microkernel-tester.h",
3258 "src/xnnpack/AlignedAllocator.h",
3259 ] + MICROKERNEL_TEST_HDRS,
3260 deps = MICROKERNEL_TEST_DEPS,
3261)
3262
3263xnnpack_unit_test(
Marat Dukhan08c4a432019-10-03 09:29:21 -07003264 name = "f32_argmaxpool_test",
3265 srcs = [
3266 "test/f32-argmaxpool.cc",
3267 "test/argmaxpool-microkernel-tester.h",
3268 "src/xnnpack/AlignedAllocator.h",
3269 ] + MICROKERNEL_TEST_HDRS,
3270 deps = MICROKERNEL_TEST_DEPS,
3271)
3272
3273xnnpack_unit_test(
Marat Dukhan99936602020-04-11 16:47:01 -07003274 name = "f32_avgpool_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003275 srcs = [
Marat Dukhan99936602020-04-11 16:47:01 -07003276 "test/f32-avgpool-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003277 "test/avgpool-microkernel-tester.h",
3278 "src/xnnpack/AlignedAllocator.h",
3279 ] + MICROKERNEL_TEST_HDRS,
3280 deps = MICROKERNEL_TEST_DEPS,
3281)
3282
3283xnnpack_unit_test(
Marat Dukhan660fd192020-03-10 04:55:30 -07003284 name = "f32_ibilinear_test",
Marat Dukhan35dacfb2019-11-07 19:18:16 -08003285 srcs = [
Marat Dukhan660fd192020-03-10 04:55:30 -07003286 "test/f32-ibilinear.cc",
3287 "test/ibilinear-microkernel-tester.h",
Marat Dukhan35dacfb2019-11-07 19:18:16 -08003288 "src/xnnpack/AlignedAllocator.h",
3289 ] + MICROKERNEL_TEST_HDRS,
3290 deps = MICROKERNEL_TEST_DEPS,
3291)
3292
3293xnnpack_unit_test(
Frank Barchardb1966592020-05-12 13:47:06 -07003294 name = "f16_clamp_test",
3295 srcs = [
3296 "test/f16-clamp.cc",
3297 "test/clamp-microkernel-tester.h",
3298 ] + MICROKERNEL_TEST_HDRS,
3299 deps = MICROKERNEL_TEST_DEPS,
3300)
3301
3302xnnpack_unit_test(
Marat Dukhan08c4a432019-10-03 09:29:21 -07003303 name = "f32_clamp_test",
3304 srcs = [
3305 "test/f32-clamp.cc",
3306 "test/clamp-microkernel-tester.h",
3307 ] + MICROKERNEL_TEST_HDRS,
3308 deps = MICROKERNEL_TEST_DEPS,
3309)
3310
3311xnnpack_unit_test(
Marat Dukhan163a7e62020-04-09 04:19:26 -07003312 name = "f32_igemm_test",
3313 srcs = [
3314 "test/f32-igemm.cc",
3315 "test/gemm-microkernel-tester.h",
3316 "src/xnnpack/AlignedAllocator.h",
3317 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3318 deps = MICROKERNEL_TEST_DEPS,
3319)
3320
3321xnnpack_unit_test(
Marat Dukhan467f6362020-05-22 23:21:55 -07003322 name = "f32_igemm_relu_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003323 srcs = [
Marat Dukhan467f6362020-05-22 23:21:55 -07003324 "test/f32-igemm-relu.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003325 "test/gemm-microkernel-tester.h",
3326 "src/xnnpack/AlignedAllocator.h",
3327 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3328 deps = MICROKERNEL_TEST_DEPS,
3329)
3330
3331xnnpack_unit_test(
Marat Dukhane207b7b2020-05-28 16:27:42 -07003332 name = "f32_igemm_minmax_test",
3333 srcs = [
3334 "test/f32-igemm-minmax.cc",
3335 "test/gemm-microkernel-tester.h",
3336 "src/xnnpack/AlignedAllocator.h",
3337 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3338 deps = MICROKERNEL_TEST_DEPS,
3339)
3340
3341xnnpack_unit_test(
Frank Barchardb0e4fae2020-05-04 15:27:51 -07003342 name = "f16_igemm_minmax_test",
3343 srcs = [
3344 "test/f16-igemm-minmax.cc",
3345 "test/gemm-microkernel-tester.h",
3346 "src/xnnpack/AlignedAllocator.h",
3347 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3348 deps = MICROKERNEL_TEST_DEPS,
3349)
3350
3351xnnpack_unit_test(
Marat Dukhan08c4a432019-10-03 09:29:21 -07003352 name = "f32_conv_hwc_test",
3353 srcs = [
3354 "test/f32-conv-hwc.cc",
3355 "test/conv-hwc-microkernel-tester.h",
3356 "src/xnnpack/AlignedAllocator.h",
3357 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3358 deps = MICROKERNEL_TEST_DEPS,
3359)
3360
3361xnnpack_unit_test(
Marat Dukhan1f29b802020-05-15 23:46:39 -07003362 name = "f32_conv_hwc2chw_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003363 srcs = [
Marat Dukhan1f29b802020-05-15 23:46:39 -07003364 "test/f32-conv-hwc2chw.cc",
3365 "test/conv-hwc2chw-microkernel-tester.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003366 "src/xnnpack/AlignedAllocator.h",
3367 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3368 deps = MICROKERNEL_TEST_DEPS,
3369)
3370
3371xnnpack_unit_test(
Marat Dukhan163a7e62020-04-09 04:19:26 -07003372 name = "f32_dwconv_test",
3373 srcs = [
3374 "test/f32-dwconv.cc",
3375 "test/dwconv-microkernel-tester.h",
3376 "src/xnnpack/AlignedAllocator.h",
3377 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3378 deps = MICROKERNEL_TEST_DEPS,
3379)
3380
3381xnnpack_unit_test(
Marat Dukhan1c587112020-04-08 20:04:28 -07003382 name = "f32_dwconv_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003383 srcs = [
Marat Dukhan1c587112020-04-08 20:04:28 -07003384 "test/f32-dwconv-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003385 "test/dwconv-microkernel-tester.h",
3386 "src/xnnpack/AlignedAllocator.h",
3387 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3388 deps = MICROKERNEL_TEST_DEPS,
3389)
3390
3391xnnpack_unit_test(
Marat Dukhan1f29b802020-05-15 23:46:39 -07003392 name = "f32_dwconv_chw_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003393 srcs = [
Marat Dukhan1f29b802020-05-15 23:46:39 -07003394 "test/f32-dwconv-chw.cc",
3395 "test/dwconv-chw-microkernel-tester.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003396 "src/xnnpack/AlignedAllocator.h",
3397 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3398 deps = MICROKERNEL_TEST_DEPS,
3399)
3400
3401xnnpack_unit_test(
Marat Dukhan99936602020-04-11 16:47:01 -07003402 name = "f32_gavgpool_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003403 srcs = [
Marat Dukhan99936602020-04-11 16:47:01 -07003404 "test/f32-gavgpool-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003405 "test/gavgpool-microkernel-tester.h",
3406 "src/xnnpack/AlignedAllocator.h",
3407 ] + MICROKERNEL_TEST_HDRS,
3408 deps = MICROKERNEL_TEST_DEPS,
3409)
3410
3411xnnpack_unit_test(
Marat Dukhan1f29b802020-05-15 23:46:39 -07003412 name = "f32_gavgpool_cw_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003413 srcs = [
Marat Dukhan1f29b802020-05-15 23:46:39 -07003414 "test/f32-gavgpool-cw.cc",
3415 "test/gavgpool-cw-microkernel-tester.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003416 "src/xnnpack/AlignedAllocator.h",
3417 ] + MICROKERNEL_TEST_HDRS,
3418 deps = MICROKERNEL_TEST_DEPS,
3419)
3420
3421xnnpack_unit_test(
Marat Dukhan163a7e62020-04-09 04:19:26 -07003422 name = "f32_gemm_test",
3423 srcs = [
3424 "test/f32-gemm.cc",
3425 "test/gemm-microkernel-tester.h",
3426 "src/xnnpack/AlignedAllocator.h",
3427 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3428 deps = MICROKERNEL_TEST_DEPS,
3429)
3430
3431xnnpack_unit_test(
Marat Dukhan467f6362020-05-22 23:21:55 -07003432 name = "f32_gemm_relu_test",
3433 srcs = [
3434 "test/f32-gemm-relu.cc",
3435 "test/gemm-microkernel-tester.h",
3436 "src/xnnpack/AlignedAllocator.h",
3437 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3438 deps = MICROKERNEL_TEST_DEPS,
3439)
3440
3441xnnpack_unit_test(
Marat Dukhan1c587112020-04-08 20:04:28 -07003442 name = "f32_gemm_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003443 srcs = [
Marat Dukhan1c587112020-04-08 20:04:28 -07003444 "test/f32-gemm-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003445 "test/gemm-microkernel-tester.h",
3446 "src/xnnpack/AlignedAllocator.h",
3447 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3448 deps = MICROKERNEL_TEST_DEPS,
3449)
3450
3451xnnpack_unit_test(
Marat Dukhan1c587112020-04-08 20:04:28 -07003452 name = "f32_gemminc_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003453 srcs = [
Marat Dukhan1c587112020-04-08 20:04:28 -07003454 "test/f32-gemminc-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003455 "test/gemm-microkernel-tester.h",
3456 "src/xnnpack/AlignedAllocator.h",
3457 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3458 deps = MICROKERNEL_TEST_DEPS,
3459)
3460
3461xnnpack_unit_test(
Frank Barchardb1966592020-05-12 13:47:06 -07003462 name = "f16_hswish_test",
3463 srcs = [
3464 "test/f16-hswish.cc",
3465 "test/hswish-microkernel-tester.h",
3466 ] + MICROKERNEL_TEST_HDRS,
3467 deps = MICROKERNEL_TEST_DEPS,
3468)
3469
3470xnnpack_unit_test(
Marat Dukhan08c4a432019-10-03 09:29:21 -07003471 name = "f32_hswish_test",
3472 srcs = [
3473 "test/f32-hswish.cc",
3474 "test/hswish-microkernel-tester.h",
3475 ] + MICROKERNEL_TEST_HDRS,
3476 deps = MICROKERNEL_TEST_DEPS,
3477)
3478
3479xnnpack_unit_test(
Marat Dukhan99936602020-04-11 16:47:01 -07003480 name = "f32_maxpool_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003481 srcs = [
Marat Dukhan99936602020-04-11 16:47:01 -07003482 "test/f32-maxpool-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003483 "test/maxpool-microkernel-tester.h",
3484 ] + MICROKERNEL_TEST_HDRS,
3485 deps = MICROKERNEL_TEST_DEPS,
3486)
3487
3488xnnpack_unit_test(
Marat Dukhan99936602020-04-11 16:47:01 -07003489 name = "f32_pavgpool_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003490 srcs = [
Marat Dukhan99936602020-04-11 16:47:01 -07003491 "test/f32-pavgpool-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003492 "test/avgpool-microkernel-tester.h",
3493 "src/xnnpack/AlignedAllocator.h",
3494 ] + MICROKERNEL_TEST_HDRS,
3495 deps = MICROKERNEL_TEST_DEPS,
3496)
3497
3498xnnpack_unit_test(
Marat Dukhan1c587112020-04-08 20:04:28 -07003499 name = "f32_ppmm_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003500 srcs = [
Marat Dukhan1c587112020-04-08 20:04:28 -07003501 "test/f32-ppmm-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003502 "test/gemm-microkernel-tester.h",
3503 "src/xnnpack/AlignedAllocator.h",
3504 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3505 deps = MICROKERNEL_TEST_DEPS,
3506)
3507
3508xnnpack_unit_test(
Frank Barchardb1966592020-05-12 13:47:06 -07003509 name = "f16_prelu_test",
3510 srcs = [
3511 "test/f16-prelu.cc",
3512 "test/prelu-microkernel-tester.h",
3513 "src/xnnpack/AlignedAllocator.h",
3514 ] + MICROKERNEL_TEST_HDRS,
3515 deps = MICROKERNEL_TEST_DEPS,
3516)
3517
3518xnnpack_unit_test(
Marat Dukhan08c4a432019-10-03 09:29:21 -07003519 name = "f32_prelu_test",
3520 srcs = [
3521 "test/f32-prelu.cc",
3522 "test/prelu-microkernel-tester.h",
3523 "src/xnnpack/AlignedAllocator.h",
3524 ] + MICROKERNEL_TEST_HDRS,
3525 deps = MICROKERNEL_TEST_DEPS,
3526)
3527
3528xnnpack_unit_test(
Marat Dukhan97579532019-10-18 16:40:39 -07003529 name = "f32_raddexpminusmax_test",
3530 srcs = [
3531 "test/f32-raddexpminusmax.cc",
3532 "test/raddexpminusmax-microkernel-tester.h",
3533 ] + MICROKERNEL_TEST_HDRS,
3534 deps = MICROKERNEL_TEST_DEPS,
3535)
3536
3537xnnpack_unit_test(
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07003538 name = "f32_raddextexp_test",
3539 srcs = [
3540 "test/f32-raddextexp.cc",
3541 "test/raddextexp-microkernel-tester.h",
3542 ] + MICROKERNEL_TEST_HDRS,
3543 deps = MICROKERNEL_TEST_DEPS,
3544)
3545
3546xnnpack_unit_test(
Marat Dukhan97579532019-10-18 16:40:39 -07003547 name = "f32_raddstoreexpminusmax_test",
3548 srcs = [
3549 "test/f32-raddstoreexpminusmax.cc",
3550 "test/raddstoreexpminusmax-microkernel-tester.h",
3551 ] + MICROKERNEL_TEST_HDRS,
3552 deps = MICROKERNEL_TEST_DEPS,
3553)
3554
3555xnnpack_unit_test(
Marat Dukhan08c4a432019-10-03 09:29:21 -07003556 name = "f32_rmax_test",
3557 srcs = [
3558 "test/f32-rmax.cc",
3559 "test/rmax-microkernel-tester.h",
3560 ] + MICROKERNEL_TEST_HDRS,
3561 deps = MICROKERNEL_TEST_DEPS,
3562)
3563
3564xnnpack_unit_test(
Marat Dukhan346a9e52019-11-15 09:06:30 -08003565 name = "f32_sigmoid_test",
3566 srcs = [
3567 "test/f32-sigmoid.cc",
Marat Dukhan1e782c42019-11-21 17:02:40 -08003568 "test/vunary-microkernel-tester.h",
Marat Dukhan346a9e52019-11-15 09:06:30 -08003569 ] + MICROKERNEL_TEST_HDRS,
3570 deps = MICROKERNEL_TEST_DEPS,
3571)
3572
3573xnnpack_unit_test(
Marat Dukhan355ab432020-04-09 19:01:52 -07003574 name = "f32_spmm_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003575 srcs = [
Marat Dukhan355ab432020-04-09 19:01:52 -07003576 "test/f32-spmm-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003577 "test/spmm-microkernel-tester.h",
3578 "src/xnnpack/AlignedAllocator.h",
3579 ] + MICROKERNEL_TEST_HDRS,
3580 deps = MICROKERNEL_TEST_DEPS,
3581)
3582
3583xnnpack_unit_test(
Frank Barchardd793f6c2020-05-08 13:37:43 -07003584 name = "f16_vadd_minmax_test",
3585 srcs = [
3586 "test/f16-vadd-minmax.cc",
3587 "test/vbinary-microkernel-tester.h",
3588 ] + MICROKERNEL_TEST_HDRS,
3589 deps = MICROKERNEL_TEST_DEPS,
3590)
3591
3592xnnpack_unit_test(
3593 name = "f16_vaddc_minmax_test",
3594 srcs = [
3595 "test/f16-vaddc-minmax.cc",
3596 "test/vbinaryc-microkernel-tester.h",
3597 ] + MICROKERNEL_TEST_HDRS,
3598 deps = MICROKERNEL_TEST_DEPS,
3599)
3600
3601xnnpack_unit_test(
3602 name = "f16_vdiv_minmax_test",
3603 srcs = [
3604 "test/f16-vdiv-minmax.cc",
3605 "test/vbinary-microkernel-tester.h",
3606 ] + MICROKERNEL_TEST_HDRS,
3607 deps = MICROKERNEL_TEST_DEPS,
3608)
3609
3610xnnpack_unit_test(
3611 name = "f16_vdivc_minmax_test",
3612 srcs = [
3613 "test/f16-vdivc-minmax.cc",
3614 "test/vbinaryc-microkernel-tester.h",
3615 ] + MICROKERNEL_TEST_HDRS,
3616 deps = MICROKERNEL_TEST_DEPS,
3617)
3618
3619xnnpack_unit_test(
3620 name = "f16_vrdivc_minmax_test",
3621 srcs = [
3622 "test/f16-vrdivc-minmax.cc",
3623 "test/vbinaryc-microkernel-tester.h",
3624 ] + MICROKERNEL_TEST_HDRS,
3625 deps = MICROKERNEL_TEST_DEPS,
3626)
3627
3628xnnpack_unit_test(
3629 name = "f16_vmax_test",
3630 srcs = [
3631 "test/f16-vmax.cc",
3632 "test/vbinary-microkernel-tester.h",
3633 ] + MICROKERNEL_TEST_HDRS,
3634 deps = MICROKERNEL_TEST_DEPS,
3635)
3636
3637xnnpack_unit_test(
3638 name = "f16_vmaxc_test",
3639 srcs = [
3640 "test/f16-vmaxc.cc",
3641 "test/vbinaryc-microkernel-tester.h",
3642 ] + MICROKERNEL_TEST_HDRS,
3643 deps = MICROKERNEL_TEST_DEPS,
3644)
3645
3646xnnpack_unit_test(
3647 name = "f16_vmin_test",
3648 srcs = [
3649 "test/f16-vmin.cc",
3650 "test/vbinary-microkernel-tester.h",
3651 ] + MICROKERNEL_TEST_HDRS,
3652 deps = MICROKERNEL_TEST_DEPS,
3653)
3654
3655xnnpack_unit_test(
3656 name = "f16_vminc_test",
3657 srcs = [
3658 "test/f16-vminc.cc",
3659 "test/vbinaryc-microkernel-tester.h",
3660 ] + MICROKERNEL_TEST_HDRS,
3661 deps = MICROKERNEL_TEST_DEPS,
3662)
3663
3664xnnpack_unit_test(
3665 name = "f16_vmul_minmax_test",
3666 srcs = [
3667 "test/f16-vmul-minmax.cc",
3668 "test/vbinary-microkernel-tester.h",
3669 ] + MICROKERNEL_TEST_HDRS,
3670 deps = MICROKERNEL_TEST_DEPS,
3671)
3672
3673xnnpack_unit_test(
3674 name = "f16_vmulc_minmax_test",
3675 srcs = [
3676 "test/f16-vmulc-minmax.cc",
3677 "test/vbinaryc-microkernel-tester.h",
3678 ] + MICROKERNEL_TEST_HDRS,
3679 deps = MICROKERNEL_TEST_DEPS,
3680)
3681
3682xnnpack_unit_test(
3683 name = "f16_vsub_minmax_test",
3684 srcs = [
3685 "test/f16-vsub-minmax.cc",
3686 "test/vbinary-microkernel-tester.h",
3687 ] + MICROKERNEL_TEST_HDRS,
3688 deps = MICROKERNEL_TEST_DEPS,
3689)
3690
3691xnnpack_unit_test(
3692 name = "f16_vsubc_minmax_test",
3693 srcs = [
3694 "test/f16-vsubc-minmax.cc",
3695 "test/vbinaryc-microkernel-tester.h",
3696 ] + MICROKERNEL_TEST_HDRS,
3697 deps = MICROKERNEL_TEST_DEPS,
3698)
3699
3700xnnpack_unit_test(
3701 name = "f16_vrsubc_minmax_test",
3702 srcs = [
3703 "test/f16-vrsubc-minmax.cc",
3704 "test/vbinaryc-microkernel-tester.h",
3705 ] + MICROKERNEL_TEST_HDRS,
3706 deps = MICROKERNEL_TEST_DEPS,
3707)
3708
3709xnnpack_unit_test(
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003710 name = "f32_vadd_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003711 srcs = [
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003712 "test/f32-vadd-minmax.cc",
Marat Dukhan1e782c42019-11-21 17:02:40 -08003713 "test/vbinary-microkernel-tester.h",
Marat Dukhanc07cb7f2019-11-14 15:32:05 -08003714 ] + MICROKERNEL_TEST_HDRS,
3715 deps = MICROKERNEL_TEST_DEPS,
3716)
3717
3718xnnpack_unit_test(
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003719 name = "f32_vaddc_minmax_test",
Marat Dukhanc07cb7f2019-11-14 15:32:05 -08003720 srcs = [
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003721 "test/f32-vaddc-minmax.cc",
Marat Dukhan1e782c42019-11-21 17:02:40 -08003722 "test/vbinaryc-microkernel-tester.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003723 ] + MICROKERNEL_TEST_HDRS,
3724 deps = MICROKERNEL_TEST_DEPS,
3725)
3726
3727xnnpack_unit_test(
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003728 name = "f32_vdiv_minmax_test",
Marat Dukhan77ca6302019-12-06 12:48:15 -08003729 srcs = [
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003730 "test/f32-vdiv-minmax.cc",
Marat Dukhan77ca6302019-12-06 12:48:15 -08003731 "test/vbinary-microkernel-tester.h",
3732 ] + MICROKERNEL_TEST_HDRS,
3733 deps = MICROKERNEL_TEST_DEPS,
3734)
3735
3736xnnpack_unit_test(
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003737 name = "f32_vdivc_minmax_test",
Marat Dukhan77ca6302019-12-06 12:48:15 -08003738 srcs = [
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003739 "test/f32-vdivc-minmax.cc",
Marat Dukhan77ca6302019-12-06 12:48:15 -08003740 "test/vbinaryc-microkernel-tester.h",
3741 ] + MICROKERNEL_TEST_HDRS,
3742 deps = MICROKERNEL_TEST_DEPS,
3743)
3744
3745xnnpack_unit_test(
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003746 name = "f32_vrdivc_minmax_test",
Marat Dukhan77ca6302019-12-06 12:48:15 -08003747 srcs = [
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003748 "test/f32-vrdivc-minmax.cc",
Marat Dukhan77ca6302019-12-06 12:48:15 -08003749 "test/vbinaryc-microkernel-tester.h",
3750 ] + MICROKERNEL_TEST_HDRS,
3751 deps = MICROKERNEL_TEST_DEPS,
3752)
3753
3754xnnpack_unit_test(
Marat Dukhan403b7d42019-12-05 12:49:11 -08003755 name = "f32_vmax_test",
3756 srcs = [
3757 "test/f32-vmax.cc",
3758 "test/vbinary-microkernel-tester.h",
3759 ] + MICROKERNEL_TEST_HDRS,
3760 deps = MICROKERNEL_TEST_DEPS,
3761)
3762
3763xnnpack_unit_test(
3764 name = "f32_vmaxc_test",
3765 srcs = [
3766 "test/f32-vmaxc.cc",
3767 "test/vbinaryc-microkernel-tester.h",
3768 ] + MICROKERNEL_TEST_HDRS,
3769 deps = MICROKERNEL_TEST_DEPS,
3770)
3771
3772xnnpack_unit_test(
3773 name = "f32_vmin_test",
3774 srcs = [
3775 "test/f32-vmin.cc",
3776 "test/vbinary-microkernel-tester.h",
3777 ] + MICROKERNEL_TEST_HDRS,
3778 deps = MICROKERNEL_TEST_DEPS,
3779)
3780
3781xnnpack_unit_test(
3782 name = "f32_vminc_test",
3783 srcs = [
3784 "test/f32-vminc.cc",
3785 "test/vbinaryc-microkernel-tester.h",
3786 ] + MICROKERNEL_TEST_HDRS,
3787 deps = MICROKERNEL_TEST_DEPS,
3788)
3789
3790xnnpack_unit_test(
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003791 name = "f32_vmul_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003792 srcs = [
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003793 "test/f32-vmul-minmax.cc",
Marat Dukhan1e782c42019-11-21 17:02:40 -08003794 "test/vbinary-microkernel-tester.h",
Marat Dukhanc07cb7f2019-11-14 15:32:05 -08003795 ] + MICROKERNEL_TEST_HDRS,
3796 deps = MICROKERNEL_TEST_DEPS,
3797)
3798
3799xnnpack_unit_test(
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003800 name = "f32_vmulc_minmax_test",
Marat Dukhanc07cb7f2019-11-14 15:32:05 -08003801 srcs = [
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003802 "test/f32-vmulc-minmax.cc",
Marat Dukhan1e782c42019-11-21 17:02:40 -08003803 "test/vbinaryc-microkernel-tester.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003804 ] + MICROKERNEL_TEST_HDRS,
3805 deps = MICROKERNEL_TEST_DEPS,
3806)
3807
3808xnnpack_unit_test(
Marat Dukhan99936602020-04-11 16:47:01 -07003809 name = "f32_vmulcaddc_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003810 srcs = [
Marat Dukhan99936602020-04-11 16:47:01 -07003811 "test/f32-vmulcaddc-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003812 "test/vmulcaddc-microkernel-tester.h",
3813 "src/xnnpack/AlignedAllocator.h",
3814 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3815 deps = MICROKERNEL_TEST_DEPS,
3816)
3817
3818xnnpack_unit_test(
Marat Dukhan05ac8e32019-10-21 15:39:33 -07003819 name = "f32_vscale_test",
3820 srcs = [
3821 "test/f32-vscale.cc",
3822 "test/vscale-microkernel-tester.h",
3823 ] + MICROKERNEL_TEST_HDRS,
3824 deps = MICROKERNEL_TEST_DEPS,
3825)
3826
3827xnnpack_unit_test(
Marat Dukhan97579532019-10-18 16:40:39 -07003828 name = "f32_vscaleexpminusmax_test",
3829 srcs = [
3830 "test/f32-vscaleexpminusmax.cc",
3831 "test/vscaleexpminusmax-microkernel-tester.h",
3832 ] + MICROKERNEL_TEST_HDRS,
3833 deps = MICROKERNEL_TEST_DEPS,
3834)
3835
3836xnnpack_unit_test(
Marat Dukhan6f8d4d32019-10-25 17:07:09 -07003837 name = "f32_vscaleextexp_test",
3838 srcs = [
3839 "test/f32-vscaleextexp.cc",
3840 "test/vscaleextexp-microkernel-tester.h",
3841 ] + MICROKERNEL_TEST_HDRS,
3842 deps = MICROKERNEL_TEST_DEPS,
3843)
3844
3845xnnpack_unit_test(
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003846 name = "f32_vsub_minmax_test",
Marat Dukhan97579532019-10-18 16:40:39 -07003847 srcs = [
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003848 "test/f32-vsub-minmax.cc",
Marat Dukhan1e782c42019-11-21 17:02:40 -08003849 "test/vbinary-microkernel-tester.h",
Marat Dukhanc07cb7f2019-11-14 15:32:05 -08003850 ] + MICROKERNEL_TEST_HDRS,
3851 deps = MICROKERNEL_TEST_DEPS,
3852)
3853
3854xnnpack_unit_test(
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003855 name = "f32_vsubc_minmax_test",
Marat Dukhanc07cb7f2019-11-14 15:32:05 -08003856 srcs = [
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003857 "test/f32-vsubc-minmax.cc",
Marat Dukhan1e782c42019-11-21 17:02:40 -08003858 "test/vbinaryc-microkernel-tester.h",
Marat Dukhanc07cb7f2019-11-14 15:32:05 -08003859 ] + MICROKERNEL_TEST_HDRS,
3860 deps = MICROKERNEL_TEST_DEPS,
3861)
3862
3863xnnpack_unit_test(
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003864 name = "f32_vrsubc_minmax_test",
Marat Dukhanc07cb7f2019-11-14 15:32:05 -08003865 srcs = [
Marat Dukhan91cd2b72020-04-09 23:57:31 -07003866 "test/f32-vrsubc-minmax.cc",
Marat Dukhan1e782c42019-11-21 17:02:40 -08003867 "test/vbinaryc-microkernel-tester.h",
Marat Dukhan97579532019-10-18 16:40:39 -07003868 ] + MICROKERNEL_TEST_HDRS,
3869 deps = MICROKERNEL_TEST_DEPS,
3870)
3871
3872xnnpack_unit_test(
Marat Dukhan99936602020-04-11 16:47:01 -07003873 name = "q8_avgpool_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003874 srcs = [
Marat Dukhan99936602020-04-11 16:47:01 -07003875 "test/q8-avgpool-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003876 "test/avgpool-microkernel-tester.h",
3877 "src/xnnpack/AlignedAllocator.h",
3878 ] + MICROKERNEL_TEST_HDRS,
3879 deps = MICROKERNEL_TEST_DEPS,
3880)
3881
3882xnnpack_unit_test(
Marat Dukhande06f492020-04-09 00:19:31 -07003883 name = "q8_igemm_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003884 srcs = [
Marat Dukhande06f492020-04-09 00:19:31 -07003885 "test/q8-igemm-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003886 "test/gemm-microkernel-tester.h",
3887 "src/xnnpack/AlignedAllocator.h",
3888 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3889 deps = MICROKERNEL_TEST_DEPS,
3890)
3891
3892xnnpack_unit_test(
Marat Dukhande06f492020-04-09 00:19:31 -07003893 name = "q8_dwconv_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003894 srcs = [
Marat Dukhande06f492020-04-09 00:19:31 -07003895 "test/q8-dwconv-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003896 "test/dwconv-microkernel-tester.h",
3897 "src/xnnpack/AlignedAllocator.h",
3898 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3899 deps = MICROKERNEL_TEST_DEPS,
3900)
3901
3902xnnpack_unit_test(
Marat Dukhan99936602020-04-11 16:47:01 -07003903 name = "q8_gavgpool_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003904 srcs = [
Marat Dukhan99936602020-04-11 16:47:01 -07003905 "test/q8-gavgpool-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003906 "test/gavgpool-microkernel-tester.h",
3907 "src/xnnpack/AlignedAllocator.h",
3908 ] + MICROKERNEL_TEST_HDRS,
3909 deps = MICROKERNEL_TEST_DEPS,
3910)
3911
3912xnnpack_unit_test(
Marat Dukhande06f492020-04-09 00:19:31 -07003913 name = "q8_gemm_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003914 srcs = [
Marat Dukhande06f492020-04-09 00:19:31 -07003915 "test/q8-gemm-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003916 "test/gemm-microkernel-tester.h",
3917 "src/xnnpack/AlignedAllocator.h",
3918 ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS,
3919 deps = MICROKERNEL_TEST_DEPS,
3920)
3921
3922xnnpack_unit_test(
Marat Dukhan99936602020-04-11 16:47:01 -07003923 name = "q8_vadd_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003924 srcs = [
Marat Dukhan99936602020-04-11 16:47:01 -07003925 "test/q8-vadd-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003926 "test/vadd-microkernel-tester.h",
3927 ] + MICROKERNEL_TEST_HDRS,
3928 deps = MICROKERNEL_TEST_DEPS,
3929)
3930
3931xnnpack_unit_test(
3932 name = "u8_clamp_test",
3933 srcs = [
3934 "test/u8-clamp.cc",
3935 "test/clamp-microkernel-tester.h",
3936 ] + MICROKERNEL_TEST_HDRS,
3937 deps = MICROKERNEL_TEST_DEPS,
3938)
3939
3940xnnpack_unit_test(
3941 name = "u8_lut32norm_test",
3942 srcs = [
3943 "test/u8-lut32norm.cc",
3944 "test/lut-norm-microkernel-tester.h",
3945 ] + MICROKERNEL_TEST_HDRS,
3946 deps = MICROKERNEL_TEST_DEPS,
3947)
3948
3949xnnpack_unit_test(
Marat Dukhan99936602020-04-11 16:47:01 -07003950 name = "u8_maxpool_minmax_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003951 srcs = [
Marat Dukhan99936602020-04-11 16:47:01 -07003952 "test/u8-maxpool-minmax.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07003953 "test/maxpool-microkernel-tester.h",
3954 ] + MICROKERNEL_TEST_HDRS,
3955 deps = MICROKERNEL_TEST_DEPS,
3956)
3957
3958xnnpack_unit_test(
3959 name = "u8_rmax_test",
3960 srcs = [
3961 "test/u8-rmax.cc",
3962 "test/rmax-microkernel-tester.h",
3963 ] + MICROKERNEL_TEST_HDRS,
3964 deps = MICROKERNEL_TEST_DEPS,
3965)
3966
3967xnnpack_unit_test(
Marat Dukhan3bb3bfc2020-05-19 17:42:46 -07003968 name = "x32_fill_test",
3969 srcs = [
3970 "test/x32-fill.cc",
3971 "test/fill-microkernel-tester.h",
3972 ] + MICROKERNEL_TEST_HDRS,
3973 deps = MICROKERNEL_TEST_DEPS,
3974)
3975
3976xnnpack_unit_test(
Marat Dukhan08c4a432019-10-03 09:29:21 -07003977 name = "x32_packx_test",
3978 srcs = [
3979 "test/x32-packx.cc",
3980 "test/pack-microkernel-tester.h",
3981 "src/xnnpack/AlignedAllocator.h",
3982 ] + MICROKERNEL_TEST_HDRS,
3983 deps = MICROKERNEL_TEST_DEPS,
3984)
3985
3986xnnpack_unit_test(
3987 name = "x32_pad_test",
3988 srcs = [
3989 "test/x32-pad.cc",
3990 "test/pad-microkernel-tester.h",
3991 ] + MICROKERNEL_TEST_HDRS,
3992 deps = MICROKERNEL_TEST_DEPS,
3993)
3994
3995xnnpack_unit_test(
3996 name = "x32_unpool_test",
3997 srcs = [
3998 "test/x32-unpool.cc",
3999 "test/unpool-microkernel-tester.h",
4000 ] + MICROKERNEL_TEST_HDRS,
4001 deps = MICROKERNEL_TEST_DEPS,
4002)
4003
4004xnnpack_unit_test(
4005 name = "x32_zip_test",
4006 srcs = [
4007 "test/x32-zip.cc",
4008 "test/zip-microkernel-tester.h",
4009 ] + MICROKERNEL_TEST_HDRS,
4010 deps = MICROKERNEL_TEST_DEPS,
4011)
4012
4013xnnpack_unit_test(
4014 name = "x8_lut_test",
4015 srcs = [
4016 "test/x8-lut.cc",
4017 "test/lut-microkernel-tester.h",
4018 ] + MICROKERNEL_TEST_HDRS,
4019 deps = MICROKERNEL_TEST_DEPS,
4020)
4021
4022xnnpack_unit_test(
4023 name = "x8_zip_test",
4024 srcs = [
4025 "test/x8-zip.cc",
4026 "test/zip-microkernel-tester.h",
4027 ] + MICROKERNEL_TEST_HDRS,
4028 deps = MICROKERNEL_TEST_DEPS,
4029)
4030
Marat Dukhanfe7acb62020-03-09 19:30:05 -07004031xnnpack_unit_test(
4032 name = "requantization_test",
4033 srcs = [
4034 "src/xnnpack/requantization-stubs.h",
4035 "test/requantization.cc",
4036 "test/requantization-tester.h",
4037 ] + MICROKERNEL_TEST_HDRS,
4038 deps = MICROKERNEL_TEST_DEPS,
4039)
4040
Marat Dukhan20c3b922020-03-10 03:45:06 -07004041########################## Size tests for the library #########################
Marat Dukhan08c4a432019-10-03 09:29:21 -07004042
4043xnnpack_binary(
Marat Dukhan20c3b922020-03-10 03:45:06 -07004044 name = "operator_size_test",
4045 srcs = ["test/operator-size.c"],
Marat Dukhan8fe54e42019-10-10 14:12:59 -07004046 deps = [":xnnpack_operators_nhwc_f32"],
Marat Dukhan08c4a432019-10-03 09:29:21 -07004047)
4048
Marat Dukhan20c3b922020-03-10 03:45:06 -07004049xnnpack_binary(
4050 name = "subgraph_size_test",
4051 srcs = ["test/subgraph-size.c"],
4052 deps = [":XNNPACK"],
4053)
4054
4055########################### Unit tests for operators ##########################
Marat Dukhan08c4a432019-10-03 09:29:21 -07004056
4057xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004058 name = "add_nc_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004059 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004060 "test/add-nc.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004061 "test/add-operator-tester.h",
4062 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004063 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004064)
4065
4066xnnpack_unit_test(
Marat Dukhanb1a0fc32019-12-02 19:32:02 -08004067 name = "add_nd_test",
4068 srcs = [
4069 "test/add-nd.cc",
4070 "test/binary-elementwise-operator-tester.h",
4071 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004072 deps = OPERATOR_TEST_DEPS,
Marat Dukhanb1a0fc32019-12-02 19:32:02 -08004073)
4074
4075xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004076 name = "argmax_pooling_nhwc_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004077 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004078 "test/argmax-pooling-nhwc.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004079 "test/argmax-pooling-operator-tester.h",
4080 ] + OPERATOR_TEST_PARAMS_HDRS,
Marat Dukhan1b354632020-03-23 12:50:22 -07004081 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004082)
4083
4084xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004085 name = "average_pooling_nhwc_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004086 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004087 "test/average-pooling-nhwc.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004088 "test/average-pooling-operator-tester.h",
4089 ] + OPERATOR_TEST_PARAMS_HDRS,
Marat Dukhan1b354632020-03-23 12:50:22 -07004090 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004091)
4092
4093xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004094 name = "channel_pad_nc_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004095 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004096 "test/channel-pad-nc.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004097 "test/channel-pad-operator-tester.h",
4098 ] + OPERATOR_TEST_PARAMS_HDRS,
Marat Dukhan1b354632020-03-23 12:50:22 -07004099 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004100)
4101
4102xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004103 name = "channel_shuffle_nc_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004104 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004105 "test/channel-shuffle-nc.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004106 "test/channel-shuffle-operator-tester.h",
4107 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004108 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004109)
4110
4111xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004112 name = "clamp_nc_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004113 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004114 "test/clamp-nc.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004115 "test/clamp-operator-tester.h",
4116 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004117 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004118)
4119
4120xnnpack_unit_test(
Marat Dukhan065b11e2020-05-22 09:49:41 -07004121 name = "constant_pad_nd_test",
4122 srcs = [
4123 "test/constant-pad-nd.cc",
4124 "test/constant-pad-operator-tester.h",
4125 ],
4126 deps = OPERATOR_TEST_DEPS,
4127)
4128
4129xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004130 name = "convolution_nhwc_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004131 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004132 "test/convolution-nhwc.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004133 "test/convolution-operator-tester.h",
4134 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004135 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004136)
4137
4138xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004139 name = "convolution_nchw_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004140 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004141 "test/convolution-nchw.cc",
4142 "test/convolution-operator-tester.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004143 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004144 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004145)
4146
4147xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004148 name = "deconvolution_nhwc_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004149 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004150 "test/deconvolution-nhwc.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004151 "test/deconvolution-operator-tester.h",
4152 ] + OPERATOR_TEST_PARAMS_HDRS,
Marat Dukhan1b354632020-03-23 12:50:22 -07004153 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004154)
4155
4156xnnpack_unit_test(
Marat Dukhan69180502019-12-06 15:00:31 -08004157 name = "divide_nd_test",
4158 srcs = [
4159 "test/binary-elementwise-operator-tester.h",
4160 "test/divide-nd.cc",
4161 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004162 deps = OPERATOR_TEST_DEPS,
Marat Dukhan69180502019-12-06 15:00:31 -08004163)
4164
4165xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004166 name = "fully_connected_nc_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004167 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004168 "test/fully-connected-nc.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004169 "test/fully-connected-operator-tester.h",
4170 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004171 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004172)
4173
4174xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004175 name = "global_average_pooling_nwc_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004176 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004177 "test/global-average-pooling-nwc.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004178 "test/global-average-pooling-operator-tester.h",
4179 ] + OPERATOR_TEST_PARAMS_HDRS,
Marat Dukhan1b354632020-03-23 12:50:22 -07004180 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004181)
4182
4183xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004184 name = "global_average_pooling_ncw_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004185 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004186 "test/global-average-pooling-ncw.cc",
4187 "test/global-average-pooling-operator-tester.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004188 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004189 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004190)
4191
4192xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004193 name = "hardswish_nc_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004194 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004195 "test/hardswish-nc.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004196 "test/hardswish-operator-tester.h",
4197 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004198 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004199)
4200
4201xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004202 name = "leaky_relu_nc_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004203 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004204 "test/leaky-relu-nc.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004205 "test/leaky-relu-operator-tester.h",
4206 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004207 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004208)
4209
4210xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004211 name = "max_pooling_nhwc_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004212 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004213 "test/max-pooling-nhwc.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004214 "test/max-pooling-operator-tester.h",
4215 ] + OPERATOR_TEST_PARAMS_HDRS,
Marat Dukhan1b354632020-03-23 12:50:22 -07004216 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004217)
4218
4219xnnpack_unit_test(
Marat Dukhan79e7f842019-12-05 14:35:50 -08004220 name = "maximum_nd_test",
4221 srcs = [
4222 "test/binary-elementwise-operator-tester.h",
4223 "test/maximum-nd.cc",
4224 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004225 deps = OPERATOR_TEST_DEPS,
Marat Dukhan79e7f842019-12-05 14:35:50 -08004226)
4227
4228xnnpack_unit_test(
4229 name = "minimum_nd_test",
4230 srcs = [
4231 "test/binary-elementwise-operator-tester.h",
4232 "test/minimum-nd.cc",
4233 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004234 deps = OPERATOR_TEST_DEPS,
Marat Dukhan79e7f842019-12-05 14:35:50 -08004235)
4236
4237xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004238 name = "multiply_nd_test",
Marat Dukhanca2733c2019-11-15 23:21:17 -08004239 srcs = [
Marat Dukhanb1a0fc32019-12-02 19:32:02 -08004240 "test/binary-elementwise-operator-tester.h",
Marat Dukhanefc47b82019-11-18 09:25:38 -08004241 "test/multiply-nd.cc",
Marat Dukhanca2733c2019-11-15 23:21:17 -08004242 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004243 deps = OPERATOR_TEST_DEPS,
Marat Dukhanca2733c2019-11-15 23:21:17 -08004244)
4245
4246xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004247 name = "prelu_nc_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004248 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004249 "test/prelu-nc.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004250 "test/prelu-operator-tester.h",
4251 ] + OPERATOR_TEST_PARAMS_HDRS,
Marat Dukhan1b354632020-03-23 12:50:22 -07004252 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004253)
4254
4255xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004256 name = "resize_bilinear_nhwc_test",
Marat Dukhan69722492019-11-11 19:55:50 -08004257 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004258 "test/resize-bilinear-nhwc.cc",
Marat Dukhan69722492019-11-11 19:55:50 -08004259 "test/resize-bilinear-operator-tester.h",
4260 ] + OPERATOR_TEST_PARAMS_HDRS,
Marat Dukhan1b354632020-03-23 12:50:22 -07004261 deps = OPERATOR_TEST_DEPS,
Marat Dukhan69722492019-11-11 19:55:50 -08004262)
4263
4264xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004265 name = "sigmoid_nc_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004266 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004267 "test/sigmoid-nc.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004268 "test/sigmoid-operator-tester.h",
4269 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004270 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004271)
4272
4273xnnpack_unit_test(
Marat Dukhanfd8e6892020-01-27 15:25:25 -08004274 name = "softmax_nc_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004275 srcs = [
Marat Dukhanfd8e6892020-01-27 15:25:25 -08004276 "test/softmax-nc.cc",
4277 "test/softmax-operator-tester.h",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004278 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004279 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004280)
4281
4282xnnpack_unit_test(
Marat Dukhan05f3f6d2019-12-03 15:13:53 -08004283 name = "subtract_nd_test",
4284 srcs = [
4285 "test/binary-elementwise-operator-tester.h",
4286 "test/subtract-nd.cc",
4287 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004288 deps = OPERATOR_TEST_DEPS,
Marat Dukhan05f3f6d2019-12-03 15:13:53 -08004289)
4290
4291xnnpack_unit_test(
Marat Dukhanefc47b82019-11-18 09:25:38 -08004292 name = "unpooling_nhwc_test",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004293 srcs = [
Marat Dukhanefc47b82019-11-18 09:25:38 -08004294 "test/unpooling-nhwc.cc",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004295 "test/unpooling-operator-tester.h",
4296 ],
Marat Dukhan1b354632020-03-23 12:50:22 -07004297 deps = OPERATOR_TEST_DEPS,
Marat Dukhan08c4a432019-10-03 09:29:21 -07004298)
4299
Chao Mei6ddfc602020-05-13 22:29:36 -07004300############################### Misc unit tests ###############################
4301
4302xnnpack_unit_test(
4303 name = "memory_planner_test",
4304 srcs = [
4305 "test/memory-planner-test.cc",
4306 ],
4307 deps = [
4308 ":XNNPACK",
4309 ":memory_planner",
4310 ],
4311)
4312
Marat Dukhan08c4a432019-10-03 09:29:21 -07004313############################# Build configurations #############################
4314
Marat Dukhanb8642352019-10-30 15:43:02 -07004315# Enables usage of assembly kernels.
Marat Dukhan08c4a432019-10-03 09:29:21 -07004316config_setting(
Marat Dukhanb8642352019-10-30 15:43:02 -07004317 name = "xnn_enable_assembly_explicit_true",
4318 define_values = {"xnn_enable_assembly": "true"},
4319)
4320
4321# Disables usage of assembly kernels.
4322config_setting(
4323 name = "xnn_enable_assembly_explicit_false",
4324 define_values = {"xnn_enable_assembly": "false"},
4325)
4326
Marat Dukhan05702cf2020-03-26 15:41:33 -07004327# Disables usage of HMP-aware optimizations.
4328config_setting(
4329 name = "xnn_enable_hmp_explicit_false",
4330 define_values = {"xnn_enable_hmp": "false"},
4331)
4332
Chao Mei6ddfc602020-05-13 22:29:36 -07004333# Enable usage of optimized memory allocation
4334config_setting(
4335 name = "xnn_enable_memopt_explicit_true",
4336 define_values = {"xnnpack_enable_memopt": "true"},
4337)
4338
4339# Disable usage of optimized memory allocation
4340config_setting(
4341 name = "xnn_enable_memopt_explicit_false",
4342 define_values = {"xnnpack_enable_memopt": "false"},
4343)
4344
Marat Dukhanb8642352019-10-30 15:43:02 -07004345# Builds with -c dbg
4346config_setting(
4347 name = "debug_build",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004348 values = {
Marat Dukhanb8642352019-10-30 15:43:02 -07004349 "compilation_mode": "dbg",
4350 },
4351)
4352
4353# Builds with -c opt
4354config_setting(
4355 name = "optimized_build",
4356 values = {
4357 "compilation_mode": "opt",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004358 },
4359)
4360
4361config_setting(
Marat Dukhanb8642352019-10-30 15:43:02 -07004362 name = "linux_k8",
4363 values = {"cpu": "k8"},
4364)
4365
4366config_setting(
Marat Dukhan582094e2020-04-30 17:21:25 -07004367 name = "linux_arm",
4368 values = {"cpu": "arm"},
Marat Dukhan4e45e662019-10-03 15:40:24 -07004369)
4370
4371config_setting(
Terry Heo68eef3f2020-04-13 22:53:52 -07004372 name = "linux_armhf",
4373 values = {"cpu": "armhf"},
4374)
4375
4376config_setting(
Marat Dukhan582094e2020-04-30 17:21:25 -07004377 name = "linux_aarch64",
4378 values = {"cpu": "aarch64"},
4379)
4380
4381config_setting(
Marat Dukhan08c4a432019-10-03 09:29:21 -07004382 name = "android",
4383 values = {"crosstool_top": "//external:android/crosstool"},
4384)
4385
4386config_setting(
4387 name = "android_armv7",
4388 values = {
4389 "crosstool_top": "//external:android/crosstool",
4390 "cpu": "armeabi-v7a",
4391 },
4392)
4393
4394config_setting(
4395 name = "android_arm64",
4396 values = {
4397 "crosstool_top": "//external:android/crosstool",
4398 "cpu": "arm64-v8a",
4399 },
4400)
4401
4402config_setting(
4403 name = "android_x86",
4404 values = {
4405 "crosstool_top": "//external:android/crosstool",
4406 "cpu": "x86",
4407 },
4408)
4409
4410config_setting(
4411 name = "android_x86_64",
4412 values = {
4413 "crosstool_top": "//external:android/crosstool",
4414 "cpu": "x86_64",
4415 },
4416)
4417
4418config_setting(
Marat Dukhan10a38082020-04-17 03:58:35 -07004419 name = "windows_x86_64",
4420 values = {"cpu": "x64_windows"},
Marat Dukhan9fe932e2020-04-11 17:14:15 -07004421)
4422
4423config_setting(
Marat Dukhan10a38082020-04-17 03:58:35 -07004424 name = "windows_x86_64_clang",
4425 values = {
4426 "compiler": "clang-cl",
4427 "cpu": "x64_windows",
4428 },
4429)
4430
4431config_setting(
4432 name = "windows_x86_64_mingw",
4433 values = {
4434 "compiler": "mingw-gcc",
4435 "cpu": "x64_windows",
4436 },
4437)
4438
4439config_setting(
4440 name = "windows_x86_64_msys",
4441 values = {
4442 "compiler": "msys-gcc",
4443 "cpu": "x64_windows",
4444 },
Marat Dukhan9fe932e2020-04-11 17:14:15 -07004445)
4446
4447config_setting(
Marat Dukhan885ca242019-10-07 09:17:32 -07004448 name = "macos_x86_64",
4449 values = {
4450 "apple_platform_type": "macos",
4451 "cpu": "darwin",
4452 },
4453)
4454
4455config_setting(
Marat Dukhan08c4a432019-10-03 09:29:21 -07004456 name = "emscripten",
Marat Dukhan1a729ec2019-10-07 09:31:44 -07004457 values = {"crosstool_top": "//toolchain:emscripten"},
Marat Dukhan08c4a432019-10-03 09:29:21 -07004458)
4459
4460config_setting(
4461 name = "emscripten_wasm",
4462 values = {
Marat Dukhan1a729ec2019-10-07 09:31:44 -07004463 "crosstool_top": "//toolchain:emscripten",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004464 "cpu": "wasm",
4465 },
4466)
4467
4468config_setting(
4469 name = "emscripten_wasmsimd",
4470 values = {
Marat Dukhan1a729ec2019-10-07 09:31:44 -07004471 "crosstool_top": "//toolchain:emscripten",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004472 "cpu": "wasm",
Marat Dukhan8c19e3c2019-10-30 12:14:58 -07004473 "features": "wasm_simd",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004474 },
4475)
4476
4477config_setting(
4478 name = "emscripten_asmjs",
4479 values = {
Marat Dukhan1a729ec2019-10-07 09:31:44 -07004480 "crosstool_top": "//toolchain:emscripten",
Marat Dukhan08c4a432019-10-03 09:29:21 -07004481 "cpu": "asmjs",
4482 },
4483)
Marat Dukhan1498d1d2020-02-11 20:00:05 -08004484
4485config_setting(
4486 name = "ios_armv7",
4487 values = {
Marat Dukhanf85fc332020-02-13 00:05:20 -08004488 "apple_platform_type": "ios",
Marat Dukhan1498d1d2020-02-11 20:00:05 -08004489 "cpu": "ios_armv7",
4490 },
4491)
4492
4493config_setting(
4494 name = "ios_arm64",
4495 values = {
Marat Dukhanf85fc332020-02-13 00:05:20 -08004496 "apple_platform_type": "ios",
Marat Dukhan1498d1d2020-02-11 20:00:05 -08004497 "cpu": "ios_arm64",
4498 },
4499)
4500
4501config_setting(
4502 name = "ios_arm64e",
4503 values = {
Marat Dukhanf85fc332020-02-13 00:05:20 -08004504 "apple_platform_type": "ios",
Marat Dukhan1498d1d2020-02-11 20:00:05 -08004505 "cpu": "ios_arm64e",
4506 },
4507)
4508
4509config_setting(
4510 name = "ios_x86",
4511 values = {
Marat Dukhanf85fc332020-02-13 00:05:20 -08004512 "apple_platform_type": "ios",
Marat Dukhan1498d1d2020-02-11 20:00:05 -08004513 "cpu": "ios_i386",
4514 },
4515)
4516
4517config_setting(
4518 name = "ios_x86_64",
4519 values = {
Marat Dukhanf85fc332020-02-13 00:05:20 -08004520 "apple_platform_type": "ios",
Marat Dukhan1498d1d2020-02-11 20:00:05 -08004521 "cpu": "ios_x86_64",
4522 },
4523)
4524
4525config_setting(
4526 name = "watchos_armv7k",
4527 values = {
Marat Dukhanf85fc332020-02-13 00:05:20 -08004528 "apple_platform_type": "watchos",
Marat Dukhan1498d1d2020-02-11 20:00:05 -08004529 "cpu": "watchos_armv7k",
4530 },
4531)
4532
4533config_setting(
4534 name = "watchos_arm64_32",
4535 values = {
Marat Dukhanf85fc332020-02-13 00:05:20 -08004536 "apple_platform_type": "watchos",
Marat Dukhan1498d1d2020-02-11 20:00:05 -08004537 "cpu": "watchos_arm64_32",
4538 },
4539)
4540
4541config_setting(
4542 name = "watchos_x86",
4543 values = {
Marat Dukhanf85fc332020-02-13 00:05:20 -08004544 "apple_platform_type": "watchos",
Marat Dukhan1498d1d2020-02-11 20:00:05 -08004545 "cpu": "watchos_i386",
4546 },
4547)
4548
4549config_setting(
4550 name = "watchos_x86_64",
4551 values = {
Marat Dukhanf85fc332020-02-13 00:05:20 -08004552 "apple_platform_type": "watchos",
Marat Dukhan1498d1d2020-02-11 20:00:05 -08004553 "cpu": "watchos_x86_64",
4554 },
4555)
4556
4557config_setting(
4558 name = "tvos_arm64",
4559 values = {
Marat Dukhanf85fc332020-02-13 00:05:20 -08004560 "apple_platform_type": "tvos",
Marat Dukhan1498d1d2020-02-11 20:00:05 -08004561 "cpu": "tvos_arm64",
4562 },
4563)
4564
4565config_setting(
4566 name = "tvos_x86_64",
4567 values = {
Marat Dukhanf85fc332020-02-13 00:05:20 -08004568 "apple_platform_type": "tvos",
Marat Dukhan1498d1d2020-02-11 20:00:05 -08004569 "cpu": "tvos_x86_64",
4570 },
4571)