diff --git a/loopy/schedule/__init__.py b/loopy/schedule/__init__.py index eb7d3b444..ad4e76e7a 100644 --- a/loopy/schedule/__init__.py +++ b/loopy/schedule/__init__.py @@ -2354,9 +2354,7 @@ def print_longest_dead_end(): sched_state, debug=debug, **schedule_gen_kwargs): debug.stop() - new_kernel = _postprocess_schedule(kernel, callables_table, gen_sched) - - yield new_kernel + yield _postprocess_schedule(kernel, callables_table, gen_sched) debug.start()