wrpll/thls: stop at return statement

This commit is contained in:
Sebastien Bourdeauducq 2019-07-19 16:27:29 +08:00
parent 34222b3f38
commit 831b3514d3
1 changed files with 2 additions and 0 deletions

View File

@ -286,6 +286,8 @@ def compile(processor, function):
arg_r = astcompiler.input(arg) arg_r = astcompiler.input(arg)
for node in body: for node in body:
astcompiler.emit(node) astcompiler.emit(node)
if isinstance(node, ast.Return):
break
scheduler = Scheduler(processor, len(astcompiler.data), astcompiler.program) scheduler = Scheduler(processor, len(astcompiler.data), astcompiler.program)
scheduler.schedule() scheduler.schedule()