aboutsummaryrefslogtreecommitdiff
path: root/verif/generator/tosa_verif_build_tests.py
diff options
context:
space:
mode:
authorJames Ward <james.ward@arm.com>2023-02-02 14:56:33 +0000
committerEric Kunze <eric.kunze@arm.com>2023-02-22 16:28:39 +0000
commit30124a8d9ab659f2661f3bb5962ae074d9ac4fa3 (patch)
tree305d31c047dd4704f8496e37026bd40a7b4c9f99 /verif/generator/tosa_verif_build_tests.py
parentc5d759328fbfe9772fba56e63c81d196b3a7620f (diff)
downloadreference_model-30124a8d9ab659f2661f3bb5962ae074d9ac4fa3.tar.gz
Bug fixes for max-batch-size/ofm-depth
* Conformance regen required for tosa-bi: conv2d conv3d depthwise_conv2d transpose_conv2d * Include bug fix for testGen.makeShape() * Include json changes to limit size of tests which now have ofm_depth > 1 Signed-off-by: James Ward <james.ward@arm.com> Change-Id: Ic8221b9a742f5737df523997bee7260f4dfcdef3
Diffstat (limited to 'verif/generator/tosa_verif_build_tests.py')
-rw-r--r--verif/generator/tosa_verif_build_tests.py12
1 files changed, 10 insertions, 2 deletions
diff --git a/verif/generator/tosa_verif_build_tests.py b/verif/generator/tosa_verif_build_tests.py
index bc1ec8e..68e44da 100644
--- a/verif/generator/tosa_verif_build_tests.py
+++ b/verif/generator/tosa_verif_build_tests.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2020-2022, ARM Limited.
+# Copyright (c) 2020-2023, ARM Limited.
# SPDX-License-Identifier: Apache-2.0
import argparse
import re
@@ -96,7 +96,7 @@ def parseArgs(argv):
"--max-batch-size",
dest="max_batch_size",
default=1,
- type=int,
+ type=positive_integer_type,
help="Maximum batch size for NHWC tests",
)
@@ -238,6 +238,14 @@ def parseArgs(argv):
return args
+def positive_integer_type(argv_str):
+ value = int(argv_str)
+ if value <= 0:
+ msg = f"{argv_str} is not a valid positive integer"
+ raise argparse.ArgumentTypeError(msg)
+ return value
+
+
def main(argv=None):
args = parseArgs(argv)