module utils.benchmark
#
Short summary#
module onnxcustom.utils.benchmark
Tools to help benchmarking.
Functions#
function |
truncated documentation |
---|---|
Measures a statement and returns the results as a dictionary. |
Documentation#
Tools to help benchmarking.
- onnxcustom.utils.benchmark.measure_time(stmt, context=None, repeat=10, number=50, div_by_number=False)#
Measures a statement and returns the results as a dictionary.
- Parameters:
stmt – string
context – variable to know in a dictionary
repeat – average over repeat experiment
number – number of executions in one row
div_by_number – divide by the number of executions
- Returns:
dictionary
Measure the processing time of a function
<<<
from onnxcustom.utils import measure_time from math import cos res = measure_time("cos(x)", context=dict(cos=cos, x=5.)) print(res)
>>>
{'average': 1.2832914944738149e-05, 'deviation': 9.344914138997379e-07, 'min_exec': 1.2260046787559986e-05, 'max_exec': 1.555995550006628e-05, 'repeat': 10, 'number': 50}
See Timer.repeat for a better understanding of parameter repeat and number. The function returns a duration corresponding to number times the execution of the main statement.