diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-01-15 15:12:17 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2024-01-18 17:42:48 +0000 |
commit | 9f5febe05901bfbd3919ef17f2caea8087cd9ccf (patch) | |
tree | 1f486997d02935b9cf5e86da097b5c24f32e07b6 /verif/generator/tosa_test_gen.py | |
parent | 3eafe6640614462cd450f4b67fbaa747f482a7fb (diff) | |
download | reference_model-9f5febe05901bfbd3919ef17f2caea8087cd9ccf.tar.gz |
Main Compliance testing for TILE
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I7f7449c652844097fec6226bcc32ca765b40bc04
Diffstat (limited to 'verif/generator/tosa_test_gen.py')
-rw-r--r-- | verif/generator/tosa_test_gen.py | 32 |
1 files changed, 24 insertions, 8 deletions
diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index d4dbef4..b1c53f5 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -1714,12 +1714,20 @@ class TosaTestGen: self.ser.addOperator(op["op"], input_list, output_list, attr) return result_tens - def build_tile(self, op, a, multiples, validator_fcns=None, error_name=None): - result_tens = OutputShaper.tileOp(self.ser, self.rng, a, multiples, error_name) + def build_tile( + self, op, inputs, args_dict, validator_fcns=None, error_name=None, qinfo=None + ): + assert len(inputs) == 1 + a = inputs[0] + multiples = args_dict["multiples"] + + result_tensor = OutputShaper.tileOp( + self.ser, self.rng, a, multiples, error_name + ) # Invalidate Input/Output list for error if checks. input_list = [a.name] - output_list = [result_tens.name] + output_list = [result_tensor.name] pCount, cCount = op["operands"] num_operands = pCount + cCount input_list, output_list = TosaErrorIfArgGen.eiInvalidateInputOutputList( @@ -1732,10 +1740,10 @@ class TosaTestGen: error_name, op=op, input_shape=a.shape, - output_shape=result_tens.shape, + output_shape=result_tensor.shape, input_dtype=a.dtype, - output_dtype=result_tens.dtype, - result_tensors=[result_tens], + output_dtype=result_tensor.dtype, + result_tensors=[result_tensor], input_list=input_list, output_list=output_list, num_operands=num_operands, @@ -1747,7 +1755,12 @@ class TosaTestGen: attr.TileAttribute(multiples) self.ser.addOperator(op["op"], input_list, output_list, attr) - return result_tens + + compliance = self.tensorComplianceMetaData( + op, a.dtype, args_dict, result_tensor, error_name + ) + + return TosaTestGen.BuildInfo(result_tensor, compliance) def build_gather( self, op, inputs, args_dict, validator_fcns=None, error_name=None, qinfo=None @@ -4269,7 +4282,7 @@ class TosaTestGen: "build_fcn": ( build_tile, TosaTensorGen.tgBasic, - TosaTensorValuesGen.tvgDefault, + TosaTensorValuesGen.tvgLazyGenDefault, TosaArgGen.agTile, ), "types": TYPE_FIB, @@ -4281,6 +4294,9 @@ class TosaTestGen: TosaErrorValidator.evRankMismatch, TosaErrorValidator.evWrongRank, ), + "data_gen": { + "fp": (gtu.DataGenType.PSEUDO_RANDOM,), + }, }, "transpose": { "op": Op.TRANSPOSE, |