LRELU (Leaky ReLU) micro-kernels
PiperOrigin-RevId: 315791296
diff --git a/BUILD.bazel b/BUILD.bazel
index 955a785..268969f 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -274,6 +274,9 @@
"src/f32-vbinary/gen/vsubc-minmax-scalar-x1.c",
"src/f32-vbinary/gen/vsubc-minmax-scalar-x2.c",
"src/f32-vbinary/gen/vsubc-minmax-scalar-x4.c",
+ "src/f32-vlrelu/gen/scalar-x1.c",
+ "src/f32-vlrelu/gen/scalar-x2.c",
+ "src/f32-vlrelu/gen/scalar-x4.c",
"src/f32-vmulcaddc/gen/c1-minmax-scalar-2x.c",
"src/f32-vmulcaddc/gen/c2-minmax-scalar-2x.c",
"src/f32-vmulcaddc/gen/c4-minmax-scalar-2x.c",
@@ -458,6 +461,9 @@
"src/f32-vbinary/gen/vsubc-minmax-wasm-x1.c",
"src/f32-vbinary/gen/vsubc-minmax-wasm-x2.c",
"src/f32-vbinary/gen/vsubc-minmax-wasm-x4.c",
+ "src/f32-vlrelu/gen/wasm-x1.c",
+ "src/f32-vlrelu/gen/wasm-x2.c",
+ "src/f32-vlrelu/gen/wasm-x4.c",
"src/f32-vmulcaddc/gen/c1-minmax-wasm-2x.c",
"src/f32-vmulcaddc/gen/c2-minmax-wasm-2x.c",
"src/f32-vmulcaddc/gen/c4-minmax-wasm-2x.c",
@@ -568,6 +574,8 @@
"src/f32-vbinary/gen/vsub-minmax-psimd-x8.c",
"src/f32-vbinary/gen/vsubc-minmax-psimd-x4.c",
"src/f32-vbinary/gen/vsubc-minmax-psimd-x8.c",
+ "src/f32-vlrelu/gen/psimd-x4.c",
+ "src/f32-vlrelu/gen/psimd-x8.c",
"src/f32-vmulcaddc/gen/c4-minmax-psimd-2x.c",
"src/f32-vmulcaddc/gen/c8-minmax-psimd-2x.c",
"src/f32-vunary/gen/vabs-psimd-x4.c",
@@ -781,6 +789,8 @@
"src/f32-vbinary/gen/vsub-minmax-neon-x8.c",
"src/f32-vbinary/gen/vsubc-minmax-neon-x4.c",
"src/f32-vbinary/gen/vsubc-minmax-neon-x8.c",
+ "src/f32-vlrelu/gen/neon-x4.c",
+ "src/f32-vlrelu/gen/neon-x8.c",
"src/f32-vmulcaddc/gen/c4-minmax-neon-2x.c",
"src/f32-vmulcaddc/gen/c8-minmax-neon-2x.c",
"src/f32-vrnd/gen/vrndne-neon-x4.c",
@@ -1267,6 +1277,8 @@
"src/f32-vbinary/gen/vsub-minmax-sse-x8.c",
"src/f32-vbinary/gen/vsubc-minmax-sse-x4.c",
"src/f32-vbinary/gen/vsubc-minmax-sse-x8.c",
+ "src/f32-vlrelu/gen/sse-x4.c",
+ "src/f32-vlrelu/gen/sse-x8.c",
"src/f32-vmulcaddc/gen/c4-minmax-sse-2x.c",
"src/f32-vmulcaddc/gen/c8-minmax-sse-2x.c",
"src/f32-vunary/gen/vabs-sse-x4.c",
@@ -1463,6 +1475,8 @@
"src/f32-vbinary/gen/vsub-minmax-avx-x16.c",
"src/f32-vbinary/gen/vsubc-minmax-avx-x8.c",
"src/f32-vbinary/gen/vsubc-minmax-avx-x16.c",
+ "src/f32-vlrelu/gen/avx-x8.c",
+ "src/f32-vlrelu/gen/avx-x16.c",
"src/f32-vscale/avx-unroll32.c",
"src/f32-vrnd/gen/vrndne-avx-x8.c",
"src/f32-vrnd/gen/vrndne-avx-x16.c",
@@ -1751,6 +1765,8 @@
"src/f32-vbinary/gen/vsub-minmax-avx512f-x32.c",
"src/f32-vbinary/gen/vsubc-minmax-avx512f-x16.c",
"src/f32-vbinary/gen/vsubc-minmax-avx512f-x32.c",
+ "src/f32-vlrelu/gen/avx512f-x16.c",
+ "src/f32-vlrelu/gen/avx512f-x32.c",
"src/f32-vscale/avx512f-unroll64.c",
"src/f32-vscaleexpminusmax/gen/avx512f-p5-scalef-x16.c",
"src/f32-vscaleexpminusmax/gen/avx512f-p5-scalef-x32.c",
@@ -4118,6 +4134,15 @@
)
xnnpack_unit_test(
+ name = "f32_vlrelu_test",
+ srcs = [
+ "test/f32-vlrelu.cc",
+ "test/vunary-microkernel-tester.h",
+ ] + MICROKERNEL_TEST_HDRS,
+ deps = MICROKERNEL_TEST_DEPS,
+)
+
+xnnpack_unit_test(
name = "f32_vneg_test",
srcs = [
"test/f32-vneg.cc",