From 7e673648db828820f775062e9634e739f0bce2d8 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Fri, 8 Jul 2011 14:12:52 -0400
Subject: [PATCH] Slight changes to timing code in response to Tim's odd
 results.

---
 loopy/__init__.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/loopy/__init__.py b/loopy/__init__.py
index a37782913..43b20c47d 100644
--- a/loopy/__init__.py
+++ b/loopy/__init__.py
@@ -1483,9 +1483,10 @@ def drive_timing_run(kernel_generator, queue, launch, flop_count=None):
                     launch(compiled_knl.cl_kernel,
                         compiled.global_size_func, compiled.local_size_func,
                         check=check))
-        queue.finish()
+        for evt in events:
+            evt.wait()
 
-        return 1e-9*sum(evt.profile.END-evt.profile.START for evt in events)/timing_rounds
+        return sum(1e-9*evt.profile.END-1e-9*evt.profile.START for evt in events)/timing_rounds
 
     soln_count = 0
     for kernel in kernel_generator:
-- 
GitLab