# 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.c",
    "src/argmax-pooling.c",
    "src/average-pooling.c",
    "src/channel-pad.c",
    "src/channel-shuffle.c",
    "src/clamp.c",
    "src/convolution-spnchw.c",
    "src/convolution.c",
    "src/deconvolution.c",
    "src/fully-connected.c",
    "src/global-average-pooling-spnchw.c",
    "src/global-average-pooling.c",
    "src/hardswish.c",
    "src/leaky-relu.c",
    "src/max-pooling.c",
    "src/prelu.c",
    "src/sigmoid.c",
    "src/softargmax.c",
    "src/unpooling.c",
]

SCALAR_UKERNELS = [
    "src/f32-argmaxpool/mp9p8q-scalar.c",
    "src/f32-argmaxpool/up4-scalar.c",
    "src/f32-argmaxpool/up9-scalar.c",
    "src/f32-avgpool/mp9p8q-scalar.c",
    "src/f32-avgpool/up9-scalar.c",
    "src/f32-bilinear/scalar-c1.c",
    "src/f32-bilinear/scalar-c2.c",
    "src/f32-bilinear/scalar-c4.c",
    "src/f32-clamp/scalar.c",
    "src/f32-conv-hwc2spchw/3x3s2p1c3x4-scalar-1x1.c",
    "src/f32-igemm/1x4-scalar.c",
    "src/f32-igemm/2x4-scalar.c",
    "src/f32-igemm/4x2-scalar.c",
    "src/f32-igemm/4x4-scalar.c",
    "src/f32-dwconv/up1x25-scalar.c",
    "src/f32-dwconv/up1x25-scalar-acc2.c",
    "src/f32-dwconv/up2x25-scalar.c",
    "src/f32-dwconv/up2x25-scalar-acc2.c",
    "src/f32-dwconv/up1x4-scalar.c",
    "src/f32-dwconv/up1x4-scalar-acc2.c",
    "src/f32-dwconv/up2x4-scalar.c",
    "src/f32-dwconv/up2x4-scalar-acc2.c",
    "src/f32-dwconv/up1x9-scalar.c",
    "src/f32-dwconv/up1x9-scalar-acc2.c",
    "src/f32-dwconv/up2x9-scalar.c",
    "src/f32-dwconv/up2x9-scalar-acc2.c",
    "src/f32-dwconv-spchw/3x3p1-scalar.c",
    "src/f32-dwconv-spchw/5x5p2-scalar.c",
    "src/f32-dwconv-spchw/3x3s2p1-scalar.c",
    "src/f32-dwconv-spchw/5x5s2p2-scalar.c",
    "src/f32-gavgpool-spchw/scalar-x1.c",
    "src/f32-gavgpool/mp7p7q-scalar.c",
    "src/f32-gavgpool/up7-scalar.c",
    "src/f32-gemm/1x4-scalar.c",
    "src/f32-gemm/2x4-scalar.c",
    "src/f32-gemm/4x2-scalar.c",
    "src/f32-gemm/4x4-scalar.c",
    "src/f32-gemminc/1x4-scalar.c",
    "src/f32-gemminc/2x4-scalar.c",
    "src/f32-gemminc/4x4-scalar.c",
    "src/f32-hswish/scalar.c",
    "src/f32-maxpool/9p8q-scalar.c",
    "src/f32-pavgpool/mp9p8q-scalar.c",
    "src/f32-pavgpool/up9-scalar.c",
    "src/f32-ppmm/2x4-scalar.c",
    "src/f32-ppmm/3x3-scalar.c",
    "src/f32-ppmm/4x2-scalar.c",
    "src/f32-ppmm/4x4-scalar.c",
    "src/f32-prelu/scalar-2x1.c",
    "src/f32-prelu/scalar-2x4.c",
    "src/f32-rmax/scalar.c",
    "src/f32-spmm/1x1-scalar-pipelined.c",
    "src/f32-spmm/1x1-scalar.c",
    "src/f32-spmm/2x1-scalar-pipelined.c",
    "src/f32-spmm/2x1-scalar.c",
    "src/f32-spmm/4x1-scalar-pipelined.c",
    "src/f32-spmm/4x1-scalar.c",
    "src/f32-spmm/8x1-scalar-pipelined.c",
    "src/f32-spmm/8x1-scalar.c",
    "src/f32-spmm/8x2-scalar.c",
    "src/f32-spmm/8x4-scalar.c",
    "src/f32-vadd/scalar.c",
    "src/f32-vmul/scalar.c",
    "src/f32-vmulcaddc/c1-scalar-2x.c",
    "src/f32-vmulcaddc/c2-scalar-2x.c",
    "src/f32-vmulcaddc/c4-scalar-2x.c",
    "src/f32-vsub/scalar.c",
    "src/q8-avgpool/mp9p8q-scalar.c",
    "src/q8-avgpool/up9-scalar.c",
    "src/q8-igemm/2x2-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-vadd/scalar.c",
    "src/u8-clamp/scalar.c",
    "src/u8-lut32norm/scalar.c",
    "src/u8-maxpool/9p8q-scalar.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",
]

