def annotate()

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