diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2023-10-18 17:22:21 +0100 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-11-02 23:22:09 +0000 |
commit | d1a08ce27ef8d0f6cf77e1b864610aade06edc5c (patch) | |
tree | 777992f45d240361f898b1d21902c2a46c58235f /verif/generator/tosa_utils.py | |
parent | b0b9e33c3500bd8dc9b12ef012d4234b1245247a (diff) | |
download | reference_model-d1a08ce27ef8d0f6cf77e1b864610aade06edc5c.tar.gz |
Compliance mode testing for CONV2D
Added CONV2D data generation.
Updated verify dot product check to latest specification.
Updated test generator and python datagenerator library to create
const files during test generation.
Add support for compliance test sets to conformance test_select.
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I5be3b761a1e3ef259c058e493877cd5a89d5778b
Diffstat (limited to 'verif/generator/tosa_utils.py')
-rw-r--r-- | verif/generator/tosa_utils.py | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/verif/generator/tosa_utils.py b/verif/generator/tosa_utils.py index 14afaa7..7fc5b52 100644 --- a/verif/generator/tosa_utils.py +++ b/verif/generator/tosa_utils.py @@ -51,15 +51,21 @@ class DataGenType(IntEnum): OP_SPECIAL = 4 -# Additional (optional) data for dot product data generator -DG_DOT_PRODUCT_OPTIONAL_INFO = ("acc_type", "kernel", "axis") - - def dtypeIsSupportedByCompliance(dtype): """Types supported by the new data generation and compliance flow.""" + if isinstance(dtype, list) or isinstance(dtype, tuple): + dtype = dtype[0] return dtype in (DType.FP32,) +def getOpNameFromOpListName(opName): + """Get the op name from a TOSA_OP_LIST name that can have suffixes.""" + for name in ("conv2d", "depthwise_conv2d", "transpose_conv2d", "conv3d"): + if opName.startswith(name): + return name + return opName + + def valueToName(item, value): """Get the name of an attribute with the given value. |