From 321ec9283545e5e3181faa225c4bd182224916c9 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 12 Apr 2015 17:50:22 -0500
Subject: [PATCH] Log scheduling time
---
loopy/schedule.py | 10 ++++++++--
1 file changed, 8 insertions(+), 2 deletions(-)
diff --git a/loopy/schedule.py b/loopy/schedule.py
index 95eaf7a8b..6ef5378d6 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
--
GitLab