commit 2b9efd8fb7df7a989e98df704afd07ca3664d5d8
author Marat Dukhan <maratek@google.com> Mon Jun 08 01:09:31 2020 -0700
committer XNNPACK Team <xnnpack-github-robot@google.com> Mon Jun 08 01:10:32 2020 -0700
tree fd323f4e47abf7687761837002fe137fff178600
parent 9d3a459441c272d82be14b579656b961066eba2c

Vector Unary (ABS/NEG/SQR) micro-kernels

PiperOrigin-RevId: 315228488

tools/generate-vunary-test.py

diff --git a/tools/generate-vunary-test.py b/tools/generate-vunary-test.py
index 4779643..d58f2b5 100755
--- a/tools/generate-vunary-test.py
+++ b/tools/generate-vunary-test.py
@@ -27,11 +27,14 @@
 
 
 def split_ukernel_name(name):
-  match = re.match(r"^xnn_(f16|f32)_(sigmoid)_(fact_)?ukernel__(.+)_x(\d+)$", name)
+  match = re.match(r"^xnn_(f16|f32)_(sigmoid|vabs|vneg|vsqr)_(fact_)?ukernel__(.+)_x(\d+)$", name)
   if match is None:
     raise ValueError("Unexpected microkernel name: " + name)
   op_type = {
+    "vabs": "Abs",
+    "vneg": "Neg",
     "sigmoid": "Sigmoid",
+    "vsqr": "Sqr",
   }[match.group(2)]
   batch_tile = int(match.group(5))
 
@@ -109,7 +112,7 @@
   _, test_name = ukernel.split("_", 1)
   _, datatype, _ = ukernel.split("_", 2)
   test_args = [
-    ukernel,
+    "xnn_f32_vunary_ukernel_function(%s)" % ukernel,
     "VUnOpMicrokernelTester::OpType::%s" % op_type,
   ]
   if not isa or isa == "psimd":