From da66f139191facfde8da3a263dc17060596fa32d Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Fri, 2 Dec 2016 23:32:57 -0600 Subject: [PATCH] Fix setting schedule recursionlimit --- loopy/schedule/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/loopy/schedule/__init__.py b/loopy/schedule/__init__.py index 9342fedb6..b449c0173 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) -- GitLab