| # Copyright 2019 Google LLC |
| # |
| # This source code is licensed under the BSD-style license found in the |
| # LICENSE file in the root directory of this source tree. |
| # |
| # Description: |
| # XNNPACK - optimized floating-point neural network operators library |
| |
| load(":build_defs.bzl", "xnnpack_aggregate_library", "xnnpack_benchmark", "xnnpack_binary", "xnnpack_cc_library", "xnnpack_min_size_copts", "xnnpack_optional_armcl_copts", "xnnpack_optional_armcl_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_copts", "xnnpack_unit_test", "xnnpack_visibility") |
| |
| licenses(["notice"]) |
| |
| exports_files(["LICENSE"]) |
| |
| OPERATOR_BENCHMARK_DEPS = [ |
| ":XNNPACK", |
| ":bench_utils", |
| "@cpuinfo", |
| "@pthreadpool", |
| ] |
| |
| MICROKERNEL_BENCHMARK_DEPS = [ |
| ":ukernels", |
| ":bench_utils", |
| ":enable_assembly", |
| "@cpuinfo", |
| "@FP16", |
| "@pthreadpool", |
| ] |
| |
| ACCURACY_EVAL_DEPS = [ |
| ":XNNPACK", |
| ":ukernels", |
| "@FP16", |
| "@pthreadpool", |
| ] |
| |
| MICROKERNEL_TEST_DEPS = [ |
| ":ukernels", |
| ":enable_assembly", |
| "@cpuinfo", |
| "@FP16", |
| "@pthreadpool", |
| ] |
| |
| OPERATOR_TEST_DEPS = [ |
| ":XNNPACK", |
| "@pthreadpool", |
| "@FP16", |
| ] |
| |
| OPERATOR_SRCS = [ |
| "src/add-nc.c", |
| "src/argmax-pooling-nhwc.c", |
| "src/average-pooling-nhwc.c", |
| "src/binary-elementwise-nd.c", |
| "src/channel-pad-nc.c", |
| "src/channel-shuffle-nc.c", |
| "src/clamp-nc.c", |
| "src/convolution-nchw.c", |
| "src/convolution-nhwc.c", |
| "src/deconvolution-nhwc.c", |
| "src/fully-connected-nc.c", |
| "src/global-average-pooling-ncw.c", |
| "src/global-average-pooling-nwc.c", |
| "src/hardswish-nc.c", |
| "src/leaky-relu-nc.c", |
| "src/max-pooling-nhwc.c", |
| "src/prelu-nc.c", |
| "src/resize-bilinear-nhwc.c", |
| "src/sigmoid-nc.c", |
| "src/softargmax-nc.c", |
| "src/unpooling-nhwc.c", |
| ] |
| |
| SCALAR_UKERNELS = [ |
| "src/f32-argmaxpool/4x-scalar-c1.c", |
| "src/f32-argmaxpool/9p8x-scalar-c1.c", |
| "src/f32-argmaxpool/9x-scalar-c1.c", |
| "src/f32-avgpool/mp9p8q-scalar.c", |
| "src/f32-avgpool/up9-scalar.c", |
| "src/f32-bilinear/gen/scalar-c1.c", |
| "src/f32-bilinear/gen/scalar-c2.c", |
| "src/f32-bilinear/gen/scalar-c4.c", |
| "src/f32-clamp/scalar.c", |
| "src/f32-conv-hwc/3x3s2p0p1c3x4-scalar-1x1.c", |
| "src/f32-conv-hwc/3x3s2p1c3x4-scalar-1x1.c", |
| "src/f32-conv-hwc2spchw/3x3s2p1c3x4-scalar-1x1.c", |
| "src/f32-dwconv-spchw/3x3p1-scalar.c", |
| "src/f32-dwconv-spchw/3x3s2p1-scalar.c", |
| "src/f32-dwconv-spchw/5x5p2-scalar.c", |
| "src/f32-dwconv-spchw/5x5s2p2-scalar.c", |
| "src/f32-dwconv/gen/up1x25-scalar-acc2.c", |
| "src/f32-dwconv/gen/up1x25-scalar.c", |
| "src/f32-dwconv/gen/up1x4-scalar-acc2.c", |
| "src/f32-dwconv/gen/up1x4-scalar.c", |
| "src/f32-dwconv/gen/up1x9-scalar-acc2.c", |
| "src/f32-dwconv/gen/up1x9-scalar.c", |
| "src/f32-dwconv/gen/up2x25-scalar-acc2.c", |
| "src/f32-dwconv/gen/up2x25-scalar.c", |
| "src/f32-dwconv/gen/up2x4-scalar-acc2.c", |
| "src/f32-dwconv/gen/up2x4-scalar.c", |
| "src/f32-dwconv/gen/up2x9-scalar-acc2.c", |
| "src/f32-dwconv/gen/up2x9-scalar.c", |
| "src/f32-gavgpool-spchw/scalar-x1.c", |
| "src/f32-gavgpool/mp7p7q-scalar.c", |
| "src/f32-gavgpool/up7-scalar.c", |
| "src/f32-gemm/gen-inc/1x4-scalar.c", |
| "src/f32-gemm/gen-inc/2x4-scalar.c", |
| "src/f32-gemm/gen-inc/4x4-scalar.c", |
| "src/f32-gemm/gen/1x4-scalar.c", |
| "src/f32-gemm/gen/2x4-scalar.c", |
| "src/f32-gemm/gen/4x2-scalar.c", |
| "src/f32-gemm/gen/4x4-scalar.c", |
| "src/f32-hswish/scalar.c", |
| "src/f32-igemm/gen/1x4-scalar.c", |
| "src/f32-igemm/gen/2x4-scalar.c", |
| "src/f32-igemm/gen/4x2-scalar.c", |
| "src/f32-igemm/gen/4x4-scalar.c", |
| "src/f32-maxpool/9p8x-scalar-c1.c", |
| "src/f32-pavgpool/mp9p8q-scalar.c", |
| "src/f32-pavgpool/up9-scalar.c", |
| "src/f32-ppmm/gen/2x4-scalar.c", |
| "src/f32-ppmm/gen/3x3-scalar.c", |
| "src/f32-ppmm/gen/4x2-scalar.c", |
| "src/f32-ppmm/gen/4x4-scalar.c", |
| "src/f32-prelu/gen/scalar-2x1.c", |
| "src/f32-prelu/gen/scalar-2x4.c", |
| "src/f32-rmax/scalar.c", |
| "src/f32-spmm/gen/1x1-scalar-pipelined.c", |
| "src/f32-spmm/gen/1x1-scalar.c", |
| "src/f32-spmm/gen/2x1-scalar-pipelined.c", |
| "src/f32-spmm/gen/2x1-scalar.c", |
| "src/f32-spmm/gen/4x1-scalar-pipelined.c", |
| "src/f32-spmm/gen/4x1-scalar.c", |
| "src/f32-spmm/gen/8x1-scalar-pipelined.c", |
| "src/f32-spmm/gen/8x1-scalar.c", |
| "src/f32-spmm/gen/8x2-scalar.c", |
| "src/f32-spmm/gen/8x4-scalar.c", |
| "src/f32-vbinary/gen/vadd-scalar-x1.c", |
| "src/f32-vbinary/gen/vadd-scalar-x2.c", |
| "src/f32-vbinary/gen/vadd-scalar-x4.c", |
| "src/f32-vbinary/gen/vaddc-scalar-x1.c", |
| "src/f32-vbinary/gen/vaddc-scalar-x2.c", |
| "src/f32-vbinary/gen/vaddc-scalar-x4.c", |
| "src/f32-vbinary/gen/vdiv-scalar-x1.c", |
| "src/f32-vbinary/gen/vdiv-scalar-x2.c", |
| "src/f32-vbinary/gen/vdiv-scalar-x4.c", |
| "src/f32-vbinary/gen/vdivc-scalar-x1.c", |
| "src/f32-vbinary/gen/vdivc-scalar-x2.c", |
| "src/f32-vbinary/gen/vdivc-scalar-x4.c", |
| "src/f32-vbinary/gen/vmax-scalar-x1.c", |
| "src/f32-vbinary/gen/vmax-scalar-x2.c", |
| "src/f32-vbinary/gen/vmax-scalar-x4.c", |
| "src/f32-vbinary/gen/vmaxc-scalar-x1.c", |
| "src/f32-vbinary/gen/vmaxc-scalar-x2.c", |
| "src/f32-vbinary/gen/vmaxc-scalar-x4.c", |
| "src/f32-vbinary/gen/vmin-scalar-x1.c", |
| "src/f32-vbinary/gen/vmin-scalar-x2.c", |
| "src/f32-vbinary/gen/vmin-scalar-x4.c", |
| "src/f32-vbinary/gen/vminc-scalar-x1.c", |
| "src/f32-vbinary/gen/vminc-scalar-x2.c", |
| "src/f32-vbinary/gen/vminc-scalar-x4.c", |
| "src/f32-vbinary/gen/vmul-scalar-x1.c", |
| "src/f32-vbinary/gen/vmul-scalar-x2.c", |
| "src/f32-vbinary/gen/vmul-scalar-x4.c", |
| "src/f32-vbinary/gen/vmulc-scalar-x1.c", |
| "src/f32-vbinary/gen/vmulc-scalar-x2.c", |
| "src/f32-vbinary/gen/vmulc-scalar-x4.c", |
| "src/f32-vbinary/gen/vrdivc-scalar-x1.c", |
| "src/f32-vbinary/gen/vrdivc-scalar-x2.c", |
| "src/f32-vbinary/gen/vrdivc-scalar-x4.c", |
| "src/f32-vbinary/gen/vrsubc-scalar-x1.c", |
| "src/f32-vbinary/gen/vrsubc-scalar-x2.c", |
| "src/f32-vbinary/gen/vrsubc-scalar-x4.c", |
| "src/f32-vbinary/gen/vsub-scalar-x1.c", |
| "src/f32-vbinary/gen/vsub-scalar-x2.c", |
| "src/f32-vbinary/gen/vsub-scalar-x4.c", |
| "src/f32-vbinary/gen/vsubc-scalar-x1.c", |
| "src/f32-vbinary/gen/vsubc-scalar-x2.c", |
| "src/f32-vbinary/gen/vsubc-scalar-x4.c", |
| "src/f32-vmulcaddc/gen/c1-scalar-2x.c", |
| "src/f32-vmulcaddc/gen/c2-scalar-2x.c", |
| "src/f32-vmulcaddc/gen/c4-scalar-2x.c", |
| "src/q8-avgpool/mp9p8q-scalar.c", |
| "src/q8-avgpool/up9-scalar.c", |
| "src/q8-dwconv/up1x9-scalar.c", |
| "src/q8-gavgpool/mp7p7q-scalar.c", |
| "src/q8-gavgpool/up7-scalar.c", |
| "src/q8-gemm/2x2-scalar.c", |
| "src/q8-igemm/2x2-scalar.c", |
| "src/q8-vadd/scalar.c", |
| "src/u8-clamp/scalar.c", |
| "src/u8-lut32norm/scalar.c", |
| "src/u8-maxpool/9p8x-scalar-c1.c", |
| "src/u8-rmax/scalar.c", |
| "src/x32-packx/x2-scalar.c", |
| "src/x32-packx/x3-scalar.c", |
| "src/x32-packx/x4-scalar.c", |
| "src/x32-pad/x2-scalar.c", |
| "src/x32-unpool/scalar.c", |
| "src/x32-zip/x2-scalar.c", |
| "src/x32-zip/x3-scalar.c", |
| "src/x32-zip/x4-scalar.c", |
| "src/x32-zip/xm-scalar.c", |
| "src/x8-lut/scalar.c", |
| "src/x8-zip/x2-scalar.c", |
| "src/x8-zip/x3-scalar.c", |
| "src/x8-zip/x4-scalar.c", |
| "src/x8-zip/xm-scalar.c", |
| ] |
| |
| WASM_UKERNELS = [ |
| "src/f32-avgpool/mp9p8q-wasm.c", |
| "src/f32-avgpool/up9-wasm.c", |
| "src/f32-clamp/wasm.c", |
| "src/f32-dwconv/gen/up1x25-wasm-acc2.c", |
| "src/f32-dwconv/gen/up1x25-wasm.c", |
| "src/f32-dwconv/gen/up1x4-wasm-acc2.c", |
| "src/f32-dwconv/gen/up1x4-wasm.c", |
| "src/f32-dwconv/gen/up1x9-wasm-acc2.c", |
| "src/f32-dwconv/gen/up1x9-wasm.c", |
| "src/f32-dwconv/gen/up2x25-wasm-acc2.c", |
| "src/f32-dwconv/gen/up2x25-wasm.c", |
| "src/f32-dwconv/gen/up2x4-wasm-acc2.c", |
| "src/f32-dwconv/gen/up2x4-wasm.c", |
| "src/f32-dwconv/gen/up2x9-wasm-acc2.c", |
| "src/f32-dwconv/gen/up2x9-wasm.c", |
| "src/f32-gavgpool/mp7p7q-wasm.c", |
| "src/f32-gavgpool/up7-wasm.c", |
| "src/f32-gemm/gen-inc/1x4-wasm.c", |
| "src/f32-gemm/gen-inc/2x4-wasm.c", |
| "src/f32-gemm/gen-inc/4x4-wasm.c", |
| "src/f32-gemm/gen/1x4-wasm.c", |
| "src/f32-gemm/gen/2x4-wasm.c", |
| "src/f32-gemm/gen/4x2-wasm.c", |
| "src/f32-gemm/gen/4x4-wasm.c", |
| "src/f32-hswish/wasm.c", |
| "src/f32-igemm/gen/1x4-wasm.c", |
| "src/f32-igemm/gen/2x4-wasm.c", |
| "src/f32-igemm/gen/4x2-wasm.c", |
| "src/f32-igemm/gen/4x4-wasm.c", |
| "src/f32-maxpool/9p8x-wasm-c1.c", |
| "src/f32-pavgpool/mp9p8q-wasm.c", |
| "src/f32-pavgpool/up9-wasm.c", |
| "src/f32-prelu/gen/wasm-2x1.c", |
| "src/f32-prelu/gen/wasm-2x4.c", |
| "src/f32-vbinary/gen/vadd-wasm-x1.c", |
| "src/f32-vbinary/gen/vadd-wasm-x2.c", |
| "src/f32-vbinary/gen/vadd-wasm-x4.c", |
| "src/f32-vbinary/gen/vaddc-wasm-x1.c", |
| "src/f32-vbinary/gen/vaddc-wasm-x2.c", |
| "src/f32-vbinary/gen/vaddc-wasm-x4.c", |
| "src/f32-vbinary/gen/vdiv-wasm-x1.c", |
| "src/f32-vbinary/gen/vdiv-wasm-x2.c", |
| "src/f32-vbinary/gen/vdiv-wasm-x4.c", |
| "src/f32-vbinary/gen/vdivc-wasm-x1.c", |
| "src/f32-vbinary/gen/vdivc-wasm-x2.c", |
| "src/f32-vbinary/gen/vdivc-wasm-x4.c", |
| "src/f32-vbinary/gen/vmax-wasm-x1.c", |
| "src/f32-vbinary/gen/vmax-wasm-x2.c", |
| "src/f32-vbinary/gen/vmax-wasm-x4.c", |
| "src/f32-vbinary/gen/vmaxc-wasm-x1.c", |
| "src/f32-vbinary/gen/vmaxc-wasm-x2.c", |
| "src/f32-vbinary/gen/vmaxc-wasm-x4.c", |
| "src/f32-vbinary/gen/vmin-wasm-x1.c", |
| "src/f32-vbinary/gen/vmin-wasm-x2.c", |
| "src/f32-vbinary/gen/vmin-wasm-x4.c", |
| "src/f32-vbinary/gen/vminc-wasm-x1.c", |
| "src/f32-vbinary/gen/vminc-wasm-x2.c", |
| "src/f32-vbinary/gen/vminc-wasm-x4.c", |
| "src/f32-vbinary/gen/vmul-wasm-x1.c", |
| "src/f32-vbinary/gen/vmul-wasm-x2.c", |
| "src/f32-vbinary/gen/vmul-wasm-x4.c", |
| "src/f32-vbinary/gen/vmulc-wasm-x1.c", |
| "src/f32-vbinary/gen/vmulc-wasm-x2.c", |
| "src/f32-vbinary/gen/vmulc-wasm-x4.c", |
| "src/f32-vbinary/gen/vrdivc-wasm-x1.c", |
| "src/f32-vbinary/gen/vrdivc-wasm-x2.c", |
| "src/f32-vbinary/gen/vrdivc-wasm-x4.c", |
| "src/f32-vbinary/gen/vrsubc-wasm-x1.c", |
| "src/f32-vbinary/gen/vrsubc-wasm-x2.c", |
| "src/f32-vbinary/gen/vrsubc-wasm-x4.c", |
| "src/f32-vbinary/gen/vsub-wasm-x1.c", |
| "src/f32-vbinary/gen/vsub-wasm-x2.c", |
| "src/f32-vbinary/gen/vsub-wasm-x4.c", |
| "src/f32-vbinary/gen/vsubc-wasm-x1.c", |
| "src/f32-vbinary/gen/vsubc-wasm-x2.c", |
| "src/f32-vbinary/gen/vsubc-wasm-x4.c", |
| "src/f32-vmulcaddc/gen/c1-wasm-2x.c", |
| "src/f32-vmulcaddc/gen/c2-wasm-2x.c", |
| "src/f32-vmulcaddc/gen/c4-wasm-2x.c", |
| ] |
| |
| PSIMD_UKERNELS = [ |
| "src/f32-argmaxpool/4x-psimd-c4.c", |
| "src/f32-argmaxpool/9p8x-psimd-c4.c", |
| "src/f32-argmaxpool/9x-psimd-c4.c", |
| "src/f32-avgpool/mp9p8q-psimd.c", |
| "src/f32-avgpool/up9-psimd.c", |
| "src/f32-bilinear/gen/psimd-c4.c", |
| "src/f32-bilinear/gen/psimd-c8.c", |
| "src/f32-clamp/psimd.c", |
| "src/f32-dwconv/gen/up4x25-psimd-acc2.c", |
| "src/f32-dwconv/gen/up4x25-psimd.c", |
| "src/f32-dwconv/gen/up4x4-psimd-acc2.c", |
| "src/f32-dwconv/gen/up4x4-psimd.c", |
| "src/f32-dwconv/gen/up4x9-psimd-acc2.c", |
| "src/f32-dwconv/gen/up4x9-psimd.c", |
| "src/f32-dwconv/gen/up8x25-psimd-acc2.c", |
| "src/f32-dwconv/gen/up8x25-psimd.c", |
| "src/f32-dwconv/gen/up8x4-psimd-acc2.c", |
| "src/f32-dwconv/gen/up8x4-psimd.c", |
| "src/f32-dwconv/gen/up8x9-psimd-acc2.c", |
| "src/f32-dwconv/gen/up8x9-psimd.c", |
| "src/f32-gavgpool/mp7p7q-psimd.c", |
| "src/f32-gavgpool/up7-psimd.c", |
| "src/f32-gemm/gen/1x8-psimd-loadsplat.c", |
| "src/f32-gemm/gen/1x8-psimd-splat.c", |
| "src/f32-gemm/gen/1x8s4-psimd.c", |
| "src/f32-gemm/gen/4x8-psimd-loadsplat.c", |
| "src/f32-gemm/gen/4x8-psimd-splat.c", |
| "src/f32-gemm/gen/4x8s4-psimd.c", |
| "src/f32-gemm/gen/6x8-psimd-loadsplat.c", |
| "src/f32-gemm/gen/6x8-psimd-splat.c", |
| "src/f32-gemm/gen/6x8s4-psimd.c", |
| "src/f32-gemm/gen-inc/1x8-psimd-loadsplat.c", |
| "src/f32-gemm/gen-inc/1x8-psimd-splat.c", |
| "src/f32-gemm/gen-inc/1x8s4-psimd.c", |
| "src/f32-gemm/gen-inc/4x8-psimd-loadsplat.c", |
| "src/f32-gemm/gen-inc/4x8-psimd-splat.c", |
| "src/f32-gemm/gen-inc/4x8s4-psimd.c", |
| "src/f32-gemm/gen-inc/6x8-psimd-loadsplat.c", |
| "src/f32-gemm/gen-inc/6x8-psimd-splat.c", |
| "src/f32-gemm/gen-inc/6x8s4-psimd.c", |
| "src/f32-hswish/psimd.c", |
| "src/f32-igemm/gen/1x8-psimd-loadsplat.c", |
| "src/f32-igemm/gen/1x8-psimd-splat.c", |
| "src/f32-igemm/gen/1x8s4-psimd.c", |
| "src/f32-igemm/gen/4x2c4-psimd.c", |
| "src/f32-igemm/gen/4x8-psimd-loadsplat.c", |
| "src/f32-igemm/gen/4x8-psimd-splat.c", |
| "src/f32-igemm/gen/4x8s4-psimd.c", |
| "src/f32-igemm/gen/6x8-psimd-loadsplat.c", |
| "src/f32-igemm/gen/6x8-psimd-splat.c", |
| "src/f32-igemm/gen/6x8s4-psimd.c", |
| "src/f32-maxpool/9p8x-psimd-c4.c", |
| "src/f32-pavgpool/mp9p8q-psimd.c", |
| "src/f32-pavgpool/up9-psimd.c", |
| "src/f32-ppmm/gen/4x8-psimd.c", |
| "src/f32-prelu/gen/psimd-2x4.c", |
| "src/f32-prelu/gen/psimd-2x8.c", |
| "src/f32-vbinary/gen/vadd-psimd-x4.c", |
| "src/f32-vbinary/gen/vadd-psimd-x8.c", |
| "src/f32-vbinary/gen/vaddc-psimd-x4.c", |
| "src/f32-vbinary/gen/vaddc-psimd-x8.c", |
| "src/f32-vbinary/gen/vdiv-psimd-x4.c", |
| "src/f32-vbinary/gen/vdiv-psimd-x8.c", |
| "src/f32-vbinary/gen/vdivc-psimd-x4.c", |
| "src/f32-vbinary/gen/vdivc-psimd-x8.c", |
| "src/f32-vbinary/gen/vmax-psimd-x4.c", |
| "src/f32-vbinary/gen/vmax-psimd-x8.c", |
| "src/f32-vbinary/gen/vmaxc-psimd-x4.c", |
| "src/f32-vbinary/gen/vmaxc-psimd-x8.c", |
| "src/f32-vbinary/gen/vmin-psimd-x4.c", |
| "src/f32-vbinary/gen/vmin-psimd-x8.c", |
| "src/f32-vbinary/gen/vminc-psimd-x4.c", |
| "src/f32-vbinary/gen/vminc-psimd-x8.c", |
| "src/f32-vbinary/gen/vmul-psimd-x4.c", |
| "src/f32-vbinary/gen/vmul-psimd-x8.c", |
| "src/f32-vbinary/gen/vmulc-psimd-x4.c", |
| "src/f32-vbinary/gen/vmulc-psimd-x8.c", |
| "src/f32-vbinary/gen/vrdivc-psimd-x4.c", |
| "src/f32-vbinary/gen/vrdivc-psimd-x8.c", |
| "src/f32-vbinary/gen/vrsubc-psimd-x4.c", |
| "src/f32-vbinary/gen/vrsubc-psimd-x8.c", |
| "src/f32-vbinary/gen/vsub-psimd-x4.c", |
| "src/f32-vbinary/gen/vsub-psimd-x8.c", |
| "src/f32-vbinary/gen/vsubc-psimd-x4.c", |
| "src/f32-vbinary/gen/vsubc-psimd-x8.c", |
| "src/f32-vmulcaddc/gen/c4-psimd-2x.c", |
| "src/f32-vmulcaddc/gen/c8-psimd-2x.c", |
| "src/x32-packx/x4-psimd.c", |
| "src/x32-pad/x2-psimd.c", |
| "src/x32-unpool/psimd.c", |
| "src/x32-zip/x2-psimd.c", |
| "src/x32-zip/x3-psimd.c", |
| "src/x32-zip/x4-psimd.c", |
| "src/x32-zip/xm-psimd.c", |
| ] |
| |
| # ISA-specific micro-kernels |
| NEON_UKERNELS = [ |
| "src/f32-avgpool/mp9p8q-neon.c", |
| "src/f32-avgpool/up9-neon.c", |
| "src/f32-bilinear/gen/neon-c4.c", |
| "src/f32-bilinear/gen/neon-c8.c", |
| "src/f32-clamp/neon.c", |
| "src/f32-dwconv/gen/up4x9-neon.c", |
| "src/f32-dwconv/gen/up4x9-neon-acc2.c", |
| "src/f32-dwconv/gen/up8x9-neon.c", |
| "src/f32-dwconv/gen/up8x9-neon-acc2.c", |
| "src/f32-gavgpool-spchw/neon-x4.c", |
| "src/f32-gavgpool/mp7p7q-neon.c", |
| "src/f32-gavgpool/up7-neon.c", |
| "src/f32-gemm/gen/1x8-neon-lane-ld64.c", |
| "src/f32-gemm/gen/4x2-neon-lane-ld64.c", |
| "src/f32-gemm/gen/4x8-neon-lane-ld128.c", |
| "src/f32-gemm/gen/4x8-neon-lane-ld64.c", |
| "src/f32-gemm/gen/5x8-neon-lane-ld64.c", |
| "src/f32-gemm/gen/6x8-neon-lane-ld64.c", |
| "src/f32-gemm/gen/6x8-neon-lane-ld128.c", |
| "src/f32-gemm/gen/1x8-neon-dup-ld64.c", |
| "src/f32-gemm/gen/4x8-neon-dup-ld128.c", |
| "src/f32-gemm/gen/4x8-neon-dup-ld64.c", |
| "src/f32-gemm/gen/6x8-neon-dup-ld64.c", |
| "src/f32-gemm/gen/6x8-neon-dup-ld128.c", |
| "src/f32-gemm/gen/1x8s4-neon.c", |
| "src/f32-gemm/gen/4x8s4-neon.c", |
| "src/f32-gemm/gen/6x8s4-neon.c", |
| "src/f32-gemm/gen/8x8s4-neon.c", |
| "src/f32-gemm/gen-inc/1x8-neon-lane-ld64.c", |
| "src/f32-gemm/gen-inc/4x8-neon-lane-ld128.c", |
| "src/f32-gemm/gen-inc/4x8-neon-lane-ld64.c", |
| "src/f32-gemm/gen-inc/5x8-neon-lane-ld64.c", |
| "src/f32-gemm/gen-inc/6x8-neon-lane-ld64.c", |
| "src/f32-gemm/gen-inc/6x8-neon-lane-ld128.c", |
| "src/f32-gemm/gen-inc/1x8-neon-dup-ld64.c", |
| "src/f32-gemm/gen-inc/4x8-neon-dup-ld128.c", |
| "src/f32-gemm/gen-inc/4x8-neon-dup-ld64.c", |
| "src/f32-gemm/gen-inc/6x8-neon-dup-ld64.c", |
| "src/f32-gemm/gen-inc/6x8-neon-dup-ld128.c", |
| "src/f32-gemm/gen-inc/1x8s4-neon.c", |
| "src/f32-gemm/gen-inc/4x8s4-neon.c", |
| "src/f32-gemm/gen-inc/6x8s4-neon.c", |
| "src/f32-gemm/gen-inc/8x8s4-neon.c", |
| "src/f32-hswish/neon.c", |
| "src/f32-igemm/gen/1x8-neon-lane-ld64.c", |
| "src/f32-igemm/gen/4x2-neon-lane-ld64.c", |
| "src/f32-igemm/gen/4x4-neon-lane-ld64.c", |
| "src/f32-igemm/gen/4x8-neon-lane-ld128.c", |
| "src/f32-igemm/gen/4x8-neon-lane-ld64.c", |
| "src/f32-igemm/gen/6x8-neon-lane-ld64.c", |
| "src/f32-igemm/gen/6x8-neon-lane-ld128.c", |
| "src/f32-igemm/gen/1x8-neon-dup-ld64.c", |
| "src/f32-igemm/gen/4x8-neon-dup-ld128.c", |
| "src/f32-igemm/gen/4x8-neon-dup-ld64.c", |
| "src/f32-igemm/gen/6x8-neon-dup-ld64.c", |
| "src/f32-igemm/gen/6x8-neon-dup-ld128.c", |
| "src/f32-igemm/gen/1x8s4-neon.c", |
| "src/f32-igemm/gen/4x8s4-neon.c", |
| "src/f32-igemm/gen/6x8s4-neon.c", |
| "src/f32-igemm/gen/8x8s4-neon.c", |
| "src/f32-pavgpool/mp9p8q-neon.c", |
| "src/f32-pavgpool/up9-neon.c", |
| "src/f32-ppmm/gen/4x8-neon.c", |
| "src/f32-ppmm/gen/8x8-neon.c", |
| "src/f32-prelu/gen/neon-2x4.c", |
| "src/f32-prelu/gen/neon-2x8.c", |
| "src/f32-rmax/neon.c", |
| "src/f32-sigmoid/gen/neon-frac-p9-p10-nr1recps-x16.c", |
| "src/f32-vbinary/gen/vadd-neon-x4.c", |
| "src/f32-vbinary/gen/vadd-neon-x8.c", |
| "src/f32-vbinary/gen/vaddc-neon-x4.c", |
| "src/f32-vbinary/gen/vaddc-neon-x8.c", |
| "src/f32-vbinary/gen/vmax-neon-x4.c", |
| "src/f32-vbinary/gen/vmax-neon-x8.c", |
| "src/f32-vbinary/gen/vmaxc-neon-x4.c", |
| "src/f32-vbinary/gen/vmaxc-neon-x8.c", |
| "src/f32-vbinary/gen/vmin-neon-x4.c", |
| "src/f32-vbinary/gen/vmin-neon-x8.c", |
| "src/f32-vbinary/gen/vminc-neon-x4.c", |
| "src/f32-vbinary/gen/vminc-neon-x8.c", |
| "src/f32-vbinary/gen/vmul-neon-x4.c", |
| "src/f32-vbinary/gen/vmul-neon-x8.c", |
| "src/f32-vbinary/gen/vmulc-neon-x4.c", |
| "src/f32-vbinary/gen/vmulc-neon-x8.c", |
| "src/f32-vbinary/gen/vrsubc-neon-x4.c", |
| "src/f32-vbinary/gen/vrsubc-neon-x8.c", |
| "src/f32-vbinary/gen/vsub-neon-x4.c", |
| "src/f32-vbinary/gen/vsub-neon-x8.c", |
| "src/f32-vbinary/gen/vsubc-neon-x4.c", |
| "src/f32-vbinary/gen/vsubc-neon-x8.c", |
| "src/f32-vmulcaddc/gen/c4-neon-2x.c", |
| "src/f32-vmulcaddc/gen/c8-neon-2x.c", |
| "src/q8-avgpool/mp9p8q-neon.c", |
| "src/q8-avgpool/up9-neon.c", |
| "src/q8-dwconv/up8x9-neon.c", |
| "src/q8-gavgpool/mp7p7q-neon.c", |
| "src/q8-gavgpool/up7-neon.c", |
| "src/q8-gemm/4x8-neon.c", |
| "src/q8-gemm/8x8-neon.c", |
| "src/q8-igemm/4x8-neon.c", |
| "src/q8-igemm/8x8-neon.c", |
| "src/q8-vadd/neon.c", |
| "src/u8-clamp/neon.c", |
| "src/u8-maxpool/9p8x-neon-c16.c", |
| "src/u8-rmax/neon.c", |
| "src/x32-packx/x4-neon-st4.c", |
| "src/x32-pad/x2-neon.c", |
| "src/x32-zip/x2-neon.c", |
| "src/x32-zip/x3-neon.c", |
| "src/x32-zip/x4-neon.c", |
| "src/x32-zip/xm-neon.c", |
| "src/x8-zip/x2-neon.c", |
| "src/x8-zip/x3-neon.c", |
| "src/x8-zip/x4-neon.c", |
| "src/x8-zip/xm-neon.c", |
| ] |
| |
| NEONFMA_UKERNELS = [ |
| "src/f32-bilinear/gen/neonfma-c4.c", |
| "src/f32-bilinear/gen/neonfma-c8.c", |
| "src/f32-igemm/gen/1x8-neonfma-dup-ld64.c", |
| "src/f32-igemm/gen/4x8-neonfma-dup-ld128.c", |
| "src/f32-igemm/gen/4x8-neonfma-dup-ld64.c", |
| "src/f32-igemm/gen/6x8-neonfma-dup-ld64.c", |
| "src/f32-igemm/gen/6x8-neonfma-dup-ld128.c", |
| "src/f32-igemm/gen/1x8s4-neonfma.c", |
| "src/f32-igemm/gen/4x8s4-neonfma.c", |
| "src/f32-igemm/gen/6x8s4-neonfma.c", |
| "src/f32-igemm/gen/8x8s4-neonfma.c", |
| "src/f32-dwconv/gen/up4x9-neonfma.c", |
| "src/f32-dwconv/gen/up4x9-neonfma-acc2.c", |
| "src/f32-dwconv/gen/up8x9-neonfma.c", |
| "src/f32-dwconv/gen/up8x9-neonfma-acc2.c", |
| "src/f32-gemm/gen/1x8-neonfma-dup-ld64.c", |
| "src/f32-gemm/gen/4x8-neonfma-dup-ld128.c", |
| "src/f32-gemm/gen/4x8-neonfma-dup-ld64.c", |
| "src/f32-gemm/gen/6x8-neonfma-dup-ld64.c", |
| "src/f32-gemm/gen/6x8-neonfma-dup-ld128.c", |
| "src/f32-gemm/gen/1x8s4-neonfma.c", |
| "src/f32-gemm/gen/4x8s4-neonfma.c", |
| "src/f32-gemm/gen/6x8s4-neonfma.c", |
| "src/f32-gemm/gen/8x8s4-neonfma.c", |
| "src/f32-gemm/gen-inc/1x8-neonfma-dup-ld64.c", |
| "src/f32-gemm/gen-inc/4x8-neonfma-dup-ld128.c", |
| "src/f32-gemm/gen-inc/4x8-neonfma-dup-ld64.c", |
| "src/f32-gemm/gen-inc/6x8-neonfma-dup-ld64.c", |
| "src/f32-gemm/gen-inc/6x8-neonfma-dup-ld128.c", |
| "src/f32-gemm/gen-inc/1x8s4-neonfma.c", |
| "src/f32-gemm/gen-inc/4x8s4-neonfma.c", |
| "src/f32-gemm/gen-inc/6x8s4-neonfma.c", |
| "src/f32-gemm/gen-inc/8x8s4-neonfma.c", |
| "src/f32-hswish/neonfma.c", |
| "src/f32-ppmm/gen/4x8-neonfma.c", |
| "src/f32-ppmm/gen/8x8-neonfma.c", |
| "src/f32-sigmoid/gen/neonfma-p5-nr2fma-x16.c", |
| "src/f32-vmulcaddc/gen/c4-neonfma-2x.c", |
| "src/f32-vmulcaddc/gen/c8-neonfma-2x.c", |
| "src/math/exp-neonfma-lut64-p2.c", |
| "src/math/exp-neonfma-p5.c", |
| "src/math/expminus-neonfma-lut2048-p1.c", |
| "src/math/expminus-neonfma-lut64-p2.c", |
| "src/math/expminus-neonfma-p5.c", |
| "src/math/sigmoid-neonfma-lut2048-p1-nr1recps1fma.c", |
| "src/math/sigmoid-neonfma-lut2048-p1-nr2fma.c", |
| "src/math/sigmoid-neonfma-lut2048-p1-nr2recps.c", |
| "src/math/sigmoid-neonfma-p5-nr1recps1fma.c", |
| "src/math/sigmoid-neonfma-p5-nr2fma.c", |
| "src/math/sigmoid-neonfma-p5-nr2recps.c", |
| ] |
| |
| AARCH64_NEONFMA_UKERNELS = [ |
| "src/f32-vbinary/gen/vdiv-neon-x4.c", |
| "src/f32-vbinary/gen/vdiv-neon-x8.c", |
| "src/f32-vbinary/gen/vdivc-neon-x4.c", |
| "src/f32-vbinary/gen/vdivc-neon-x8.c", |
| "src/f32-vbinary/gen/vrdivc-neon-x4.c", |
| "src/f32-vbinary/gen/vrdivc-neon-x8.c", |
| "src/f32-gemm/gen/1x8-neonfma-lane-ld64.c", |
| "src/f32-gemm/gen/4x2-neonfma-lane-ld64.c", |
| "src/f32-gemm/gen/4x8-neonfma-lane-ld128.c", |
| "src/f32-gemm/gen/4x8-neonfma-lane-ld64.c", |
| "src/f32-gemm/gen/5x8-neonfma-lane-ld64.c", |
| "src/f32-gemm/gen/6x8-neonfma-lane-ld64.c", |
| "src/f32-gemm/gen/6x8-neonfma-lane-ld128.c", |
| "src/f32-gemm/gen-inc/1x8-neonfma-lane-ld64.c", |
| "src/f32-gemm/gen-inc/4x8-neonfma-lane-ld128.c", |
| "src/f32-gemm/gen-inc/4x8-neonfma-lane-ld64.c", |
| "src/f32-gemm/gen-inc/5x8-neonfma-lane-ld64.c", |
| "src/f32-gemm/gen-inc/6x8-neonfma-lane-ld64.c", |
| "src/f32-gemm/gen-inc/6x8-neonfma-lane-ld128.c", |
| "src/f32-igemm/gen/1x8-neonfma-lane-ld64.c", |
| "src/f32-igemm/gen/4x2-neonfma-lane-ld64.c", |
| "src/f32-igemm/gen/4x4-neonfma-lane-ld64.c", |
| "src/f32-igemm/gen/4x8-neonfma-lane-ld128.c", |
| "src/f32-igemm/gen/4x8-neonfma-lane-ld64.c", |
| "src/f32-igemm/gen/6x8-neonfma-lane-ld64.c", |
| "src/f32-igemm/gen/6x8-neonfma-lane-ld128.c", |
| "src/f32-conv-hwc/3x3s2p1c3x4-neonfma-2x2.c", |
| "src/f32-conv-hwc/3x3s2p1c3x8-neonfma-2x2.c", |
| "src/f32-conv-hwc2spchw/3x3s2p1c3x4-neonfma-2x2.c", |
| "src/f32-dwconv-spchw/3x3p1-neonfma.c", |
| "src/f32-dwconv-spchw/5x5p2-neonfma.c", |
| "src/f32-dwconv-spchw/3x3s2p1-neonfma.c", |
| "src/f32-dwconv-spchw/5x5s2p2-neonfma.c", |
| "src/f32-spmm/gen/12x1-neonfma.c", |
| "src/f32-spmm/gen/12x2-neonfma.c", |
| "src/f32-spmm/gen/12x4-neonfma.c", |
| "src/f32-spmm/gen/16x1-neonfma-pipelined.c", |
| "src/f32-spmm/gen/16x1-neonfma-unroll2.c", |
| "src/f32-spmm/gen/16x1-neonfma.c", |
| "src/f32-spmm/gen/16x2-neonfma.c", |
| "src/f32-spmm/gen/16x4-neonfma.c", |
| "src/f32-spmm/gen/4x1-neonfma-pipelined.c", |
| "src/f32-spmm/gen/4x1-neonfma-unroll2.c", |
| "src/f32-spmm/gen/4x1-neonfma.c", |
| "src/f32-spmm/gen/4x2-neonfma.c", |
| "src/f32-spmm/gen/4x4-neonfma.c", |
| "src/f32-spmm/gen/8x1-neonfma-pipelined.c", |
| "src/f32-spmm/gen/8x1-neonfma-unroll2.c", |
| "src/f32-spmm/gen/8x1-neonfma.c", |
| "src/f32-spmm/gen/8x2-neonfma.c", |
| "src/f32-spmm/gen/8x4-neonfma.c", |
| "src/math/sigmoid-neonfma-p5-div.c", |
| "src/math/sigmoid-neonfma-lut2048-p1-div.c", |
| ] |
| |
| AARCH64_NEONFP16ARITH_UKERNELS = [ |
| "src/f16-gemm/gen/4x8-neonfp16arith-ld64.c", |
| "src/f16-gemm/gen/6x8-neonfp16arith-ld64.c", |
| "src/f16-gemm/gen/8x8-neonfp16arith-ld64.c", |
| ] |
| |
| SSE_UKERNELS = [ |
| "src/f32-avgpool/mp9p8q-sse.c", |
| "src/f32-avgpool/up9-sse.c", |
| "src/f32-bilinear/gen/sse-c4.c", |
| "src/f32-bilinear/gen/sse-c8.c", |
| "src/f32-clamp/sse.c", |
| "src/f32-dwconv-spchw/3x3p1-sse.c", |
| "src/f32-dwconv-spchw/3x3s2p1-sse.c", |
| "src/f32-dwconv/gen/up4x25-sse-acc2.c", |
| "src/f32-dwconv/gen/up4x25-sse.c", |
| "src/f32-dwconv/gen/up4x4-sse-acc2.c", |
| "src/f32-dwconv/gen/up4x4-sse.c", |
| "src/f32-dwconv/gen/up4x9-sse-acc2.c", |
| "src/f32-dwconv/gen/up4x9-sse.c", |
| "src/f32-dwconv/gen/up8x25-sse-acc2.c", |
| "src/f32-dwconv/gen/up8x25-sse.c", |
| "src/f32-dwconv/gen/up8x4-sse-acc2.c", |
| "src/f32-dwconv/gen/up8x4-sse.c", |
| "src/f32-dwconv/gen/up8x9-sse-acc2.c", |
| "src/f32-dwconv/gen/up8x9-sse.c", |
| "src/f32-gavgpool-spchw/sse-x4.c", |
| "src/f32-gavgpool/mp7p7q-sse.c", |
| "src/f32-gavgpool/up7-sse.c", |
| "src/f32-gemm/gen/1x8-sse-dup.c", |
| "src/f32-gemm/gen/1x8-sse-load1.c", |
| "src/f32-gemm/gen/1x8s4-sse.c", |
| "src/f32-gemm/gen/4x8-sse-dup.c", |
| "src/f32-gemm/gen/4x8-sse-load1.c", |
| "src/f32-gemm/gen/4x8s4-sse.c", |
| "src/f32-gemm/gen-inc/1x8-sse-dup.c", |
| "src/f32-gemm/gen-inc/1x8-sse-load1.c", |
| "src/f32-gemm/gen-inc/1x8s4-sse.c", |
| "src/f32-gemm/gen-inc/4x8-sse-dup.c", |
| "src/f32-gemm/gen-inc/4x8-sse-load1.c", |
| "src/f32-gemm/gen-inc/4x8s4-sse.c", |
| "src/f32-hswish/sse.c", |
| "src/f32-igemm/gen/1x8-sse-dup.c", |
| "src/f32-igemm/gen/1x8-sse-load1.c", |
| "src/f32-igemm/gen/1x8s4-sse.c", |
| "src/f32-igemm/gen/4x2c4-sse.c", |
| "src/f32-igemm/gen/4x8-sse-dup.c", |
| "src/f32-igemm/gen/4x8-sse-load1.c", |
| "src/f32-igemm/gen/4x8s4-sse.c", |
| "src/f32-maxpool/9p8x-sse-c4.c", |
| "src/f32-pavgpool/mp9p8q-sse.c", |
| "src/f32-pavgpool/up9-sse.c", |
| "src/f32-ppmm/gen/4x8-sse.c", |
| "src/f32-rmax/sse.c", |
| "src/f32-spmm/gen/4x1-sse.c", |
| "src/f32-spmm/gen/8x1-sse.c", |
| "src/f32-vbinary/gen/vadd-sse-x4.c", |
| "src/f32-vbinary/gen/vadd-sse-x8.c", |
| "src/f32-vbinary/gen/vaddc-sse-x4.c", |
| "src/f32-vbinary/gen/vaddc-sse-x8.c", |
| "src/f32-vbinary/gen/vdiv-sse-x4.c", |
| "src/f32-vbinary/gen/vdiv-sse-x8.c", |
| "src/f32-vbinary/gen/vdivc-sse-x4.c", |
| "src/f32-vbinary/gen/vdivc-sse-x8.c", |
| "src/f32-vbinary/gen/vmax-sse-x4.c", |
| "src/f32-vbinary/gen/vmax-sse-x8.c", |
| "src/f32-vbinary/gen/vmaxc-sse-x4.c", |
| "src/f32-vbinary/gen/vmaxc-sse-x8.c", |
| "src/f32-vbinary/gen/vmin-sse-x4.c", |
| "src/f32-vbinary/gen/vmin-sse-x8.c", |
| "src/f32-vbinary/gen/vminc-sse-x4.c", |
| "src/f32-vbinary/gen/vminc-sse-x8.c", |
| "src/f32-vbinary/gen/vmul-sse-x4.c", |
| "src/f32-vbinary/gen/vmul-sse-x8.c", |
| "src/f32-vbinary/gen/vmulc-sse-x4.c", |
| "src/f32-vbinary/gen/vmulc-sse-x8.c", |
| "src/f32-vbinary/gen/vrdivc-sse-x4.c", |
| "src/f32-vbinary/gen/vrdivc-sse-x8.c", |
| "src/f32-vbinary/gen/vrsubc-sse-x4.c", |
| "src/f32-vbinary/gen/vrsubc-sse-x8.c", |
| "src/f32-vbinary/gen/vsub-sse-x4.c", |
| "src/f32-vbinary/gen/vsub-sse-x8.c", |
| "src/f32-vbinary/gen/vsubc-sse-x4.c", |
| "src/f32-vbinary/gen/vsubc-sse-x8.c", |
| "src/f32-vmulcaddc/gen/c4-sse-2x.c", |
| "src/f32-vmulcaddc/gen/c8-sse-2x.c", |
| "src/x32-packx/x4-sse.c", |
| ] |
| |
| SSE2_UKERNELS = [ |
| "src/f32-argmaxpool/9p8x-sse2-c4.c", |
| "src/f32-argmaxpool/4x-sse2-c4.c", |
| "src/f32-argmaxpool/9x-sse2-c4.c", |
| "src/f32-prelu/gen/sse2-2x4.c", |
| "src/f32-prelu/gen/sse2-2x8.c", |
| "src/f32-sigmoid/gen/sse2-p5-div-x8.c", |
| "src/f32-sigmoid/gen/sse2-p5-div-x16.c", |
| "src/q8-avgpool/mp9p8q-sse2.c", |
| "src/q8-avgpool/up9-sse2.c", |
| "src/q8-igemm/4x4c2-sse2.c", |
| "src/q8-dwconv/up8x9-sse2.c", |
| "src/q8-gavgpool/mp7p7q-sse2.c", |
| "src/q8-gavgpool/up7-sse2.c", |
| "src/q8-gemm/2x4c8-sse2.c", |
| "src/q8-gemm/4x4c2-sse2.c", |
| "src/q8-vadd/sse2.c", |
| "src/u8-clamp/sse2.c", |
| "src/u8-maxpool/9p8x-sse2-c16.c", |
| "src/u8-rmax/sse2.c", |
| "src/x32-pad/x2-sse2.c", |
| "src/x32-zip/x2-sse2.c", |
| "src/x32-zip/x3-sse2.c", |
| "src/x32-zip/x4-sse2.c", |
| "src/x32-zip/xm-sse2.c", |
| "src/x8-zip/x2-sse2.c", |
| "src/x8-zip/x3-sse2.c", |
| "src/x8-zip/x4-sse2.c", |
| "src/x8-zip/xm-sse2.c", |
| "src/math/exp-sse2-p5.c", |
| "src/math/expminus-sse2-p5.c", |
| "src/math/sigmoid-sse2-p5-div.c", |
| ] |
| |
| SSE41_UKERNELS = [ |
| "src/f32-prelu/gen/sse41-2x4.c", |
| "src/f32-prelu/gen/sse41-2x8.c", |
| "src/f32-sigmoid/gen/sse41-p5-div-x8.c", |
| "src/f32-sigmoid/gen/sse41-p5-div-x16.c", |
| ] |
| |
| AVX_UKERNELS = [ |
| "src/f32-clamp/avx.c", |
| "src/f32-dwconv/gen/up16x4-avx-acc2.c", |
| "src/f32-dwconv/gen/up16x4-avx.c", |
| "src/f32-dwconv/gen/up8x4-avx-acc2.c", |
| "src/f32-dwconv/gen/up8x4-avx.c", |
| "src/f32-dwconv/gen/up16x9-avx-acc2.c", |
| "src/f32-dwconv/gen/up16x9-avx.c", |
| "src/f32-dwconv/gen/up8x9-avx-acc2.c", |
| "src/f32-dwconv/gen/up8x9-avx.c", |
| "src/f32-dwconv/gen/up16x25-avx-acc2.c", |
| "src/f32-dwconv/gen/up16x25-avx.c", |
| "src/f32-dwconv/gen/up8x25-avx-acc2.c", |
| "src/f32-dwconv/gen/up8x25-avx.c", |
| "src/f32-gemm/gen/1x8-avx-broadcast.c", |
| "src/f32-gemm/gen/4x8-avx-broadcast.c", |
| "src/f32-gemm/gen/5x8-avx-broadcast.c", |
| "src/f32-gemm/gen/6x8-avx-broadcast.c", |
| "src/f32-gemm/gen/7x8-avx-broadcast.c", |
| "src/f32-gemm/gen/1x16-avx-broadcast.c", |
| "src/f32-gemm/gen/3x16-avx-broadcast.c", |
| "src/f32-gemm/gen/4x16-avx-broadcast.c", |
| "src/f32-gemm/gen/5x16-avx-broadcast.c", |
| "src/f32-gemm/gen-inc/1x8-avx-broadcast.c", |
| "src/f32-gemm/gen-inc/4x8-avx-broadcast.c", |
| "src/f32-gemm/gen-inc/5x8-avx-broadcast.c", |
| "src/f32-gemm/gen-inc/6x8-avx-broadcast.c", |
| "src/f32-gemm/gen-inc/7x8-avx-broadcast.c", |
| "src/f32-gemm/gen-inc/1x16-avx-broadcast.c", |
| "src/f32-gemm/gen-inc/3x16-avx-broadcast.c", |
| "src/f32-gemm/gen-inc/4x16-avx-broadcast.c", |
| "src/f32-gemm/gen-inc/5x16-avx-broadcast.c", |
| "src/f32-igemm/gen/1x8-avx-broadcast.c", |
| "src/f32-igemm/gen/4x8-avx-broadcast.c", |
| "src/f32-igemm/gen/5x8-avx-broadcast.c", |
| "src/f32-igemm/gen/6x8-avx-broadcast.c", |
| "src/f32-igemm/gen/7x8-avx-broadcast.c", |
| "src/f32-igemm/gen/1x16-avx-broadcast.c", |
| "src/f32-igemm/gen/3x16-avx-broadcast.c", |
| "src/f32-igemm/gen/4x16-avx-broadcast.c", |
| "src/f32-igemm/gen/5x16-avx-broadcast.c", |
| "src/f32-rmax/avx.c", |
| "src/f32-vscale/avx-unroll32.c", |
| ] |
| |
| FMA3_UKERNELS = [ |
| "src/f32-dwconv/gen/up16x4-fma3-acc2.c", |
| "src/f32-dwconv/gen/up16x4-fma3.c", |
| "src/f32-dwconv/gen/up8x4-fma3-acc2.c", |
| "src/f32-dwconv/gen/up8x4-fma3.c", |
| "src/f32-dwconv/gen/up16x9-fma3-acc2.c", |
| "src/f32-dwconv/gen/up16x9-fma3.c", |
| "src/f32-dwconv/gen/up8x9-fma3-acc2.c", |
| "src/f32-dwconv/gen/up8x9-fma3.c", |
| "src/f32-dwconv/gen/up16x25-fma3-acc2.c", |
| "src/f32-dwconv/gen/up16x25-fma3.c", |
| "src/f32-dwconv/gen/up8x25-fma3-acc2.c", |
| "src/f32-dwconv/gen/up8x25-fma3.c", |
| "src/f32-gemm/gen/1x8-fma3-broadcast.c", |
| "src/f32-gemm/gen/4x8-fma3-broadcast.c", |
| "src/f32-gemm/gen/5x8-fma3-broadcast.c", |
| "src/f32-gemm/gen/6x8-fma3-broadcast.c", |
| "src/f32-gemm/gen/7x8-fma3-broadcast.c", |
| "src/f32-gemm/gen/8x8-fma3-broadcast.c", |
| "src/f32-gemm/gen/1x16-fma3-broadcast.c", |
| "src/f32-gemm/gen/3x16-fma3-broadcast.c", |
| "src/f32-gemm/gen/4x16-fma3-broadcast.c", |
| "src/f32-gemm/gen/5x16-fma3-broadcast.c", |
| "src/f32-gemm/gen-inc/1x8-fma3-broadcast.c", |
| "src/f32-gemm/gen-inc/4x8-fma3-broadcast.c", |
| "src/f32-gemm/gen-inc/5x8-fma3-broadcast.c", |
| "src/f32-gemm/gen-inc/6x8-fma3-broadcast.c", |
| "src/f32-gemm/gen-inc/7x8-fma3-broadcast.c", |
| "src/f32-gemm/gen-inc/8x8-fma3-broadcast.c", |
| "src/f32-gemm/gen-inc/1x16-fma3-broadcast.c", |
| "src/f32-gemm/gen-inc/3x16-fma3-broadcast.c", |
| "src/f32-gemm/gen-inc/4x16-fma3-broadcast.c", |
| "src/f32-gemm/gen-inc/5x16-fma3-broadcast.c", |
| "src/f32-igemm/gen/1x8-fma3-broadcast.c", |
| "src/f32-igemm/gen/4x8-fma3-broadcast.c", |
| "src/f32-igemm/gen/5x8-fma3-broadcast.c", |
| "src/f32-igemm/gen/6x8-fma3-broadcast.c", |
| "src/f32-igemm/gen/7x8-fma3-broadcast.c", |
| "src/f32-igemm/gen/8x8-fma3-broadcast.c", |
| "src/f32-igemm/gen/1x16-fma3-broadcast.c", |
| "src/f32-igemm/gen/3x16-fma3-broadcast.c", |
| "src/f32-igemm/gen/4x16-fma3-broadcast.c", |
| "src/f32-igemm/gen/5x16-fma3-broadcast.c", |
| ] |
| |
| AVX2_UKERNELS = [ |
| "src/f32-raddexpminusmax/avx2-p5-unroll64.c", |
| "src/f32-raddextexp/avx2-p5-unroll64.c", |
| "src/f32-raddstoreexpminusmax/avx2-p5-unroll64.c", |
| "src/f32-vscaleexpminusmax/avx2-p5-unroll64.c", |
| "src/f32-vscaleextexp/avx2-p5-unroll64.c", |
| "src/math/exp-avx2-p5.c", |
| "src/math/exp-avx2-perm-p3.c", |
| "src/math/exp-avx2-perm-p4.c", |
| "src/math/expminus-avx2-p5.c", |
| "src/math/extexp-avx2-p5.c", |
| ] |
| |
| AVX512F_UKERNELS = [ |
| "src/f32-clamp/avx512f.c", |
| "src/f32-dwconv/gen/up32x4-avx512f-acc2.c", |
| "src/f32-dwconv/gen/up32x4-avx512f.c", |
| "src/f32-dwconv/gen/up16x4-avx512f-acc2.c", |
| "src/f32-dwconv/gen/up16x4-avx512f.c", |
| "src/f32-dwconv/gen/up32x9-avx512f-acc2.c", |
| "src/f32-dwconv/gen/up32x9-avx512f.c", |
| "src/f32-dwconv/gen/up16x9-avx512f-acc2.c", |
| "src/f32-dwconv/gen/up16x9-avx512f.c", |
| "src/f32-dwconv/gen/up32x25-avx512f-acc2.c", |
| "src/f32-dwconv/gen/up32x25-avx512f.c", |
| "src/f32-dwconv/gen/up16x25-avx512f-acc2.c", |
| "src/f32-dwconv/gen/up16x25-avx512f.c", |
| "src/f32-gemm/gen/1x16-avx512f-broadcast.c", |
| "src/f32-gemm/gen/4x16-avx512f-broadcast.c", |
| "src/f32-gemm/gen/5x16-avx512f-broadcast.c", |
| "src/f32-gemm/gen/6x16-avx512f-broadcast.c", |
| "src/f32-gemm/gen/7x16-avx512f-broadcast.c", |
| "src/f32-gemm/gen/8x16-avx512f-broadcast.c", |
| "src/f32-gemm/gen-inc/1x16-avx512f-broadcast.c", |
| "src/f32-gemm/gen-inc/4x16-avx512f-broadcast.c", |
| "src/f32-gemm/gen-inc/5x16-avx512f-broadcast.c", |
| "src/f32-gemm/gen-inc/6x16-avx512f-broadcast.c", |
| "src/f32-gemm/gen-inc/7x16-avx512f-broadcast.c", |
| "src/f32-gemm/gen-inc/8x16-avx512f-broadcast.c", |
| "src/f32-igemm/gen/1x16-avx512f-broadcast.c", |
| "src/f32-igemm/gen/4x16-avx512f-broadcast.c", |
| "src/f32-igemm/gen/5x16-avx512f-broadcast.c", |
| "src/f32-igemm/gen/6x16-avx512f-broadcast.c", |
| "src/f32-igemm/gen/7x16-avx512f-broadcast.c", |
| "src/f32-igemm/gen/8x16-avx512f-broadcast.c", |
| "src/f32-raddexpminusmax/avx512f-p5-scalef-unroll128.c", |
| "src/f32-raddextexp/avx512f-p5-scalef-unroll128.c", |
| "src/f32-raddstoreexpminusmax/avx512f-p5-scalef-unroll128.c", |
| "src/f32-rmax/avx512f.c", |
| "src/f32-vscale/avx512f-unroll64.c", |
| "src/f32-vscaleexpminusmax/avx512f-p5-scalef-unroll128.c", |
| "src/f32-vscaleextexp/avx512f-p5-scalef-unroll128.c", |
| "src/math/exp-avx512f-p5-scalef.c", |
| "src/math/exp-avx512f-p5.c", |
| "src/math/exp-avx512f-perm-p3.c", |
| "src/math/exp-avx512f-perm2-p2.c", |
| "src/math/extexp-avx512f-p5.c", |
| ] |
| |
| AARCH32_ASM_UKERNELS = [ |
| "src/q8-dwconv/up8x9-aarch32-neon.S", |
| "src/f32-gemm/4x8-aarch32-neon-cortex-a75.S", |
| "src/f32-gemm/4x8-aarch32-neon-ld64.S", |
| ] |
| |
| AARCH64_ASM_UKERNELS = [ |
| "src/f32-dwconv/up4x9-aarch64-neonfma-cortex-a55.S", |
| "src/f32-dwconv/up4x9-aarch64-neonfma.S", |
| "src/f32-gemm/gen/1x12-aarch64-neonfma-cortex-a53.S", |
| "src/f32-gemm/gen/1x8-aarch64-neonfma-cortex-a53.S", |
| "src/f32-gemm/gen/1x8-aarch64-neonfma-cortex-a57.S", |
| "src/f32-gemm/gen/1x8-aarch64-neonfma-cortex-a75.S", |
| "src/f32-gemm/gen/4x12-aarch64-neonfma-cortex-a53.S", |
| "src/f32-gemm/gen/4x8-aarch64-neonfma-cortex-a53.S", |
| "src/f32-gemm/gen/4x8-aarch64-neonfma-cortex-a57.S", |
| "src/f32-gemm/gen/4x8-aarch64-neonfma-cortex-a75.S", |
| "src/f32-gemm/gen/4x8-aarch64-neonfma-ld128.S", |
| "src/f32-gemm/gen/4x8-aarch64-neonfma-ld64.S", |
| "src/f32-gemm/gen/5x8-aarch64-neonfma-cortex-a75.S", |
| "src/f32-gemm/gen/6x8-aarch64-neonfma-cortex-a53.S", |
| "src/f32-gemm/gen/6x8-aarch64-neonfma-cortex-a57.S", |
| "src/f32-gemm/gen/6x8-aarch64-neonfma-cortex-a73.S", |
| "src/f32-gemm/gen/6x8-aarch64-neonfma-cortex-a75.S", |
| "src/f32-gemm/gen/6x8-aarch64-neonfma-ld128.S", |
| "src/f32-gemm/gen/6x8-aarch64-neonfma-ld64.S", |
| "src/f32-gemm/gen-inc/1x12-aarch64-neonfma-cortex-a53.S", |
| "src/f32-gemm/gen-inc/1x8-aarch64-neonfma-cortex-a53.S", |
| "src/f32-gemm/gen-inc/1x8-aarch64-neonfma-cortex-a57.S", |
| "src/f32-gemm/gen-inc/1x8-aarch64-neonfma-cortex-a75.S", |
| "src/f32-gemm/gen-inc/4x12-aarch64-neonfma-cortex-a53.S", |
| "src/f32-gemm/gen-inc/4x8-aarch64-neonfma-cortex-a53.S", |
| "src/f32-gemm/gen-inc/4x8-aarch64-neonfma-cortex-a57.S", |
| "src/f32-gemm/gen-inc/4x8-aarch64-neonfma-cortex-a75.S", |
| "src/f32-gemm/gen-inc/4x8-aarch64-neonfma-ld128.S", |
| "src/f32-gemm/gen-inc/4x8-aarch64-neonfma-ld64.S", |
| "src/f32-gemm/gen-inc/5x8-aarch64-neonfma-cortex-a75.S", |
| "src/f32-gemm/gen-inc/6x8-aarch64-neonfma-cortex-a53.S", |
| "src/f32-gemm/gen-inc/6x8-aarch64-neonfma-cortex-a57.S", |
| "src/f32-gemm/gen-inc/6x8-aarch64-neonfma-cortex-a73.S", |
| "src/f32-gemm/gen-inc/6x8-aarch64-neonfma-cortex-a75.S", |
| "src/f32-gemm/gen-inc/6x8-aarch64-neonfma-ld128.S", |
| "src/f32-gemm/gen-inc/6x8-aarch64-neonfma-ld64.S", |
| "src/f32-igemm/1x12-aarch64-neonfma-cortex-a53.S", |
| "src/f32-igemm/1x8-aarch64-neonfma-cortex-a53.S", |
| "src/f32-igemm/1x8-aarch64-neonfma-cortex-a57.S", |
| "src/f32-igemm/1x8-aarch64-neonfma-cortex-a75.S", |
| "src/f32-igemm/4x12-aarch64-neonfma-cortex-a53.S", |
| "src/f32-igemm/4x8-aarch64-neonfma-cortex-a53.S", |
| "src/f32-igemm/4x8-aarch64-neonfma-cortex-a75.S", |
| "src/f32-igemm/5x8-aarch64-neonfma-cortex-a75.S", |
| "src/f32-igemm/6x8-aarch64-neonfma-cortex-a53.S", |
| "src/f32-igemm/6x8-aarch64-neonfma-cortex-a57.S", |
| "src/f32-igemm/6x8-aarch64-neonfma-cortex-a73.S", |
| "src/f32-igemm/6x8-aarch64-neonfma-cortex-a75.S", |
| ] |
| |
| INTERNAL_MICROKERNEL_HDRS = [ |
| "src/xnnpack/argmaxpool.h", |
| "src/xnnpack/avgpool.h", |
| "src/xnnpack/bilinear.h", |
| "src/xnnpack/clamp.h", |
| "src/xnnpack/common.h", |
| "src/xnnpack/conv.h", |
| "src/xnnpack/dwconv.h", |
| "src/xnnpack/gavgpool.h", |
| "src/xnnpack/gemm.h", |
| "src/xnnpack/hswish.h", |
| "src/xnnpack/igemm.h", |
| "src/xnnpack/intrinsics-polyfill.h", |
| "src/xnnpack/lut.h", |
| "src/xnnpack/math.h", |
| "src/xnnpack/maxpool.h", |
| "src/xnnpack/memory.h", |
| "src/xnnpack/packx.h", |
| "src/xnnpack/pad.h", |
| "src/xnnpack/params.h", |
| "src/xnnpack/pavgpool.h", |
| "src/xnnpack/ppmm.h", |
| "src/xnnpack/prelu.h", |
| "src/xnnpack/raddexpminusmax.h", |
| "src/xnnpack/raddextexp.h", |
| "src/xnnpack/raddstoreexpminusmax.h", |
| "src/xnnpack/rmax.h", |
| "src/xnnpack/scalar-utils.h", |
| "src/xnnpack/spmm.h", |
| "src/xnnpack/unpool.h", |
| "src/xnnpack/vadd.h", |
| "src/xnnpack/vbinary.h", |
| "src/xnnpack/vmulcaddc.h", |
| "src/xnnpack/vscale.h", |
| "src/xnnpack/vscaleexpminusmax.h", |
| "src/xnnpack/vscaleextexp.h", |
| "src/xnnpack/vunary.h", |
| "src/xnnpack/zip.h", |
| ] |
| |
| INTERNAL_HDRS = INTERNAL_MICROKERNEL_HDRS + [ |
| "include/xnnpack.h", |
| "src/xnnpack/allocator.h", |
| "src/xnnpack/compute.h", |
| "src/xnnpack/im2col.h", |
| "src/xnnpack/indirection.h", |
| "src/xnnpack/math-stubs.h", |
| "src/xnnpack/operator.h", |
| "src/xnnpack/pack.h", |
| "src/xnnpack/params-init.h", |
| "src/xnnpack/requantization-stubs.h", |
| "src/xnnpack/requantization.h", |
| ] |
| |
| ACCURACY_EVAL_HDRS = INTERNAL_MICROKERNEL_HDRS + [ |
| "src/xnnpack/math-stubs.h", |
| ] |
| |
| MICROKERNEL_BENCHMARK_HDRS = INTERNAL_MICROKERNEL_HDRS + [ |
| "src/xnnpack/params-init.h", |
| "include/xnnpack.h", |
| ] |
| |
| MICROKERNEL_TEST_HDRS = INTERNAL_MICROKERNEL_HDRS + [ |
| "src/xnnpack/isa-checks.h", |
| "src/xnnpack/params-init.h", |
| "src/xnnpack/requantization.h", |
| "include/xnnpack.h", |
| ] |
| |
| OPERATOR_TEST_PARAMS_HDRS = [ |
| "src/xnnpack/params.h", |
| "src/xnnpack/common.h", |
| ] |
| |
| WEIGHTS_PACK_HDRS = [ |
| "src/xnnpack/pack.h", |
| "src/xnnpack/operator.h", |
| "src/xnnpack/compute.h", |
| ] |
| |
| LOGGING_COPTS = select({ |
| # No logging in optimized mode |
| ":optimized_build": ["-DXNN_LOG_LEVEL=0"], |
| # Full logging in debug mode |
| ":debug_build": ["-DXNN_LOG_LEVEL=5"], |
| # Error-only logging in default (fastbuild) mode |
| "//conditions:default": ["-DXNN_LOG_LEVEL=2"], |
| }) |
| |
| LOGGING_HDRS = [ |
| "src/xnnpack/log.h", |
| ] |
| |
| xnnpack_cc_library( |
| name = "scalar_ukernels", |
| srcs = SCALAR_UKERNELS, |
| hdrs = INTERNAL_HDRS, |
| aarch32_copts = ["-marm"], |
| copts = xnnpack_std_copts(), |
| deps = [ |
| "@FP16", |
| "@FXdiv", |
| "@pthreadpool", |
| ], |
| ) |
| |
| xnnpack_cc_library( |
| name = "wasm_ukernels", |
| hdrs = INTERNAL_HDRS, |
| copts = xnnpack_std_copts(), |
| wasm_srcs = WASM_UKERNELS, |
| deps = [ |
| "@FP16", |
| "@FXdiv", |
| "@pthreadpool", |
| ], |
| ) |
| |
| xnnpack_cc_library( |
| name = "psimd_ukernels", |
| srcs = PSIMD_UKERNELS, |
| hdrs = INTERNAL_HDRS, |
| aarch32_copts = [ |
| "-marm", |
| "-mfpu=neon", |
| ], |
| copts = xnnpack_std_copts(), |
| optimized_copts = [ |
| "-O3", |
| "-ffast-math", |
| ], |
| deps = [ |
| "@FP16", |
| "@psimd", |
| "@pthreadpool", |
| ], |
| ) |
| |
| xnnpack_cc_library( |
| name = "neon_ukernels", |
| hdrs = INTERNAL_HDRS, |
| aarch32_copts = [ |
| "-marm", |
| "-mfpu=neon", |
| ], |
| aarch32_srcs = NEON_UKERNELS, |
| aarch64_srcs = NEON_UKERNELS, |
| copts = xnnpack_std_copts(), |
| deps = [ |
| "@FP16", |
| "@pthreadpool", |
| ], |
| ) |
| |
| xnnpack_cc_library( |
| name = "neonfma_ukernels", |
| hdrs = INTERNAL_HDRS, |
| aarch32_copts = [ |
| "-marm", |
| "-mfpu=neon-vfpv4", |
| ], |
| aarch32_srcs = NEONFMA_UKERNELS, |
| aarch64_srcs = NEONFMA_UKERNELS + AARCH64_NEONFMA_UKERNELS, |
| copts = xnnpack_std_copts(), |
| deps = [ |
| "@FP16", |
| "@pthreadpool", |
| ], |
| ) |
| |
| xnnpack_cc_library( |
| name = "neonfp16arith_ukernels", |
| hdrs = INTERNAL_HDRS, |
| aarch64_copts = ["-march=armv8.2-a+fp16"], |
| aarch64_srcs = AARCH64_NEONFP16ARITH_UKERNELS, |
| copts = xnnpack_std_copts(), |
| deps = [ |
| "@FP16", |
| "@pthreadpool", |
| ], |
| ) |
| |
| xnnpack_cc_library( |
| name = "sse2_ukernels", |
| hdrs = INTERNAL_HDRS, |
| copts = xnnpack_std_copts(), |
| x86_copts = ["-msse2"], |
| x86_srcs = SSE_UKERNELS + SSE2_UKERNELS, |
| deps = [ |
| "@FP16", |
| "@pthreadpool", |
| ], |
| ) |
| |
| xnnpack_cc_library( |
| name = "sse41_ukernels", |
| hdrs = INTERNAL_HDRS, |
| copts = xnnpack_std_copts(), |
| x86_copts = ["-msse4.1"], |
| x86_srcs = SSE41_UKERNELS, |
| deps = [ |
| "@FP16", |
| "@pthreadpool", |
| ], |
| ) |
| |
| xnnpack_cc_library( |
| name = "avx_ukernels", |
| hdrs = INTERNAL_HDRS, |
| copts = xnnpack_std_copts(), |
| x86_copts = ["-mavx"], |
| x86_srcs = AVX_UKERNELS, |
| deps = [ |
| "@FP16", |
| "@pthreadpool", |
| ], |
| ) |
| |
| xnnpack_cc_library( |
| name = "fma3_ukernels", |
| hdrs = INTERNAL_HDRS, |
| copts = xnnpack_std_copts(), |
| x86_copts = [ |
| "-mfma", |
| ], |
| x86_srcs = FMA3_UKERNELS, |
| deps = [ |
| "@FP16", |
| "@pthreadpool", |
| ], |
| ) |
| |
| xnnpack_cc_library( |
| name = "avx2_ukernels", |
| hdrs = INTERNAL_HDRS, |
| copts = xnnpack_std_copts(), |
| x86_copts = [ |
| "-mfma", |
| "-mavx2", |
| ], |
| x86_srcs = AVX2_UKERNELS, |
| deps = [ |
| "@FP16", |
| "@pthreadpool", |
| ], |
| ) |
| |
| xnnpack_cc_library( |
| name = "avx512f_ukernels", |
| hdrs = INTERNAL_HDRS, |
| copts = xnnpack_std_copts(), |
| x86_copts = ["-mavx512f"], |
| x86_srcs = AVX512F_UKERNELS, |
| deps = [ |
| "@FP16", |
| "@pthreadpool", |
| ], |
| ) |
| |
| xnnpack_cc_library( |
| name = "asm_ukernels", |
| hdrs = ["src/xnnpack/assembly.h"], |
| aarch32_srcs = AARCH32_ASM_UKERNELS, |
| aarch64_srcs = AARCH64_ASM_UKERNELS, |
| ) |
| |
| xnnpack_aggregate_library( |
| name = "ukernels", |
| aarch32_deps = [ |
| ":psimd_ukernels", |
| ":neon_ukernels", |
| ":neonfma_ukernels", |
| ":asm_ukernels", |
| ], |
| aarch64_deps = [ |
| ":psimd_ukernels", |
| ":neon_ukernels", |
| ":neonfma_ukernels", |
| ":neonfp16arith_ukernels", |
| ":asm_ukernels", |
| ], |
| generic_deps = [":scalar_ukernels"], |
| wasm_deps = [ |
| ":wasm_ukernels", |
| ], |
| wasmsimd_deps = [ |
| ":wasm_ukernels", |
| ":psimd_ukernels", |
| ], |
| x86_deps = [ |
| ":psimd_ukernels", |
| ":sse2_ukernels", |
| ":sse41_ukernels", |
| ":avx_ukernels", |
| ":fma3_ukernels", |
| ":avx2_ukernels", |
| ":avx512f_ukernels", |
| ], |
| ) |
| |
| xnnpack_cc_library( |
| name = "im2col", |
| srcs = ["src/im2col.c"], |
| hdrs = [ |
| "src/xnnpack/common.h", |
| "src/xnnpack/im2col.h", |
| ], |
| copts = xnnpack_std_copts(), |
| ) |
| |
| xnnpack_cc_library( |
| name = "indirection", |
| srcs = ["src/indirection.c"], |
| hdrs = INTERNAL_HDRS, |
| copts = xnnpack_std_copts(), |
| deps = [ |
| "@FP16", |
| "@FXdiv", |
| "@pthreadpool", |
| ], |
| ) |
| |
| xnnpack_cc_library( |
| name = "operator_run", |
| srcs = ["src/operator-run.c"], |
| hdrs = INTERNAL_HDRS + LOGGING_HDRS, |
| copts = xnnpack_std_copts() + LOGGING_COPTS + [ |
| # Wrappers for multi-pass microkernels use VLAs for temporary buffers. |
| "-Wno-vla", |
| ], |
| deps = [ |
| "@FP16", |
| "@FXdiv", |
| "@clog", |
| "@pthreadpool", |
| ], |
| ) |
| |
| cc_library( |
| name = "enable_assembly", |
| defines = select({ |
| ":xnn_enable_assembly_explicit_true": ["XNN_ENABLE_ASSEMBLY=1"], |
| ":xnn_enable_assembly_explicit_false": ["XNN_ENABLE_ASSEMBLY=0"], |
| "//conditions:default": ["XNN_ENABLE_ASSEMBLY=1"], |
| }), |
| ) |
| |
| xnnpack_cc_library( |
| name = "operators", |
| srcs = OPERATOR_SRCS + [ |
| "src/memory.c", |
| "src/operator-delete.c", |
| ], |
| hdrs = INTERNAL_HDRS + LOGGING_HDRS, |
| copts = xnnpack_std_copts() + LOGGING_COPTS + [ |
| "-Isrc", |
| "-Iinclude", |
| ] + select({ |
| ":debug_build": [], |
| "//conditions:default": xnnpack_min_size_copts(), |
| }), |
| wasm_srcs = ["src/wasm-stubs.c"], |
| wasmsimd_srcs = ["src/wasm-stubs.c"], |
| deps = [ |
| ":indirection", |
| "@FP16", |
| "@FXdiv", |
| "@clog", |
| "@pthreadpool", |
| ], |
| ) |
| |
| cc_library( |
| name = "XNNPACK", |
| srcs = [ |
| "src/init.c", |
| ], |
| copts = xnnpack_std_copts() + LOGGING_COPTS + [ |
| "-Isrc", |
| "-Iinclude", |
| ] + select({ |
| ":debug_build": [], |
| "//conditions:default": xnnpack_min_size_copts(), |
| }), |
| includes = ["include"], |
| linkstatic = True, |
| textual_hdrs = ["include/xnnpack.h"], |
| visibility = xnnpack_visibility(), |
| deps = [ |
| ":enable_assembly", |
| ":ukernels", |
| ":operator_run", |
| ":operators", |
| "@clog", |
| "@pthreadpool", |
| ] + select({ |
| ":emscripten": [], |
| "//conditions:default": ["@cpuinfo"], |
| }), |
| ) |
| |
| cc_library( |
| name = "xnnpack_operators_nhwc_f32", |
| srcs = [ |
| "src/init.c", |
| ], |
| copts = xnnpack_std_copts() + LOGGING_COPTS + [ |
| "-Isrc", |
| "-Iinclude", |
| ] + select({ |
| ":debug_build": [], |
| "//conditions:default": xnnpack_min_size_copts(), |
| }), |
| defines = [ |
| "XNN_NO_Q8_OPERATORS", |
| "XNN_NO_U8_OPERATORS", |
| "XNN_NO_X8_OPERATORS", |
| "XNN_NO_NCHW_OPERATORS", |
| ], |
| includes = ["include"], |
| linkstatic = True, |
| textual_hdrs = ["include/xnnpack.h"], |
| visibility = xnnpack_visibility(), |
| deps = [ |
| ":enable_assembly", |
| ":ukernels", |
| ":operator_run", |
| ":operators", |
| "@clog", |
| "@pthreadpool", |
| ] + select({ |
| ":emscripten": [], |
| "//conditions:default": ["@cpuinfo"], |
| }), |
| ) |
| |
| xnnpack_cc_library( |
| name = "bench_utils", |
| srcs = ["bench/utils.cc"], |
| hdrs = ["bench/utils.h"], |
| copts = ["-Wno-unused-result"], |
| deps = [ |
| "@com_google_benchmark//:benchmark", |
| "@cpuinfo", |
| ], |
| ) |
| |
| ######################### Benchmarks for micro-kernels ######################### |
| |
| xnnpack_benchmark( |
| name = "q8_gemm_bench", |
| srcs = [ |
| "bench/gemm.h", |
| "bench/q8-gemm.cc", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_BENCHMARK_HDRS, |
| copts = ["-Wno-unused-function"] + xnnpack_optional_ruy_copts() + xnnpack_optional_gemmlowp_copts(), |
| deps = MICROKERNEL_BENCHMARK_DEPS + xnnpack_optional_ruy_deps() + xnnpack_optional_gemmlowp_deps(), |
| ) |
| |
| xnnpack_benchmark( |
| name = "f16_gemm_bench", |
| srcs = [ |
| "bench/f16-gemm.cc", |
| "bench/gemm.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_BENCHMARK_HDRS, |
| copts = ["-Wno-unused-function"], |
| deps = MICROKERNEL_BENCHMARK_DEPS, |
| ) |
| |
| xnnpack_benchmark( |
| name = "f32_igemm_bench", |
| srcs = [ |
| "bench/f32-igemm.cc", |
| "bench/conv.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + MICROKERNEL_BENCHMARK_HDRS, |
| deps = MICROKERNEL_BENCHMARK_DEPS + [":indirection"], |
| ) |
| |
| xnnpack_benchmark( |
| name = "f32_conv_hwc_bench", |
| srcs = [ |
| "bench/f32-conv-hwc.cc", |
| "bench/dconv.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_BENCHMARK_HDRS, |
| copts = ["-Wno-unused-function"], |
| deps = MICROKERNEL_BENCHMARK_DEPS, |
| ) |
| |
| xnnpack_benchmark( |
| name = "f32_conv_hwc2spchw_bench", |
| srcs = [ |
| "bench/f32-conv-hwc2spchw.cc", |
| "bench/dconv.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_BENCHMARK_HDRS, |
| copts = ["-Wno-unused-function"], |
| deps = MICROKERNEL_BENCHMARK_DEPS, |
| ) |
| |
| xnnpack_benchmark( |
| name = "f32_dwconv_bench", |
| srcs = [ |
| "bench/f32-dwconv.cc", |
| "bench/dwconv.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + MICROKERNEL_BENCHMARK_HDRS, |
| deps = MICROKERNEL_BENCHMARK_DEPS + [":indirection"], |
| ) |
| |
| xnnpack_benchmark( |
| name = "f32_dwconv_spchw_bench", |
| srcs = [ |
| "bench/f32-dwconv-spchw.cc", |
| "bench/dwconv.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_BENCHMARK_HDRS, |
| deps = MICROKERNEL_BENCHMARK_DEPS + [":indirection"], |
| ) |
| |
| xnnpack_benchmark( |
| name = "f32_gemm_bench", |
| srcs = [ |
| "bench/f32-gemm.cc", |
| "bench/gemm.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_BENCHMARK_HDRS, |
| copts = ["-Wno-unused-function"] + xnnpack_optional_ruy_copts(), |
| deps = MICROKERNEL_BENCHMARK_DEPS + xnnpack_optional_ruy_deps(), |
| ) |
| |
| xnnpack_benchmark( |
| name = "f32_rmax_bench", |
| srcs = [ |
| "bench/f32-rmax.cc", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + MICROKERNEL_BENCHMARK_HDRS, |
| deps = MICROKERNEL_BENCHMARK_DEPS, |
| ) |
| |
| xnnpack_benchmark( |
| name = "f32_sigmoid_bench", |
| srcs = [ |
| "bench/f32-sigmoid.cc", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + MICROKERNEL_BENCHMARK_HDRS, |
| copts = ["-Wno-unused-function"], |
| deps = MICROKERNEL_BENCHMARK_DEPS, |
| ) |
| |
| xnnpack_benchmark( |
| name = "f32_spmm_bench", |
| srcs = [ |
| "bench/f32-spmm.cc", |
| "bench/gemm.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + MICROKERNEL_BENCHMARK_HDRS, |
| copts = ["-Wno-unused-function"], |
| deps = MICROKERNEL_BENCHMARK_DEPS, |
| ) |
| |
| xnnpack_benchmark( |
| name = "f32_softargmax_bench", |
| srcs = [ |
| "bench/f32-softargmax.cc", |
| ] + MICROKERNEL_BENCHMARK_HDRS, |
| copts = ["-Wno-unused-function"], |
| deps = MICROKERNEL_BENCHMARK_DEPS, |
| ) |
| |
| xnnpack_benchmark( |
| name = "f32_im2col_gemm_bench", |
| srcs = [ |
| "bench/f32-im2col-gemm.cc", |
| "bench/conv.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_BENCHMARK_HDRS, |
| deps = MICROKERNEL_BENCHMARK_DEPS + [":im2col"], |
| ) |
| |
| ########################### Benchmarks for operators ########################### |
| |
| xnnpack_benchmark( |
| name = "add_bench", |
| srcs = ["bench/add.cc"], |
| deps = OPERATOR_BENCHMARK_DEPS, |
| ) |
| |
| xnnpack_benchmark( |
| name = "average_pooling_bench", |
| srcs = ["bench/average-pooling.cc"], |
| deps = OPERATOR_BENCHMARK_DEPS, |
| ) |
| |
| xnnpack_benchmark( |
| name = "channel_shuffle_bench", |
| srcs = ["bench/channel-shuffle.cc"], |
| deps = OPERATOR_BENCHMARK_DEPS, |
| ) |
| |
| xnnpack_benchmark( |
| name = "convolution_bench", |
| srcs = ["bench/convolution.cc"], |
| copts = xnnpack_optional_tflite_copts() + xnnpack_optional_armcl_copts(), |
| deps = OPERATOR_BENCHMARK_DEPS + xnnpack_optional_tflite_deps() + xnnpack_optional_armcl_deps(), |
| ) |
| |
| xnnpack_benchmark( |
| name = "deconvolution_bench", |
| srcs = ["bench/deconvolution.cc"], |
| copts = xnnpack_optional_tflite_copts(), |
| deps = OPERATOR_BENCHMARK_DEPS + xnnpack_optional_tflite_deps(), |
| ) |
| |
| xnnpack_benchmark( |
| name = "global_average_pooling_bench", |
| srcs = ["bench/global-average-pooling.cc"], |
| deps = OPERATOR_BENCHMARK_DEPS, |
| ) |
| |
| xnnpack_benchmark( |
| name = "max_pooling_bench", |
| srcs = ["bench/max-pooling.cc"], |
| deps = OPERATOR_BENCHMARK_DEPS, |
| ) |
| |
| xnnpack_benchmark( |
| name = "sigmoid_bench", |
| srcs = ["bench/sigmoid.cc"], |
| copts = xnnpack_optional_tflite_copts(), |
| deps = OPERATOR_BENCHMARK_DEPS + xnnpack_optional_tflite_deps(), |
| ) |
| |
| xnnpack_benchmark( |
| name = "prelu_bench", |
| srcs = ["bench/prelu.cc"], |
| copts = xnnpack_optional_tflite_copts(), |
| deps = OPERATOR_BENCHMARK_DEPS + xnnpack_optional_tflite_deps(), |
| ) |
| |
| xnnpack_benchmark( |
| name = "softargmax_bench", |
| srcs = ["bench/softargmax.cc"], |
| deps = OPERATOR_BENCHMARK_DEPS, |
| ) |
| |
| ############################# End-to-end benchmarks ############################ |
| |
| cc_library( |
| name = "mobilenet_v1", |
| srcs = ["models/mobilenet-v1.cc"], |
| hdrs = ["models/models.h"], |
| linkstatic = True, |
| deps = [ |
| ":XNNPACK", |
| "@pthreadpool", |
| ], |
| ) |
| |
| cc_library( |
| name = "mobilenet_v2", |
| srcs = ["models/mobilenet-v2.cc"], |
| hdrs = ["models/models.h"], |
| linkstatic = True, |
| deps = [ |
| ":XNNPACK", |
| "@pthreadpool", |
| ], |
| ) |
| |
| xnnpack_benchmark( |
| name = "f32_dwconv_e2e_bench", |
| srcs = ["bench/f32-dwconv-e2e.cc"] + MICROKERNEL_BENCHMARK_HDRS, |
| copts = ["-Wno-unused-function"], |
| deps = MICROKERNEL_BENCHMARK_DEPS + [ |
| ":XNNPACK", |
| ":mobilenet_v1", |
| ":mobilenet_v2", |
| ], |
| ) |
| |
| xnnpack_benchmark( |
| name = "f32_gemm_e2e_bench", |
| srcs = ["bench/f32-gemm-e2e.cc"] + MICROKERNEL_BENCHMARK_HDRS, |
| copts = ["-Wno-unused-function"], |
| deps = MICROKERNEL_BENCHMARK_DEPS + [ |
| ":XNNPACK", |
| ":mobilenet_v1", |
| ":mobilenet_v2", |
| ], |
| ) |
| |
| xnnpack_benchmark( |
| name = "end2end_bench", |
| srcs = ["bench/end2end.cc"], |
| deps = [ |
| ":XNNPACK", |
| ":bench_utils", |
| ":mobilenet_v1", |
| ":mobilenet_v2", |
| "@pthreadpool", |
| ], |
| ) |
| |
| #################### Accuracy evaluation for math functions #################### |
| |
| xnnpack_benchmark( |
| name = "f32_exp_eval", |
| srcs = [ |
| "eval/f32-exp.cc", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + ACCURACY_EVAL_HDRS, |
| deps = ACCURACY_EVAL_DEPS, |
| ) |
| |
| xnnpack_benchmark( |
| name = "f32_expminus_eval", |
| srcs = [ |
| "eval/f32-expminus.cc", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + ACCURACY_EVAL_HDRS, |
| deps = ACCURACY_EVAL_DEPS, |
| ) |
| |
| xnnpack_benchmark( |
| name = "f32_extexp_eval", |
| srcs = [ |
| "eval/f32-extexp.cc", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + ACCURACY_EVAL_HDRS, |
| deps = ACCURACY_EVAL_DEPS, |
| ) |
| |
| xnnpack_benchmark( |
| name = "f32_sigmoid_eval", |
| srcs = [ |
| "eval/f32-sigmoid.cc", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + ACCURACY_EVAL_HDRS, |
| deps = ACCURACY_EVAL_DEPS, |
| ) |
| |
| ######################### Unit tests for micro-kernels ######################### |
| |
| xnnpack_unit_test( |
| name = "f16_gemm_test", |
| srcs = [ |
| "test/f16-gemm.cc", |
| "test/gemm-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_argmaxpool_test", |
| srcs = [ |
| "test/f32-argmaxpool.cc", |
| "test/argmaxpool-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_avgpool_test", |
| srcs = [ |
| "test/f32-avgpool.cc", |
| "test/avgpool-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_bilinear_test", |
| srcs = [ |
| "test/f32-bilinear.cc", |
| "test/bilinear-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_clamp_test", |
| srcs = [ |
| "test/f32-clamp.cc", |
| "test/clamp-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_igemm_test", |
| srcs = [ |
| "test/f32-igemm.cc", |
| "test/gemm-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_conv_hwc_test", |
| srcs = [ |
| "test/f32-conv-hwc.cc", |
| "test/conv-hwc-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_conv_hwc2spchw_test", |
| srcs = [ |
| "test/f32-conv-hwc2spchw.cc", |
| "test/conv-hwc2spchw-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_dwconv_test", |
| srcs = [ |
| "test/f32-dwconv.cc", |
| "test/dwconv-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_dwconv_spchw_test", |
| srcs = [ |
| "test/f32-dwconv-spchw.cc", |
| "test/dwconv-spchw-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_gavgpool_test", |
| srcs = [ |
| "test/f32-gavgpool.cc", |
| "test/gavgpool-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_gavgpool_spchw_test", |
| srcs = [ |
| "test/f32-gavgpool-spchw.cc", |
| "test/gavgpool-spchw-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_gemm_test", |
| srcs = [ |
| "test/f32-gemm.cc", |
| "test/gemm-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_gemminc_test", |
| srcs = [ |
| "test/f32-gemminc.cc", |
| "test/gemm-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_hswish_test", |
| srcs = [ |
| "test/f32-hswish.cc", |
| "test/hswish-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_maxpool_test", |
| srcs = [ |
| "test/f32-maxpool.cc", |
| "test/maxpool-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_pavgpool_test", |
| srcs = [ |
| "test/f32-pavgpool.cc", |
| "test/avgpool-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_ppmm_test", |
| srcs = [ |
| "test/f32-ppmm.cc", |
| "test/gemm-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_prelu_test", |
| srcs = [ |
| "test/f32-prelu.cc", |
| "test/prelu-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_raddexpminusmax_test", |
| srcs = [ |
| "test/f32-raddexpminusmax.cc", |
| "test/raddexpminusmax-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_raddextexp_test", |
| srcs = [ |
| "test/f32-raddextexp.cc", |
| "test/raddextexp-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_raddstoreexpminusmax_test", |
| srcs = [ |
| "test/f32-raddstoreexpminusmax.cc", |
| "test/raddstoreexpminusmax-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_rmax_test", |
| srcs = [ |
| "test/f32-rmax.cc", |
| "test/rmax-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_sigmoid_test", |
| srcs = [ |
| "test/f32-sigmoid.cc", |
| "test/vunary-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_spmm_test", |
| srcs = [ |
| "test/f32-spmm.cc", |
| "test/spmm-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vadd_test", |
| srcs = [ |
| "test/f32-vadd.cc", |
| "test/vbinary-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vaddc_test", |
| srcs = [ |
| "test/f32-vaddc.cc", |
| "test/vbinaryc-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vdiv_test", |
| srcs = [ |
| "test/f32-vdiv.cc", |
| "test/vbinary-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vdivc_test", |
| srcs = [ |
| "test/f32-vdivc.cc", |
| "test/vbinaryc-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vrdivc_test", |
| srcs = [ |
| "test/f32-vrdivc.cc", |
| "test/vbinaryc-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vmax_test", |
| srcs = [ |
| "test/f32-vmax.cc", |
| "test/vbinary-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vmaxc_test", |
| srcs = [ |
| "test/f32-vmaxc.cc", |
| "test/vbinaryc-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vmin_test", |
| srcs = [ |
| "test/f32-vmin.cc", |
| "test/vbinary-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vminc_test", |
| srcs = [ |
| "test/f32-vminc.cc", |
| "test/vbinaryc-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vmul_test", |
| srcs = [ |
| "test/f32-vmul.cc", |
| "test/vbinary-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vmulc_test", |
| srcs = [ |
| "test/f32-vmulc.cc", |
| "test/vbinaryc-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vmulcaddc_test", |
| srcs = [ |
| "test/f32-vmulcaddc.cc", |
| "test/vmulcaddc-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vscale_test", |
| srcs = [ |
| "test/f32-vscale.cc", |
| "test/vscale-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vscaleexpminusmax_test", |
| srcs = [ |
| "test/f32-vscaleexpminusmax.cc", |
| "test/vscaleexpminusmax-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vscaleextexp_test", |
| srcs = [ |
| "test/f32-vscaleextexp.cc", |
| "test/vscaleextexp-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vsub_test", |
| srcs = [ |
| "test/f32-vsub.cc", |
| "test/vbinary-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vsubc_test", |
| srcs = [ |
| "test/f32-vsubc.cc", |
| "test/vbinaryc-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "f32_vrsubc_test", |
| srcs = [ |
| "test/f32-vrsubc.cc", |
| "test/vbinaryc-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "q8_avgpool_test", |
| srcs = [ |
| "test/q8-avgpool.cc", |
| "test/avgpool-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "q8_igemm_test", |
| srcs = [ |
| "test/q8-igemm.cc", |
| "test/gemm-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "q8_dwconv_test", |
| srcs = [ |
| "test/q8-dwconv.cc", |
| "test/dwconv-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "q8_gavgpool_test", |
| srcs = [ |
| "test/q8-gavgpool.cc", |
| "test/gavgpool-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "q8_gemm_test", |
| srcs = [ |
| "test/q8-gemm.cc", |
| "test/gemm-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + WEIGHTS_PACK_HDRS + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "q8_vadd_test", |
| srcs = [ |
| "test/q8-vadd.cc", |
| "test/vadd-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "u8_clamp_test", |
| srcs = [ |
| "test/u8-clamp.cc", |
| "test/clamp-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "u8_lut32norm_test", |
| srcs = [ |
| "test/u8-lut32norm.cc", |
| "test/lut-norm-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "u8_maxpool_test", |
| srcs = [ |
| "test/u8-maxpool.cc", |
| "test/maxpool-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "u8_rmax_test", |
| srcs = [ |
| "test/u8-rmax.cc", |
| "test/rmax-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "x32_packx_test", |
| srcs = [ |
| "test/x32-packx.cc", |
| "test/pack-microkernel-tester.h", |
| "src/xnnpack/AlignedAllocator.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "x32_pad_test", |
| srcs = [ |
| "test/x32-pad.cc", |
| "test/pad-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "x32_unpool_test", |
| srcs = [ |
| "test/x32-unpool.cc", |
| "test/unpool-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "x32_zip_test", |
| srcs = [ |
| "test/x32-zip.cc", |
| "test/zip-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "x8_lut_test", |
| srcs = [ |
| "test/x8-lut.cc", |
| "test/lut-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "x8_zip_test", |
| srcs = [ |
| "test/x8-zip.cc", |
| "test/zip-microkernel-tester.h", |
| ] + MICROKERNEL_TEST_HDRS, |
| deps = MICROKERNEL_TEST_DEPS, |
| ) |
| |
| ########################### Size test for the library ########################## |
| |
| xnnpack_binary( |
| name = "size_test", |
| srcs = ["test/size.c"], |
| deps = [":xnnpack_operators_nhwc_f32"], |
| ) |
| |
| ########################### Unit tests for operators ########################### |
| |
| xnnpack_unit_test( |
| name = "add_nc_test", |
| srcs = [ |
| "test/add-nc.cc", |
| "test/add-operator-tester.h", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "add_nd_test", |
| srcs = [ |
| "test/add-nd.cc", |
| "test/binary-elementwise-operator-tester.h", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "argmax_pooling_nhwc_test", |
| srcs = [ |
| "test/argmax-pooling-nhwc.cc", |
| "test/argmax-pooling-operator-tester.h", |
| ] + OPERATOR_TEST_PARAMS_HDRS, |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "average_pooling_nhwc_test", |
| srcs = [ |
| "test/average-pooling-nhwc.cc", |
| "test/average-pooling-operator-tester.h", |
| ] + OPERATOR_TEST_PARAMS_HDRS, |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "channel_pad_nc_test", |
| srcs = [ |
| "test/channel-pad-nc.cc", |
| "test/channel-pad-operator-tester.h", |
| ] + OPERATOR_TEST_PARAMS_HDRS, |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "channel_shuffle_nc_test", |
| srcs = [ |
| "test/channel-shuffle-nc.cc", |
| "test/channel-shuffle-operator-tester.h", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "clamp_nc_test", |
| srcs = [ |
| "test/clamp-nc.cc", |
| "test/clamp-operator-tester.h", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "convolution_nhwc_test", |
| srcs = [ |
| "test/convolution-nhwc.cc", |
| "test/convolution-operator-tester.h", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "convolution_nchw_test", |
| srcs = [ |
| "test/convolution-nchw.cc", |
| "test/convolution-operator-tester.h", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "deconvolution_nhwc_test", |
| srcs = [ |
| "test/deconvolution-nhwc.cc", |
| "test/deconvolution-operator-tester.h", |
| ] + OPERATOR_TEST_PARAMS_HDRS, |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "divide_nd_test", |
| srcs = [ |
| "test/binary-elementwise-operator-tester.h", |
| "test/divide-nd.cc", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "fully_connected_nc_test", |
| srcs = [ |
| "test/fully-connected-nc.cc", |
| "test/fully-connected-operator-tester.h", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "global_average_pooling_nwc_test", |
| srcs = [ |
| "test/global-average-pooling-nwc.cc", |
| "test/global-average-pooling-operator-tester.h", |
| ] + OPERATOR_TEST_PARAMS_HDRS, |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "global_average_pooling_ncw_test", |
| srcs = [ |
| "test/global-average-pooling-ncw.cc", |
| "test/global-average-pooling-operator-tester.h", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "hardswish_nc_test", |
| srcs = [ |
| "test/hardswish-nc.cc", |
| "test/hardswish-operator-tester.h", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "leaky_relu_nc_test", |
| srcs = [ |
| "test/leaky-relu-nc.cc", |
| "test/leaky-relu-operator-tester.h", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "max_pooling_nhwc_test", |
| srcs = [ |
| "test/max-pooling-nhwc.cc", |
| "test/max-pooling-operator-tester.h", |
| ] + OPERATOR_TEST_PARAMS_HDRS, |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "maximum_nd_test", |
| srcs = [ |
| "test/binary-elementwise-operator-tester.h", |
| "test/maximum-nd.cc", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "minimum_nd_test", |
| srcs = [ |
| "test/binary-elementwise-operator-tester.h", |
| "test/minimum-nd.cc", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "multiply_nd_test", |
| srcs = [ |
| "test/binary-elementwise-operator-tester.h", |
| "test/multiply-nd.cc", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "prelu_nc_test", |
| srcs = [ |
| "test/prelu-nc.cc", |
| "test/prelu-operator-tester.h", |
| ] + OPERATOR_TEST_PARAMS_HDRS, |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "resize_bilinear_nhwc_test", |
| srcs = [ |
| "test/resize-bilinear-nhwc.cc", |
| "test/resize-bilinear-operator-tester.h", |
| ] + OPERATOR_TEST_PARAMS_HDRS, |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "sigmoid_nc_test", |
| srcs = [ |
| "test/sigmoid-nc.cc", |
| "test/sigmoid-operator-tester.h", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "softargmax_nc_test", |
| srcs = [ |
| "test/softargmax-nc.cc", |
| "test/softargmax-operator-tester.h", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "subtract_nd_test", |
| srcs = [ |
| "test/binary-elementwise-operator-tester.h", |
| "test/subtract-nd.cc", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| xnnpack_unit_test( |
| name = "unpooling_nhwc_test", |
| srcs = [ |
| "test/unpooling-nhwc.cc", |
| "test/unpooling-operator-tester.h", |
| ], |
| deps = OPERATOR_TEST_DEPS, |
| ) |
| |
| ############################# Build configurations ############################# |
| |
| # Enables usage of assembly kernels. |
| config_setting( |
| name = "xnn_enable_assembly_explicit_true", |
| define_values = {"xnn_enable_assembly": "true"}, |
| ) |
| |
| # Disables usage of assembly kernels. |
| config_setting( |
| name = "xnn_enable_assembly_explicit_false", |
| define_values = {"xnn_enable_assembly": "false"}, |
| ) |
| |
| # Builds with -c dbg |
| config_setting( |
| name = "debug_build", |
| values = { |
| "compilation_mode": "dbg", |
| }, |
| ) |
| |
| # Builds with -c opt |
| config_setting( |
| name = "optimized_build", |
| values = { |
| "compilation_mode": "opt", |
| }, |
| ) |
| |
| config_setting( |
| name = "linux_k8", |
| values = {"cpu": "k8"}, |
| ) |
| |
| config_setting( |
| name = "linux_aarch64", |
| values = {"cpu": "aarch64"}, |
| ) |
| |
| config_setting( |
| name = "android", |
| values = {"crosstool_top": "//external:android/crosstool"}, |
| ) |
| |
| config_setting( |
| name = "android_armv7", |
| values = { |
| "crosstool_top": "//external:android/crosstool", |
| "cpu": "armeabi-v7a", |
| }, |
| ) |
| |
| config_setting( |
| name = "android_arm64", |
| values = { |
| "crosstool_top": "//external:android/crosstool", |
| "cpu": "arm64-v8a", |
| }, |
| ) |
| |
| config_setting( |
| name = "android_x86", |
| values = { |
| "crosstool_top": "//external:android/crosstool", |
| "cpu": "x86", |
| }, |
| ) |
| |
| config_setting( |
| name = "android_x86_64", |
| values = { |
| "crosstool_top": "//external:android/crosstool", |
| "cpu": "x86_64", |
| }, |
| ) |
| |
| config_setting( |
| name = "macos_x86_64", |
| values = { |
| "apple_platform_type": "macos", |
| "cpu": "darwin", |
| }, |
| ) |
| |
| config_setting( |
| name = "emscripten", |
| values = {"crosstool_top": "//toolchain:emscripten"}, |
| ) |
| |
| config_setting( |
| name = "emscripten_wasm", |
| values = { |
| "crosstool_top": "//toolchain:emscripten", |
| "cpu": "wasm", |
| }, |
| ) |
| |
| config_setting( |
| name = "emscripten_wasmsimd", |
| values = { |
| "crosstool_top": "//toolchain:emscripten", |
| "cpu": "wasm", |
| "features": "wasm_simd", |
| }, |
| ) |
| |
| config_setting( |
| name = "emscripten_asmjs", |
| values = { |
| "crosstool_top": "//toolchain:emscripten", |
| "cpu": "asmjs", |
| }, |
| ) |