commit 74342e522ec61e85fde64fe801da9e750b3e2d86
author Won Jeon <won.jeon@arm.com> Tue Jan 09 00:34:40 2024 +0000
committer Eric Kunze <eric.kunze@arm.com> Wed Jan 24 21:01:20 2024 +0000
tree 473a02dcbccb5dcf7aee009682454aa2b914bb64
parent 1f75232dab1b50162ebc420e6e076edeb8a58341

Add conformance testing for shape operators

Signed-off-by: Won Jeon <won.jeon@arm.com>
Change-Id: Ie80570146601c470a3be7c04a9d6e1016a7c547c

verif/conformance/test_select.py

diff --git a/verif/conformance/test_select.py b/verif/conformance/test_select.py
index cebdf62..55eef58 100644
--- a/verif/conformance/test_select.py
+++ b/verif/conformance/test_select.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2021-2023, ARM Limited.
+# Copyright (c) 2021-2024, ARM Limited.
 # SPDX-License-Identifier: Apache-2.0
 """Select generated tests."""
 import argparse
@@ -437,6 +437,12 @@
     name = "add"
 
 
+class AddShapeOperator(Operator):
+    """Test selector for the ADD_SHAPE operator."""
+
+    name = "add_shape"
+
+
 class ArgmaxOperator(Operator):
     """Test selector for the ARGMAX operator."""
 
@@ -507,6 +513,12 @@
     param_names = ["shape", "type", "axis"]
 
 
+class ConcatShapeOperator(Operator):
+    """Test selector for the CONCAT_SHAPE operator."""
+
+    name = "concat_shape"
+
+
 class CondIfOperator(Operator):
     """Test selector for the COND_IF operator."""
 
@@ -520,6 +532,12 @@
     name = "const"
 
 
+class ConstShapeOperator(Operator):
+    """Test selector for the CONST_SHAPE operator."""
+
+    name = "const_shape"
+
+
 class Conv2dOperator(Operator):
     """Test selector for the CONV2D operator."""
 
@@ -548,6 +566,12 @@
     param_names = ["shape", "type", "axis"]
 
 
+class DivShapeOperator(Operator):
+    """Test selector for the DIV_SHAPE operator."""
+
+    name = "div_shape"
+
+
 class EqualOperator(Operator):
     """Test selector for the EQUAL operator."""
 
@@ -696,6 +720,12 @@
     param_names = ["shape", "type", "perm", "shift"]
 
 
+class MulShapeOperator(Operator):
+    """Test selector for the MUL_SHAPE operator."""
+
+    name = "mul_shape"
+
+
 class NegateOperator(Operator):
     """Test selector for the Negate operator."""
 
@@ -849,6 +879,12 @@
     name = "sub"
 
 
+class SubShapeOperator(Operator):
+    """Test selector for the SUB_SHAPE operator."""
+
+    name = "sub_shape"
+
+
 class TableOperator(Operator):
     """Test selector for the TABLE operator."""