PSIMD_UKERNELS = [
    "src/f32-argmaxpool/mp9p8q-psimd.c",
    "src/f32-argmaxpool/up4-psimd.c",
    "src/f32-argmaxpool/up9-psimd.c",
    "src/f32-avgpool/mp9p8q-psimd.c",
    "src/f32-avgpool/up9-psimd.c",
    "src/f32-bilinear/psimd-c4.c",
    "src/f32-bilinear/psimd-c8.c",
    "src/f32-clamp/psimd.c",
    "src/f32-igemm/1x8-psimd-loadsplat.c",
    "src/f32-igemm/1x8-psimd-splat.c",
    "src/f32-igemm/1x8s4-psimd.c",
    "src/f32-igemm/4x2c4-psimd.c",
    "src/f32-igemm/4x8-psimd-loadsplat.c",
    "src/f32-igemm/4x8-psimd-splat.c",
    "src/f32-igemm/4x8s4-psimd.c",
    "src/f32-igemm/6x8-psimd-loadsplat.c",
    "src/f32-igemm/6x8-psimd-splat.c",
    "src/f32-igemm/6x8s4-psimd.c",
    "src/f32-dwconv/up4x25-psimd.c",
    "src/f32-dwconv/up4x25-psimd-acc2.c",
    "src/f32-dwconv/up8x25-psimd.c",
    "src/f32-dwconv/up8x25-psimd-acc2.c",
    "src/f32-dwconv/up4x4-psimd.c",
    "src/f32-dwconv/up4x4-psimd-acc2.c",
    "src/f32-dwconv/up8x4-psimd.c",
    "src/f32-dwconv/up8x4-psimd-acc2.c",
    "src/f32-dwconv/up4x9-psimd.c",
    "src/f32-dwconv/up4x9-psimd-acc2.c",
    "src/f32-dwconv/up8x9-psimd.c",
    "src/f32-dwconv/up8x9-psimd-acc2.c",
    "src/f32-gavgpool/mp7p7q-psimd.c",
    "src/f32-gavgpool/up7-psimd.c",
    "src/f32-gemm/1x8-psimd-loadsplat.c",
    "src/f32-gemm/1x8-psimd-splat.c",
    "src/f32-gemm/1x8s4-psimd.c",
    "src/f32-gemm/4x8-psimd-loadsplat.c",
    "src/f32-gemm/4x8-psimd-splat.c",
    "src/f32-gemm/4x8s4-psimd.c",
    "src/f32-gemm/6x8-psimd-loadsplat.c",
    "src/f32-gemm/6x8-psimd-splat.c",
    "src/f32-gemm/6x8s4-psimd.c",
    "src/f32-gemminc/1x8-psimd-loadsplat.c",
    "src/f32-gemminc/1x8-psimd-splat.c",
    "src/f32-gemminc/1x8s4-psimd.c",
    "src/f32-gemminc/4x8-psimd-loadsplat.c",
    "src/f32-gemminc/4x8-psimd-splat.c",
    "src/f32-gemminc/4x8s4-psimd.c",
    "src/f32-gemminc/6x8-psimd-loadsplat.c",
    "src/f32-gemminc/6x8-psimd-splat.c",
    "src/f32-gemminc/6x8s4-psimd.c",
    "src/f32-hswish/psimd.c",
    "src/f32-maxpool/9p8q-psimd.c",
    "src/f32-pavgpool/mp9p8q-psimd.c",
    "src/f32-pavgpool/up9-psimd.c",
    "src/f32-ppmm/4x8-psimd.c",
    "src/f32-prelu/psimd-2x4.c",
    "src/f32-prelu/psimd-2x8.c",
    "src/f32-vadd/psimd.c",
    "src/f32-vmul/psimd.c",
    "src/f32-vmulcaddc/c4-psimd-2x.c",
    "src/f32-vmulcaddc/c8-psimd-2x.c",
    "src/f32-vsub/psimd.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/neon-c4.c",
    "src/f32-bilinear/neon-c8.c",
    "src/f32-clamp/neon.c",
    "src/f32-dwconv/up4x9-neon.c",
    "src/f32-dwconv/up4x9-neon-acc2.c",
    "src/f32-dwconv/up8x9-neon.c",
    "src/f32-dwconv/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/1x8-neon-ld64.c",
    "src/f32-gemm/4x2-neon-ld64.c",
    "src/f32-gemm/4x8-neon-ld128.c",
    "src/f32-gemm/4x8-neon-ld64.c",
    "src/f32-gemm/5x8-neon-ld64.c",
    "src/f32-gemm/6x8-neon-ld64.c",
    "src/f32-gemminc/1x8-neon-ld64.c",
    "src/f32-gemminc/4x8-neon-ld128.c",
    "src/f32-gemminc/4x8-neon-ld64.c",
    "src/f32-gemminc/5x8-neon-ld64.c",
    "src/f32-gemminc/6x8-neon-ld64.c",
    "src/f32-hswish/neon.c",
    "src/f32-igemm/1x8-neon-ld64.c",
    "src/f32-igemm/4x2-neon-ld64.c",
    "src/f32-igemm/4x4-neon-ld64.c",
    "src/f32-igemm/4x8-neon-ld128.c",
    "src/f32-igemm/4x8-neon-ld64.c",
    "src/f32-igemm/6x8-neon-ld64.c",
    "src/f32-pavgpool/mp9p8q-neon.c",
    "src/f32-pavgpool/up9-neon.c",
    "src/f32-ppmm/4x8-neon.c",
    "src/f32-ppmm/8x8-neon.c",
    "src/f32-prelu/neon-2x4.c",
    "src/f32-prelu/neon-2x8.c",
    "src/f32-rmax/neon.c",
    "src/f32-vmulcaddc/c4-neon-2x.c",
    "src/f32-vmulcaddc/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/9p8q-neon.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/neonfma-c4.c",
    "src/f32-bilinear/neonfma-c8.c",
    "src/f32-igemm/1x8-neonfma-ld64.c",
    "src/f32-igemm/4x2-neonfma-ld64.c",
    "src/f32-igemm/4x4-neonfma-ld64.c",
    "src/f32-igemm/4x8-neonfma-ld128.c",
    "src/f32-igemm/4x8-neonfma-ld64.c",
    "src/f32-igemm/6x8-neonfma-ld64.c",
    "src/f32-dwconv/up4x9-neonfma.c",
    "src/f32-dwconv/up4x9-neonfma-acc2.c",
    "src/f32-dwconv/up8x9-neonfma.c",
    "src/f32-dwconv/up8x9-neonfma-acc2.c",
    "src/f32-gemm/1x8-neonfma-ld64.c",
    "src/f32-gemm/4x2-neonfma-ld64.c",
    "src/f32-gemm/4x8-neonfma-ld128.c",
    "src/f32-gemm/4x8-neonfma-ld64.c",
    "src/f32-gemm/5x8-neonfma-ld64.c",
    "src/f32-gemm/6x8-neonfma-ld64.c",
    "src/f32-gemminc/1x8-neonfma-ld64.c",
    "src/f32-gemminc/4x8-neonfma-ld128.c",
    "src/f32-gemminc/4x8-neonfma-ld64.c",
    "src/f32-gemminc/5x8-neonfma-ld64.c",
    "src/f32-gemminc/6x8-neonfma-ld64.c",
    "src/f32-hswish/neonfma.c",
    "src/f32-ppmm/4x8-neonfma.c",
    "src/f32-ppmm/8x8-neonfma.c",
    "src/f32-vmulcaddc/c4-neonfma-2x.c",
    "src/f32-vmulcaddc/c8-neonfma-2x.c",
]

