#######################################################
# 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
[docs]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
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.4