From bb0935f868a5ab09403cf3628848655b06ac1dec Mon Sep 17 00:00:00 2001 From: Jeremy Johnson Date: Thu, 14 Sep 2023 16:43:48 +0100 Subject: Update verifier library data-type support Make compliance meta-data data-type required for all. Add data-type checking for all verifier modes. Add initial enum support for new ROUND compliance mode. Improve print out information from library. Use numpy ctypes.data_as to get f16 support compared to ctypes_lib. Signed-off-by: Jeremy Johnson Change-Id: Ie983ba4ea958a88556f30c09b3ebc19cd9ec96b7 --- verif/generator/tosa_test_gen.py | 14 ++++++++++---- verif/generator/tosa_utils.py | 1 + 2 files changed, 11 insertions(+), 4 deletions(-) (limited to 'verif/generator') diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index d15f785..8beb2ae 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -282,17 +282,21 @@ class TosaTestGen: ) def tensorComplianceMetaData(self, op, argsDict, outputTensor, errorName): - if errorName: - # No compliance for error tests + if errorName or not gtu.dtypeIsFloat(outputTensor.dtype): + # No compliance for error tests or integer tests currently return None + # Create compliance meta data for expected output tensor - compliance_tens = {"mode": None} + compliance_tens = { + "mode": None, + # Data type is needed for all FP runs, as refmodel precise mode produces FP64 + "data_type": gtu.DTYPE_ATTRIBUTES[outputTensor.dtype]["json"], + } if argsDict["dg_type"] == gtu.DataGenType.DOT_PRODUCT: mode = gtu.ComplianceMode.DOT_PRODUCT compliance_tens["dot_product_info"] = { "s": argsDict["s"], "ks": argsDict["ks"], - "data_type": gtu.DTYPE_ATTRIBUTES[outputTensor.dtype]["json"], } elif argsDict["dg_type"] == gtu.DataGenType.OP_SPECIAL: mode = gtu.ComplianceMode.FP_SPECIAL @@ -301,6 +305,8 @@ class TosaTestGen: compliance_tens["ulp_info"] = {"ulp": op["compliance"]["ulp"]} elif op["op"] == Op.REDUCE_PRODUCT: mode = gtu.ComplianceMode.REDUCE_PRODUCT + elif op["op"] in (Op.ADD, Op.MUL, Op.SUB, Op.CEIL, Op.FLOOR, Op.CAST): + mode = gtu.ComplianceMode.ROUND else: mode = gtu.ComplianceMode.EXACT compliance_tens["mode"] = gtu.ComplianceMode(mode).name diff --git a/verif/generator/tosa_utils.py b/verif/generator/tosa_utils.py index 75a0df5..dddc320 100644 --- a/verif/generator/tosa_utils.py +++ b/verif/generator/tosa_utils.py @@ -38,6 +38,7 @@ class ComplianceMode(IntEnum): ULP = 2 FP_SPECIAL = 3 REDUCE_PRODUCT = 4 + ROUND = 5 class DataGenType(IntEnum): -- cgit v1.2.1