AARCH64_NEONFMA_UKERNELS = [
    "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/12x1-neonfma.c",
    "src/f32-spmm/12x2-neonfma.c",
    "src/f32-spmm/12x4-neonfma.c",
    "src/f32-spmm/16x1-neonfma-pipelined.c",
    "src/f32-spmm/16x1-neonfma-unroll2.c",
    "src/f32-spmm/16x1-neonfma.c",
    "src/f32-spmm/16x2-neonfma.c",
    "src/f32-spmm/16x4-neonfma.c",
    "src/f32-spmm/4x1-neonfma-pipelined.c",
    "src/f32-spmm/4x1-neonfma-unroll2.c",
    "src/f32-spmm/4x1-neonfma.c",
    "src/f32-spmm/4x2-neonfma.c",
    "src/f32-spmm/4x4-neonfma.c",
    "src/f32-spmm/8x1-neonfma-pipelined.c",
    "src/f32-spmm/8x1-neonfma-unroll2.c",
    "src/f32-spmm/8x1-neonfma.c",
    "src/f32-spmm/8x2-neonfma.c",
    "src/f32-spmm/8x4-neonfma.c",
]

AARCH64_NEONFP16ARITH_UKERNELS = [
    "src/f16-gemm/4x8-neonfp16arith-ld64.c",
    "src/f16-gemm/6x8-neonfp16arith-ld64.c",
    "src/f16-gemm/8x8-neonfp16arith-ld64.c",
]

