diff --git a/loopy/schedule/__init__.py b/loopy/schedule/__init__.py index 9342fedb67a675012e3fcdf39a5a9fdb1a4999da..b449c01739a22ac79b2128106207dca51758a543 100644 --- a/loopy/schedule/__init__.py +++ b/loopy/schedule/__init__.py @@ -1763,7 +1763,8 @@ def generate_loop_schedules(kernel, debug_args={}): new_limit = max(rec_limit, len(kernel.instructions) * 2) sys.setrecursionlimit(new_limit) try: - return generate_loop_schedules_inner(kernel, debug_args=debug_args) + for sched in generate_loop_schedules_inner(kernel, debug_args=debug_args): + yield sched finally: sys.setrecursionlimit(rec_limit)