Outline variables in BUILD files
PiperOrigin-RevId: 302496577
diff --git a/BUILD.bazel b/BUILD.bazel
index b56029f..fc0ea18 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -12,6 +12,13 @@
exports_files(["LICENSE"])
+OPERATOR_BENCHMARK_DEPS = [
+ ":XNNPACK",
+ ":bench_utils",
+ "@cpuinfo",
+ "@pthreadpool",
+]
+
MICROKERNEL_BENCHMARK_DEPS = [
":ukernels",
":bench_utils",
@@ -36,6 +43,12 @@
"@pthreadpool",
]
+OPERATOR_TEST_DEPS = [
+ ":XNNPACK",
+ "@pthreadpool",
+ "@FP16",
+]
+
OPERATOR_SRCS = [
"src/add-nc.c",
"src/argmax-pooling-nhwc.c",
@@ -1466,7 +1479,7 @@
"src/f32-igemm/gen/6x8-aarch64-neonfma-cortex-a75.S",
]
-INTERNAL_HDRS = [
+INTERNAL_MICROKERNEL_HDRS = [
"src/requantization/gemmlowp-requantization.h",
"src/xnnpack/argmaxpool.h",
"src/xnnpack/avgpool.h",
@@ -1505,6 +1518,9 @@
"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",
@@ -1519,130 +1535,16 @@
"src/xnnpack/subgraph.h",
]
-ACCURACY_EVAL_HDRS = [
- "src/requantization/gemmlowp-requantization.h",
- "src/xnnpack/argmaxpool.h",
- "src/xnnpack/avgpool.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/ibilinear.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",
+ACCURACY_EVAL_HDRS = INTERNAL_MICROKERNEL_HDRS + [
"src/xnnpack/math-stubs.h",
]
-MICROKERNEL_BENCHMARK_HDRS = [
- "src/requantization/gemmlowp-requantization.h",
- "src/xnnpack/argmaxpool.h",
- "src/xnnpack/avgpool.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/ibilinear.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",
+MICROKERNEL_BENCHMARK_HDRS = INTERNAL_MICROKERNEL_HDRS + [
"src/xnnpack/params-init.h",
"include/xnnpack.h",
]
-MICROKERNEL_TEST_HDRS = [
- "src/requantization/gemmlowp-requantization.h",
- "src/xnnpack/argmaxpool.h",
- "src/xnnpack/avgpool.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/ibilinear.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",
+MICROKERNEL_TEST_HDRS = INTERNAL_MICROKERNEL_HDRS + [
"src/xnnpack/isa-checks.h",
"src/xnnpack/params-init.h",
"src/xnnpack/requantization.h",
@@ -2295,117 +2197,67 @@
xnnpack_benchmark(
name = "add_bench",
srcs = ["bench/add.cc"],
- deps = [
- ":XNNPACK",
- ":bench_utils",
- "@cpuinfo",
- "@pthreadpool",
- ],
+ deps = OPERATOR_BENCHMARK_DEPS,
)
xnnpack_benchmark(
name = "average_pooling_bench",
srcs = ["bench/average-pooling.cc"],
copts = xnnpack_optional_tflite_copts(),
- deps = [
- ":XNNPACK",
- ":bench_utils",
- "@cpuinfo",
- "@pthreadpool",
- ] + xnnpack_optional_tflite_deps(),
+ deps = OPERATOR_BENCHMARK_DEPS + xnnpack_optional_tflite_deps(),
)
xnnpack_benchmark(
name = "channel_shuffle_bench",
srcs = ["bench/channel-shuffle.cc"],
- deps = [
- ":XNNPACK",
- ":bench_utils",
- "@cpuinfo",
- "@pthreadpool",
- ],
+ deps = OPERATOR_BENCHMARK_DEPS,
)
xnnpack_benchmark(
name = "convolution_bench",
srcs = ["bench/convolution.cc"],
copts = xnnpack_optional_tflite_copts() + xnnpack_optional_armcl_copts(),
- deps = [
- ":XNNPACK",
- ":bench_utils",
- "@cpuinfo",
- "@pthreadpool",
- ] + xnnpack_optional_tflite_deps() + xnnpack_optional_armcl_deps(),
+ 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 = [
- ":XNNPACK",
- ":bench_utils",
- "@cpuinfo",
- "@pthreadpool",
- ] + xnnpack_optional_tflite_deps(),
+ deps = OPERATOR_BENCHMARK_DEPS + xnnpack_optional_tflite_deps(),
)
xnnpack_benchmark(
name = "global_average_pooling_bench",
srcs = ["bench/global-average-pooling.cc"],
- deps = [
- ":XNNPACK",
- ":bench_utils",
- "@cpuinfo",
- "@pthreadpool",
- ],
+ deps = OPERATOR_BENCHMARK_DEPS,
)
xnnpack_benchmark(
name = "max_pooling_bench",
srcs = ["bench/max-pooling.cc"],
- deps = [
- ":XNNPACK",
- ":bench_utils",
- "@cpuinfo",
- "@pthreadpool",
- ],
+ deps = OPERATOR_BENCHMARK_DEPS,
)
xnnpack_benchmark(
name = "sigmoid_bench",
srcs = ["bench/sigmoid.cc"],
copts = xnnpack_optional_tflite_copts(),
- deps = [
- ":XNNPACK",
- ":bench_utils",
- "@cpuinfo",
- "@pthreadpool",
- ] + xnnpack_optional_tflite_deps(),
+ deps = OPERATOR_BENCHMARK_DEPS + xnnpack_optional_tflite_deps(),
)
xnnpack_benchmark(
name = "prelu_bench",
srcs = ["bench/prelu.cc"],
copts = xnnpack_optional_tflite_copts(),
- deps = [
- ":XNNPACK",
- ":bench_utils",
- "@cpuinfo",
- "@pthreadpool",
- ] + xnnpack_optional_tflite_deps(),
+ deps = OPERATOR_BENCHMARK_DEPS + xnnpack_optional_tflite_deps(),
)
xnnpack_benchmark(
name = "softmax_bench",
srcs = ["bench/softmax.cc"],
copts = xnnpack_optional_tflite_copts(),
- deps = [
- ":XNNPACK",
- ":bench_utils",
- "@cpuinfo",
- "@pthreadpool",
- ] + xnnpack_optional_tflite_deps(),
+ deps = OPERATOR_BENCHMARK_DEPS + xnnpack_optional_tflite_deps(),
)
############################# End-to-end benchmarks ############################
@@ -3141,11 +2993,7 @@
"test/add-nc.cc",
"test/add-operator-tester.h",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3154,11 +3002,7 @@
"test/add-nd.cc",
"test/binary-elementwise-operator-tester.h",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3167,11 +3011,7 @@
"test/argmax-pooling-nhwc.cc",
"test/argmax-pooling-operator-tester.h",
] + OPERATOR_TEST_PARAMS_HDRS,
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3180,11 +3020,7 @@
"test/average-pooling-nhwc.cc",
"test/average-pooling-operator-tester.h",
] + OPERATOR_TEST_PARAMS_HDRS,
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3193,11 +3029,7 @@
"test/channel-pad-nc.cc",
"test/channel-pad-operator-tester.h",
] + OPERATOR_TEST_PARAMS_HDRS,
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3206,11 +3038,7 @@
"test/channel-shuffle-nc.cc",
"test/channel-shuffle-operator-tester.h",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3219,11 +3047,7 @@
"test/clamp-nc.cc",
"test/clamp-operator-tester.h",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3232,11 +3056,7 @@
"test/convolution-nhwc.cc",
"test/convolution-operator-tester.h",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3245,11 +3065,7 @@
"test/convolution-nchw.cc",
"test/convolution-operator-tester.h",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3258,11 +3074,7 @@
"test/deconvolution-nhwc.cc",
"test/deconvolution-operator-tester.h",
] + OPERATOR_TEST_PARAMS_HDRS,
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3271,11 +3083,7 @@
"test/binary-elementwise-operator-tester.h",
"test/divide-nd.cc",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3284,11 +3092,7 @@
"test/fully-connected-nc.cc",
"test/fully-connected-operator-tester.h",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3297,11 +3101,7 @@
"test/global-average-pooling-nwc.cc",
"test/global-average-pooling-operator-tester.h",
] + OPERATOR_TEST_PARAMS_HDRS,
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3310,11 +3110,7 @@
"test/global-average-pooling-ncw.cc",
"test/global-average-pooling-operator-tester.h",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3323,11 +3119,7 @@
"test/hardswish-nc.cc",
"test/hardswish-operator-tester.h",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3336,11 +3128,7 @@
"test/leaky-relu-nc.cc",
"test/leaky-relu-operator-tester.h",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3349,11 +3137,7 @@
"test/max-pooling-nhwc.cc",
"test/max-pooling-operator-tester.h",
] + OPERATOR_TEST_PARAMS_HDRS,
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3362,11 +3146,7 @@
"test/binary-elementwise-operator-tester.h",
"test/maximum-nd.cc",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3375,11 +3155,7 @@
"test/binary-elementwise-operator-tester.h",
"test/minimum-nd.cc",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3388,11 +3164,7 @@
"test/binary-elementwise-operator-tester.h",
"test/multiply-nd.cc",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3401,11 +3173,7 @@
"test/prelu-nc.cc",
"test/prelu-operator-tester.h",
] + OPERATOR_TEST_PARAMS_HDRS,
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3414,11 +3182,7 @@
"test/resize-bilinear-nhwc.cc",
"test/resize-bilinear-operator-tester.h",
] + OPERATOR_TEST_PARAMS_HDRS,
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3427,11 +3191,7 @@
"test/sigmoid-nc.cc",
"test/sigmoid-operator-tester.h",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3440,11 +3200,7 @@
"test/softmax-nc.cc",
"test/softmax-operator-tester.h",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3453,11 +3209,7 @@
"test/binary-elementwise-operator-tester.h",
"test/subtract-nd.cc",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
xnnpack_unit_test(
@@ -3466,11 +3218,7 @@
"test/unpooling-nhwc.cc",
"test/unpooling-operator-tester.h",
],
- deps = [
- ":XNNPACK",
- "@FP16",
- "@pthreadpool",
- ],
+ deps = OPERATOR_TEST_DEPS,
)
############################# Build configurations #############################