SSE_UKERNELS = [
    "src/f32-avgpool/mp9p8q-sse.c",
    "src/f32-avgpool/up9-sse.c",
    "src/f32-bilinear/sse-c4.c",
    "src/f32-bilinear/sse-c8.c",
    "src/f32-clamp/sse.c",
    "src/f32-igemm/1x8-sse-dup.c",
    "src/f32-igemm/1x8-sse-load1.c",
    "src/f32-igemm/1x8s4-sse.c",
    "src/f32-igemm/4x2c4-sse.c",
    "src/f32-igemm/4x8-sse-dup.c",
    "src/f32-igemm/4x8-sse-load1.c",
    "src/f32-igemm/4x8s4-sse.c",
    "src/f32-dwconv/up4x25-sse.c",
    "src/f32-dwconv/up4x4-sse.c",
    "src/f32-dwconv/up4x9-sse.c",
    "src/f32-dwconv/up4x25-sse-acc2.c",
    "src/f32-dwconv/up4x4-sse-acc2.c",
    "src/f32-dwconv/up4x9-sse-acc2.c",
    "src/f32-dwconv/up8x25-sse.c",
    "src/f32-dwconv/up8x4-sse.c",
    "src/f32-dwconv/up8x9-sse.c",
    "src/f32-dwconv/up8x25-sse-acc2.c",
    "src/f32-dwconv/up8x4-sse-acc2.c",
    "src/f32-dwconv/up8x9-sse-acc2.c",
    "src/f32-gavgpool-spchw/sse-x4.c",
    "src/f32-gavgpool/mp7p7q-sse.c",
    "src/f32-gavgpool/up7-sse.c",
    "src/f32-gemm/1x8-sse-dup.c",
    "src/f32-gemm/1x8-sse-load1.c",
    "src/f32-gemm/1x8s4-sse.c",
    "src/f32-gemm/4x8-sse-dup.c",
    "src/f32-gemm/4x8-sse-load1.c",
    "src/f32-gemm/4x8s4-sse.c",
    "src/f32-gemminc/1x8-sse-dup.c",
    "src/f32-gemminc/1x8-sse-load1.c",
    "src/f32-gemminc/1x8s4-sse.c",
    "src/f32-gemminc/4x8-sse-dup.c",
    "src/f32-gemminc/4x8-sse-load1.c",
    "src/f32-gemminc/4x8s4-sse.c",
    "src/f32-hswish/sse.c",
    "src/f32-maxpool/9p8q-sse.c",
    "src/f32-pavgpool/mp9p8q-sse.c",
    "src/f32-pavgpool/up9-sse.c",
    "src/f32-dwconv-spchw/3x3p1-sse.c",
    "src/f32-dwconv-spchw/3x3s2p1-sse.c",
    "src/f32-ppmm/4x8-sse.c",
    "src/f32-rmax/sse.c",
    "src/f32-spmm/4x1-sse.c",
    "src/f32-spmm/8x1-sse.c",
    "src/f32-vadd/sse.c",
    "src/f32-vmul/sse.c",
    "src/f32-vmulcaddc/c4-sse-2x.c",
    "src/f32-vmulcaddc/c8-sse-2x.c",
    "src/f32-vsub/sse.c",
    "src/x32-packx/x4-sse.c",
]

