in utils/benchmark_utils.py [0:0]
def annotate(f, title): @functools.wraps(f) def _f(*args, **kwargs): with record_function(title): return f(*args, **kwargs) return _f