diff options
| author | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-01-10 11:07:29 +0000 |
|---|---|---|
| committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-01-17 14:23:06 +0000 |
| commit | 7b9abced233128f4128d84294a0f9d6b432a24cf (patch) | |
| tree | fa5b6bf1923226701afe66e9840509ebb2e88ac8 /verif/generator/tosa_test_gen.py | |
| parent | 7203835d65afb8a06a5cf98072718d93a9b71567 (diff) | |
| download | reference_model-7b9abced233128f4128d84294a0f9d6b432a24cf.tar.gz | |
Main Compliance testing for SELECT
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I7276f2db39e67314c950e972cc1a97b7796dcd18
Diffstat (limited to 'verif/generator/tosa_test_gen.py')
| -rw-r--r-- | verif/generator/tosa_test_gen.py | 28 |
1 files changed, 21 insertions, 7 deletions
diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index 0d072ac..7a759e9 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -601,12 +601,19 @@ class TosaTestGen: return result_tens - def build_select(self, op, cond, a, b, validator_fcns=None, error_name=None): - result_tens = OutputShaper.selectOp(self.ser, self.rng, cond, a, b, error_name) + def build_select( + self, op, inputs, args_dict, validator_fcns=None, error_name=None, qinfo=None + ): + assert len(inputs) == 3 + cond, a, b = inputs + + result_tensor = OutputShaper.selectOp( + self.ser, self.rng, cond, a, b, error_name + ) # Invalidate Input/Output list for error if checks. input_list = [cond.name, a.name, b.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( @@ -623,8 +630,8 @@ class TosaTestGen: input3=b, input_shape=a.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, @@ -636,7 +643,11 @@ class TosaTestGen: input_list, output_list, ) - return result_tens + compliance = self.tensorComplianceMetaData( + op, a.dtype, args_dict, result_tensor, error_name + ) + + return TosaTestGen.BuildInfo(result_tensor, compliance) def build_comparison( self, op, inputs, args_dict, validator_fcns=None, error_name=None, qinfo=None @@ -3882,7 +3893,7 @@ class TosaTestGen: build_select, TosaTensorGen.tgBroadcastFuzz, TosaTensorValuesGen.tvgSelect, - None, + TosaArgGen.agNone, ), "types": TYPE_FIB, "error_if_validators": ( @@ -3894,6 +3905,9 @@ class TosaTestGen: TosaErrorValidator.evDimensionMismatch, TosaErrorValidator.evBroadcastShapesMismatch, ), + "data_gen": { + "fp": (gtu.DataGenType.PSEUDO_RANDOM,), + }, }, # Comparison operators "equal": { |