SSE2_UKERNELS = [
    "src/f32-argmaxpool/mp9p8q-sse2.c",
    "src/f32-argmaxpool/up4-sse2.c",
    "src/f32-argmaxpool/up9-sse2.c",
    "src/f32-prelu/sse2-2x4.c",
    "src/f32-prelu/sse2-2x8.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/9p8q-sse2.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",
]

SSE41_UKERNELS = [
    "src/f32-prelu/sse41-2x4.c",
    "src/f32-prelu/sse41-2x8.c",
]

AVX_UKERNELS = [
    "src/f32-rmax/avx.c",
    "src/f32-vscale/avx-unroll32.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-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",
]

AARCH64_ASM_UKERNELS = [
    "src/f32-dwconv/up4x9-aarch64-neonfma-cortex-a55.S",
    "src/f32-dwconv/up4x9-aarch64-neonfma.S",
    "src/f32-gemm/1x12-aarch64-neonfma-cortex-a53.S",
    "src/f32-gemm/1x8-aarch64-neonfma-cortex-a53.S",
    "src/f32-gemm/1x8-aarch64-neonfma-cortex-a57.S",
    "src/f32-gemm/1x8-aarch64-neonfma-cortex-a75.S",
    "src/f32-gemm/4x12-aarch64-neonfma-cortex-a53.S",
    "src/f32-gemm/4x8-aarch64-neonfma-cortex-a53.S",
    "src/f32-gemm/4x8-aarch64-neonfma-cortex-a57.S",
    "src/f32-gemm/4x8-aarch64-neonfma-cortex-a75.S",
    "src/f32-gemm/4x8-aarch64-neonfma-ld128.S",
    "src/f32-gemm/4x8-aarch64-neonfma-ld64.S",
    "src/f32-gemm/5x8-aarch64-neonfma-cortex-a75.S",
    "src/f32-gemm/6x8-aarch64-neonfma-cortex-a53.S",
    "src/f32-gemm/6x8-aarch64-neonfma-cortex-a57.S",
    "src/f32-gemm/6x8-aarch64-neonfma-cortex-a73.S",
    "src/f32-gemm/6x8-aarch64-neonfma-cortex-a75.S",
    "src/f32-gemm/6x8-aarch64-neonfma-ld128.S",
    "src/f32-gemm/6x8-aarch64-neonfma-ld64.S",
    "src/f32-gemminc/1x12-aarch64-neonfma-cortex-a53.S",
    "src/f32-gemminc/1x8-aarch64-neonfma-cortex-a53.S",
    "src/f32-gemminc/1x8-aarch64-neonfma-cortex-a57.S",
    "src/f32-gemminc/1x8-aarch64-neonfma-cortex-a75.S",
    "src/f32-gemminc/4x12-aarch64-neonfma-cortex-a53.S",
    "src/f32-gemminc/4x8-aarch64-neonfma-cortex-a53.S",
    "src/f32-gemminc/4x8-aarch64-neonfma-cortex-a57.S",
    "src/f32-gemminc/4x8-aarch64-neonfma-cortex-a75.S",
    "src/f32-gemminc/4x8-aarch64-neonfma-ld128.S",
    "src/f32-gemminc/4x8-aarch64-neonfma-ld64.S",
    "src/f32-gemminc/5x8-aarch64-neonfma-cortex-a75.S",
    "src/f32-gemminc/6x8-aarch64-neonfma-cortex-a53.S",
    "src/f32-gemminc/6x8-aarch64-neonfma-cortex-a57.S",
    "src/f32-gemminc/6x8-aarch64-neonfma-cortex-a73.S",
    "src/f32-gemminc/6x8-aarch64-neonfma-cortex-a75.S",
    "src/f32-gemminc/6x8-aarch64-neonfma-ld128.S",
    "src/f32-gemminc/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/lut.h",
    "src/xnnpack/math.h",
    "src/xnnpack/maxpool.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/vmul.h",
    "src/xnnpack/vmulcaddc.h",
    "src/xnnpack/vscale.h",
    "src/xnnpack/vscaleexpminusmax.h",
    "src/xnnpack/vscaleextexp.h",
    "src/xnnpack/vsub.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",
    ],
)

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",
    ],
)

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"],
)

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"],
)

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"],
)

