/usr/lib/python3/dist-packages/arrayfire/timer.py is in python3-arrayfire 3.3.20160624-2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 | #######################################################
# Copyright (c) 2015, ArrayFire
# All rights reserved.
#
# This file is distributed under 3-clause BSD license.
# The complete license agreement can be obtained at:
# http://arrayfire.com/licenses/BSD-3-Clause
########################################################
"""
Functions to time arrayfire.
"""
from .library import *
from .device import (sync, eval)
from time import time
import math
def timeit(af_func, *args):
"""
Function to time arrayfire functions.
Parameters
----------
af_func : arrayfire function
*args : arguments to `af_func`
Returns
--------
t : Time in seconds
"""
sample_trials = 3
sample_time = 1E20
for i in range(sample_trials):
start = time()
res = af_func(*args)
eval(res)
sync()
sample_time = min(sample_time, time() - start)
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
|