7b89b8fc3f
Differential Revision: D90237984 fbshipit-source-id: 526fd760f303bf31be4f743bdcd77760496de0de
71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
# flake8: noqa
|
|
|
|
# pyre-unsafe
|
|
|
|
import inspect
|
|
from functools import wraps
|
|
from time import perf_counter
|
|
|
|
DO_TIMING = False
|
|
DISPLAY_LESS_PROGRESS = False
|
|
timer_dict = {}
|
|
counter = 0
|
|
|
|
|
|
def time(f):
|
|
@wraps(f)
|
|
def wrap(*args, **kw):
|
|
if DO_TIMING:
|
|
# Run function with timing
|
|
ts = perf_counter()
|
|
result = f(*args, **kw)
|
|
te = perf_counter()
|
|
tt = te - ts
|
|
|
|
# Get function name
|
|
arg_names = inspect.getfullargspec(f)[0]
|
|
if arg_names[0] == "self" and DISPLAY_LESS_PROGRESS:
|
|
return result
|
|
elif arg_names[0] == "self":
|
|
method_name = type(args[0]).__name__ + "." + f.__name__
|
|
else:
|
|
method_name = f.__name__
|
|
|
|
# Record accumulative time in each function for analysis
|
|
if method_name in timer_dict.keys():
|
|
timer_dict[method_name] += tt
|
|
else:
|
|
timer_dict[method_name] = tt
|
|
|
|
# If code is finished, display timing summary
|
|
if method_name == "Evaluator.evaluate":
|
|
print("")
|
|
print("Timing analysis:")
|
|
for key, value in timer_dict.items():
|
|
print("%-70s %2.4f sec" % (key, value))
|
|
else:
|
|
# Get function argument values for printing special arguments of interest
|
|
arg_titles = ["tracker", "seq", "cls"]
|
|
arg_vals = []
|
|
for i, a in enumerate(arg_names):
|
|
if a in arg_titles:
|
|
arg_vals.append(args[i])
|
|
arg_text = "(" + ", ".join(arg_vals) + ")"
|
|
|
|
# Display methods and functions with different indentation.
|
|
if arg_names[0] == "self":
|
|
print("%-74s %2.4f sec" % (" " * 4 + method_name + arg_text, tt))
|
|
elif arg_names[0] == "test":
|
|
pass
|
|
else:
|
|
global counter
|
|
counter += 1
|
|
print("%i %-70s %2.4f sec" % (counter, method_name + arg_text, tt))
|
|
|
|
return result
|
|
else:
|
|
# If config["TIME_PROGRESS"] is false, or config["USE_PARALLEL"] is true, run functions normally without timing.
|
|
return f(*args, **kw)
|
|
|
|
return wrap
|