diff --git a/loopy/schedule.py b/loopy/schedule.py index fa224040d19c48348d85c3e291dd381f1c8ed3ca..ca017f1c75b685a9e150ef473cab1437fd06e3a5 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):