comparison contrib/perf.py @ 40680:9d88ae5c635b

perf: add a `setup` argument to run code outside of the timed section

With this new argument, it is possible to perform special setup and cleanup
outside of code actually timed. This is useful to provide more accurate
benchmark.

40679:20d2fd6036ed

    # experimental config: perf.all-timing
    displayall = ui.configbool(b"perf", b"all-timing", False)
    return functools.partial(_timer, fm, displayall=displayall), fm

def stub_timer(fm, func, title=None):
    func()

@contextlib.contextmanager
def timeone():
    r = []

    cstop = util.timer()
    ostop = os.times()
    a, b = ostart, ostop
    r.append((cstop - cstart, b[0] - a[0], b[1]-a[1]))

def _timer(fm, func, title=None, displayall=False):
    gc.collect()
    results = []
    begin = util.timer()
    count = 0
    while True:
        with timeone() as item:
            r = func()
        count += 1
        results.append(item[0])
        cstop = util.timer()

40680:9d88ae5c635b

    # experimental config: perf.all-timing
    displayall = ui.configbool(b"perf", b"all-timing", False)
    return functools.partial(_timer, fm, displayall=displayall), fm

def stub_timer(fm, func, setup=None, title=None):
    func()

@contextlib.contextmanager
def timeone():
    r = []

    cstop = util.timer()
    ostop = os.times()
    a, b = ostart, ostop
    r.append((cstop - cstart, b[0] - a[0], b[1]-a[1]))

def _timer(fm, func, setup=None, title=None, displayall=False):
    gc.collect()
    results = []
    begin = util.timer()
    count = 0
    while True:
        if setup is not None:
            setup()
        with timeone() as item:
            r = func()
        count += 1
        results.append(item[0])
        cstop = util.timer()