diff options
| author | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-02-01 15:54:07 +0000 |
|---|---|---|
| committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-02-08 11:14:04 +0000 |
| commit | 32d0b5af61d978d9932ac5576b42203e57881168 (patch) | |
| tree | eb9f3d6d845edc0f53da06285db1e4736282b4c3 /verif/generator/tosa_arg_gen.py | |
| parent | 6f57e6e665094959aed40c0e388ac81fbd118720 (diff) | |
| download | reference_model-32d0b5af61d978d9932ac5576b42203e57881168.tar.gz | |
Main Compliance: Add RESIZE support
Add RELATIVE verify mode for RESIZE.
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I4fe352579507211dae7a048bf080c24426ce42a2
Diffstat (limited to 'verif/generator/tosa_arg_gen.py')
| -rw-r--r-- | verif/generator/tosa_arg_gen.py | 41 |
1 files changed, 33 insertions, 8 deletions
diff --git a/verif/generator/tosa_arg_gen.py b/verif/generator/tosa_arg_gen.py index a4bced3..4630f35 100644 --- a/verif/generator/tosa_arg_gen.py +++ b/verif/generator/tosa_arg_gen.py @@ -1394,6 +1394,22 @@ class TosaTensorValuesGen: testGen, opName, dtypeList, shapeList, argsDict, error_name ) + @staticmethod + def tvgResize(testGen, opName, dtypeList, shapeList, argsDict, error_name=None): + data_range = TosaTensorValuesGen._get_data_range( + testGen, + dtypeList[0], + TosaTensorValuesGen.TVG_FLOAT_HIGH_VALUE, + ) + if data_range: + argsDict["data_range"] = data_range + # Needed for compliance + argsDict["max_abs_value"] = data_range[1] + + return TosaTensorValuesGen.tvgLazyGenDefault( + testGen, opName, dtypeList, shapeList, argsDict, error_name + ) + # Set the POW exponent high data range TVG_FLOAT_HIGH_VALUE_POW_EXP = { DType.FP32: 10.0, @@ -3343,14 +3359,13 @@ class TosaArgGen: border[0], border[1], ), - [ - mode, - scale, - offset, - border, - dtype, - outputDTypeNew, - ], + { + "mode": mode, + "scale": scale, + "offset": offset, + "border": border, + "output_dtype": outputDTypeNew, + }, ) if arg_to_append in arg_list: # Skip already generated test params @@ -3359,6 +3374,16 @@ class TosaArgGen: # Valid permutation perm += 1 arg_list.append(arg_to_append) + + # Now add data generator types + arg_list = TosaArgGen._add_data_generators( + testGen, + opName, + dtype, + arg_list, + error_name, + ) + # Return list of tuples: (arg_str, args_dict) return arg_list @staticmethod |