diff --git a/loopy/schedule.py b/loopy/schedule.py index 95eaf7a8b75f7b8058be0c2ab3a0f076e14353dd..6ef5378d6456ca022f9dd27ee87e593810981e98 100644 --- a/loopy/schedule.py +++ b/loopy/schedule.py @@ -1025,8 +1025,6 @@ def generate_loop_schedules(kernel, debug_args={}): from loopy.check import pre_schedule_checks pre_schedule_checks(kernel) - logger.info("%s: schedule start" % kernel.name) - schedule_count = 0 debug = ScheduleDebugger(**debug_args) @@ -1134,6 +1132,11 @@ def get_one_scheduled_kernel(kernel): kernel_count = 0 + from time import time + start_time = time() + + logger.info("%s: schedule start" % kernel.name) + for scheduled_kernel in generate_loop_schedules(kernel): kernel_count += 1 @@ -1145,6 +1148,9 @@ def get_one_scheduled_kernel(kernel): ambiguous = True break + logger.info("%s: scheduling done after %.2f s" % ( + kernel.name, time()-start_time)) + if ambiguous: from warnings import warn from loopy.diagnostic import LoopyWarning