wrpll/thls: stop at return statement

2019-07-19 16:27:29 +08:00 · 2019-07-19 16:27:29 +08:00 · 831b3514d3
parent 34222b3f38
commit 831b3514d3
1 changed files with 2 additions and 0 deletions
--- a/artiq/gateware/wrpll/thls.py
+++ b/artiq/gateware/wrpll/thls.py
@ -286,6 +286,8 @@ def compile(processor, function):
    arg_r = astcompiler.input(arg)
    for node in body:
        astcompiler.emit(node)
+        if isinstance(node, ast.Return):
+            break

    scheduler = Scheduler(processor, len(astcompiler.data), astcompiler.program)
    scheduler.schedule()