From e352f7302544b34d1dee2abf618f345bf2dfd0fc Mon Sep 17 00:00:00 2001 From: Tim Warburton Date: Tue, 25 Oct 2011 22:07:54 -0500 Subject: [PATCH] Split off preprocess_kernel from scheduling. This allows easier access (during debugging) to the kernel that actually gets scheduled. --- loopy/schedule.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/loopy/schedule.py b/loopy/schedule.py index fa224040d..ca017f1c7 100644 --- a/loopy/schedule.py +++ b/loopy/schedule.py @@ -833,7 +833,7 @@ def insert_barriers(kernel, schedule, level=0): # {{{ main scheduling entrypoint -def generate_loop_schedules(kernel, loop_priority=[]): +def preprocess_kernel(kernel): kernel = realize_reduction(kernel) # {{{ check that all CSEs have been realized @@ -859,6 +859,11 @@ def generate_loop_schedules(kernel, loop_priority=[]): chk.check_for_inactive_iname_access(kernel) chk.check_for_write_races(kernel) + return kernel + +def generate_loop_schedules(kernel, loop_priority=[]): + kernel = preprocess_kernel(kernel) + schedule_count = 0 for gen_sched in generate_loop_schedules_internal(kernel, loop_priority): -- GitLab