diff options
| author | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-01-10 13:13:35 +0000 |
|---|---|---|
| committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-01-17 14:23:50 +0000 |
| commit | 3eafe6640614462cd450f4b67fbaa747f482a7fb (patch) | |
| tree | 61fddb6ad731a14f6b46b8e7220fae76a8624c28 /verif/generator/tosa_arg_gen.py | |
| parent | 7b9abced233128f4128d84294a0f9d6b432a24cf (diff) | |
| download | reference_model-3eafe6640614462cd450f4b67fbaa747f482a7fb.tar.gz | |
Main Compliance testing for CONCAT
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: If08c560eac171124641dac66d7de9d5053db72df
Diffstat (limited to 'verif/generator/tosa_arg_gen.py')
| -rw-r--r-- | verif/generator/tosa_arg_gen.py | 22 |
1 files changed, 14 insertions, 8 deletions
diff --git a/verif/generator/tosa_arg_gen.py b/verif/generator/tosa_arg_gen.py index bfe7f0d..e6d7df8 100644 --- a/verif/generator/tosa_arg_gen.py +++ b/verif/generator/tosa_arg_gen.py @@ -697,6 +697,14 @@ class TosaTensorValuesGen: ): # Variable inputs versus constants pCount, cCount = testGen.TOSA_OP_LIST[opName]["operands"] + if "p_count" in argsDict: + # Override for operators like CONCAT + pCount = argsDict["p_count"] + cCount = argsDict["c_count"] + assert pCount + cCount == len( + shapeList + ), "Placeholders & Constant tensors must match shapes list" + tens_ser_list = [] if ( @@ -1154,15 +1162,13 @@ class TosaTensorValuesGen: testGen, shapeList, argsDict["axis"], error_name ) - tens_ser_list = [] - tens_ser_list.extend( - testGen.buildPlaceholderTensors(shapeList[0:count], dtypeList[0:count]) - ) - tens_ser_list.extend( - testGen.buildConstTensors(shapeList[count:], dtypeList[count:]) - ) + # Override default pCount/cCount for operator + argsDict["p_count"] = count + argsDict["c_count"] = len(shapeList) - count - return TosaTensorValuesGen.TVGInfo(tens_ser_list, None) + return TosaTensorValuesGen.tvgLazyGenDefault( + testGen, opName, dtypeList, shapeList, argsDict, error_name + ) @staticmethod def tvgLogicalShift( |