diff options
author | Won Jeon <won.jeon@arm.com> | 2024-01-18 06:31:55 +0000 |
---|---|---|
committer | Won Jeon <won.jeon@arm.com> | 2024-01-23 16:56:33 +0000 |
commit | 64e4bfe627ded0ba44ff60b23db28c1ff5d73d13 (patch) | |
tree | e4744f2124d90882c97af2159c460ba1d7c46d86 /verif/generator/tosa_test_gen.py | |
parent | 8690a0873fac28acccbb0acb15c16e8337e14df1 (diff) | |
download | reference_model-64e4bfe627ded0ba44ff60b23db28c1ff5d73d13.tar.gz |
Update RESHAPE and TILE conformance testing
Signed-off-by: Won Jeon <won.jeon@arm.com>
Change-Id: Iaf59472cb32e03a92bad87ae9dba1b7548f20268
Diffstat (limited to 'verif/generator/tosa_test_gen.py')
-rw-r--r-- | verif/generator/tosa_test_gen.py | 30 |
1 files changed, 7 insertions, 23 deletions
diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index 67ac367..159ee83 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -1558,22 +1558,14 @@ class TosaTestGen: ): assert len(inputs) == 2 a = inputs[0] - # second input is not properly generated yet - # new_shape = inputs[1] - - # modify inputs[1] by a shape tensor from new_shape arg value - new_shape_attr = args_dict["new_shape"] - shape_array = np.array(new_shape_attr) - shape = shape_array.shape - new_shape = self.ser.addPlaceholder(shape, DType.SHAPE, shape_array) - inputs[1] = new_shape - + shape = inputs[1] + shape_attr = args_dict["new_shape"] result_tensor = OutputShaper.reshapeOp( - self.ser, self.rng, a, new_shape_attr, error_name + self.ser, self.rng, a, shape_attr, error_name ) # Invalidate Input/Output list for error if checks. - input_list = [a.name, new_shape.name] + input_list = [a.name, shape.name] output_list = [result_tensor.name] pCount, cCount = op["operands"] num_operands = pCount + cCount @@ -1725,16 +1717,8 @@ class TosaTestGen: ): assert len(inputs) == 2 a = inputs[0] - # second input is not properly generated yet - # multiples = inputs[1] - - # modify inputs[1] by a shape tensor from multiples arg value + multiples = inputs[1] multiples_attr = args_dict["multiples"] - shape_array = np.int64(np.array(multiples_attr)) - shape = shape_array.shape - multiples = self.ser.addPlaceholder(shape, DType.SHAPE, shape_array) - inputs[1] = multiples - result_tensor = OutputShaper.tileOp( self.ser, self.rng, a, multiples_attr, error_name ) @@ -4236,7 +4220,7 @@ class TosaTestGen: "build_fcn": ( build_reshape, TosaTensorGen.tgBasic, - TosaTensorValuesGen.tvgLazyGenDefault, + TosaTensorValuesGen.tvgReshape, TosaArgGen.agReshape, ), "types": TYPE_FIB, @@ -4302,7 +4286,7 @@ class TosaTestGen: "build_fcn": ( build_tile, TosaTensorGen.tgBasic, - TosaTensorValuesGen.tvgLazyGenDefault, + TosaTensorValuesGen.tvgTile, TosaArgGen.agTile, ), "types": TYPE_FIB, |