diff options
| -rw-r--r-- | OPTIONS.md | 16 | ||||
| -rw-r--r-- | ethosu/vela/vela.py | 8 |
2 files changed, 24 insertions, 0 deletions
@@ -223,6 +223,22 @@ be a power of two and greater or equal to 16. vela network.tflite --allocation-alignment 128 ``` +### Recursion Limit + +Sets the Python internal limit to depth of recursion. It may be +necessary to increase this from the default for very large networks +due to the recursive nature of the graph traversal algorithm. +If Vela fails with a `RecursionError`, try increasing the limit using +this option to see if it resolves the issue. +Please note that this option may not work as intended on Microsoft Windows +systems, as there is a hard limit on thread stack size. +**Type: Integer** +**Default: 1000** + +```bash +vela network.tflite --recursion-limit 2000 +``` + ## Verbose Print Options All of the options below are disabled by default and enabling them will add diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py index b8a3b9f2..9e237f84 100644 --- a/ethosu/vela/vela.py +++ b/ethosu/vela/vela.py @@ -398,6 +398,12 @@ def main(args=None): " operator inputs and outputs (default: %(default)s)" ), ) + parser.add_argument( + "--recursion-limit", + type=int, + default=1000, + help="Set the recursion depth limit, may result in RecursionError if too low (default: %(default)s)", + ) args = parser.parse_args(args=args) # Generate the supported ops report and exit @@ -431,6 +437,8 @@ def main(args=None): if v.startswith("verbose") and v != "verbose_all": setattr(args, v, True) + sys.setrecursionlimit(args.recursion_limit) + arch = architecture_features.ArchitectureFeatures( vela_config_files=args.config, system_config=args.system_config, |