xnnpack_cc_library(
    name = "sse2_ukernels",
    hdrs = INTERNAL_HDRS,
    copts = xnnpack_std_copts(),
    x86_copts = ["-msse2"],
    x86_srcs = SSE_UKERNELS + SSE2_UKERNELS,
    deps = ["@FP16"],
)

xnnpack_cc_library(
    name = "sse41_ukernels",
    hdrs = INTERNAL_HDRS,
    copts = xnnpack_std_copts(),
    x86_copts = ["-msse4.1"],
    x86_srcs = SSE41_UKERNELS,
    deps = ["@FP16"],
)

xnnpack_cc_library(
    name = "avx_ukernels",
    hdrs = INTERNAL_HDRS,
    copts = xnnpack_std_copts(),
    x86_copts = ["-mavx"],
    x86_srcs = AVX_UKERNELS,
    deps = ["@FP16"],
)

xnnpack_cc_library(
    name = "avx2_ukernels",
    hdrs = INTERNAL_HDRS,
    copts = xnnpack_std_copts(),
    x86_copts = [
        "-mfma",
        "-mavx2",
    ],
    x86_srcs = AVX2_UKERNELS,
    deps = ["@FP16"],
)

xnnpack_cc_library(
    name = "avx512f_ukernels",
    hdrs = INTERNAL_HDRS,
    copts = xnnpack_std_copts(),
    x86_copts = ["-mavx512f"],
    x86_srcs = AVX512F_UKERNELS,
    deps = ["@FP16"],
)

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"],
    wasmsimd_deps = [
        ":psimd_ukernels",
    ],
    x86_deps = [
        ":psimd_ukernels",
        ":sse2_ukernels",
        ":sse41_ukernels",
        ":avx_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/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_SPNCHW_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_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"],
    deps = OPERATOR_BENCHMARK_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,
)

