[python-arrayfire] 169/250: Fixing timeit to work with python2

[Ghislain Vaillant]

This is an automated email from the git hooks/post-receive script.

ghisvail-guest pushed a commit to branch debian/master
in repository python-arrayfire.

commit caeef361de4688c56675b1ad53a3cea21691df7a
Author: Pavan Yalamanchili <pavan at arrayfire.com>
Date:   Wed Dec 9 18:49:26 2015 -0500

    Fixing timeit to work with python2
---
 arrayfire/timer.py | 38 ++++++++++++++++++++++----------------
 1 file changed, 22 insertions(+), 16 deletions(-)

diff --git a/arrayfire/timer.py b/arrayfire/timer.py
index ff33407..6f6c3ef 100644
--- a/arrayfire/timer.py
+++ b/arrayfire/timer.py
@@ -13,8 +13,9 @@ Functions to time arrayfire functions
 from .library import *
 from .device import (sync, eval)
 from time import time
+import math
 
-def timeit(af_func, *args, min_iters = 10):
+def timeit(af_func, *args):
     """
     Function to time arrayfire functions.
 
@@ -25,27 +26,32 @@ def timeit(af_func, *args, min_iters = 10):
 
     *args      : arguments to `af_func`
 
-    min_iters  : Minimum number of iterations to be run for `af_func`
-
     Returns
     --------
 
     t   : Time in seconds
     """
 
-    res = af_func(*args)
-    eval(res)
-    sync()
+    sample_trials = 3
 
-    start = time()
-    elapsed = 0
-    num_iters = 0
-    while elapsed < 1:
-        for n in range(min_iters):
-            res = af_func(*args)
-            eval(res)
+    sample_time = 1E20
+
+    for i in range(sample_trials):
+        start = time()
+        res = af_func(*args)
+        eval(res)
         sync()
-        elapsed += time() - start
-        num_iters += min_iters
+        sample_time = min(sample_time, time() - start)
 
-    return elapsed / num_iters
+    if (sample_time >= 0.5):
+        return sample_time
+
+    num_iters = max(math.ceil(1.0 / sample_time), 3.0)
+
+    start = time()
+    for i in range(int(num_iters)):
+        res = af_func(*args)
+        eval(res)
+    sync()
+    sample_time = (time() - start) / num_iters
+    return sample_time

-- 
Alioth's /usr/local/bin/git-commit-notice on /srv/git.debian.org/git/debian-science/packages/python-arrayfire.git