diff options
-rw-r--r-- | verif/tosa_test_gen.py | 2 | ||||
-rwxr-xr-x | verif/tosa_verif_build_tests.py | 3 |
2 files changed, 3 insertions, 2 deletions
diff --git a/verif/tosa_test_gen.py b/verif/tosa_test_gen.py index 0e57a7b..24d7b7b 100644 --- a/verif/tosa_test_gen.py +++ b/verif/tosa_test_gen.py @@ -683,7 +683,7 @@ class TosaArgGen: # Randomly generate legal output dimensions and shift # and then compute the stride and offset based on them - output_dims = [ testGen.randInt(), testGen.randInt() ] + output_dims = [ testGen.randInt(1), testGen.randInt(1) ] in_center_h = (ifm_shape[1] - 1) / 2.0 in_center_w = (ifm_shape[2] - 1) / 2.0 out_center_h = (output_dims[0] - 1) / 2.0 diff --git a/verif/tosa_verif_build_tests.py b/verif/tosa_verif_build_tests.py index 19eb2f4..b8a24dd 100755 --- a/verif/tosa_verif_build_tests.py +++ b/verif/tosa_verif_build_tests.py @@ -127,7 +127,8 @@ def main(): print('{} matching tests'.format(len(testList))) for opName, testStr, dtype, shapeList, testArgs in testList: - print(testStr) + if args.verbose: + print(testStr) ttg.serializeTest(opName, testStr, dtype, shapeList, testArgs) print('Done creating {} tests'.format(len(testList))) |