######################### 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_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/vadd-microkernel-tester.h",
    ] + MICROKERNEL_TEST_HDRS,
    deps = MICROKERNEL_TEST_DEPS,
)

xnnpack_unit_test(
    name = "f32_vmul_test",
    srcs = [
        "test/f32-vmul.cc",
        "test/vmul-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/vsub-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_test",
    srcs = [
        "test/add.cc",
        "test/add-operator-tester.h",
    ],
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "argmax_pooling_test",
    srcs = [
        "test/argmax-pooling.cc",
        "test/argmax-pooling-operator-tester.h",
    ] + OPERATOR_TEST_PARAMS_HDRS,
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "average_pooling_test",
    srcs = [
        "test/average-pooling.cc",
        "test/average-pooling-operator-tester.h",
    ] + OPERATOR_TEST_PARAMS_HDRS,
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "channel_pad_test",
    srcs = [
        "test/channel-pad.cc",
        "test/channel-pad-operator-tester.h",
    ] + OPERATOR_TEST_PARAMS_HDRS,
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "channel_shuffle_test",
    srcs = [
        "test/channel-shuffle.cc",
        "test/channel-shuffle-operator-tester.h",
    ],
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "clamp_test",
    srcs = [
        "test/clamp.cc",
        "test/clamp-operator-tester.h",
    ],
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "convolution_test",
    srcs = [
        "test/convolution.cc",
        "test/convolution-operator-tester.h",
    ],
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "convolution_spnchw_test",
    srcs = [
        "test/convolution-spnchw.cc",
        "test/convolution-spnchw-operator-tester.h",
    ],
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "deconvolution_test",
    srcs = [
        "test/deconvolution.cc",
        "test/deconvolution-operator-tester.h",
    ] + OPERATOR_TEST_PARAMS_HDRS,
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "fully_connected_test",
    srcs = [
        "test/fully-connected.cc",
        "test/fully-connected-operator-tester.h",
    ],
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "global_average_pooling_test",
    srcs = [
        "test/global-average-pooling.cc",
        "test/global-average-pooling-operator-tester.h",
    ] + OPERATOR_TEST_PARAMS_HDRS,
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "global_average_pooling_spnchw_test",
    srcs = [
        "test/global-average-pooling-spnchw.cc",
        "test/global-average-pooling-spnchw-operator-tester.h",
    ],
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "hardswish_test",
    srcs = [
        "test/hardswish.cc",
        "test/hardswish-operator-tester.h",
    ],
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "leaky_relu_test",
    srcs = [
        "test/leaky-relu.cc",
        "test/leaky-relu-operator-tester.h",
    ],
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "max_pooling_test",
    srcs = [
        "test/max-pooling.cc",
        "test/max-pooling-operator-tester.h",
    ] + OPERATOR_TEST_PARAMS_HDRS,
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "prelu_test",
    srcs = [
        "test/prelu.cc",
        "test/prelu-operator-tester.h",
    ] + OPERATOR_TEST_PARAMS_HDRS,
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "sigmoid_test",
    srcs = [
        "test/sigmoid.cc",
        "test/sigmoid-operator-tester.h",
    ],
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "softargmax_test",
    srcs = [
        "test/softargmax.cc",
        "test/softargmax-operator-tester.h",
    ],
    deps = OPERATOR_TEST_DEPS,
)

xnnpack_unit_test(
    name = "unpooling_test",
    srcs = [
        "test/unpooling.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",
    },
)