[PATCH] annotate: pre-calculate the "needed" dictionary
Jun Wu
quark at fb.com
Fri Sep 2 15:10:30 UTC 2016
# HG changeset patch
# User Jun Wu <quark at fb.com>
# Date 1472826059 -3600
# Fri Sep 02 15:20:59 2016 +0100
# Node ID c9a9040c95add098eb8e280fa60edaa766154141
# Parent d130a38ef41f3c9e2d2f26df3535d89a93f87301
# Available At https://bitbucket.org/quark-zju/hg-draft
# hg pull https://bitbucket.org/quark-zju/hg-draft -r c9a9040c95ad
annotate: pre-calculate the "needed" dictionary
The "needed" dict is used as a reference counter to free items in the giant
"hist" dict. However, currently it is not very accurate and can lead to
dropping "hist" items unnecessarily, for example, with the following DAG,
-3-
/ \
--1--2--4--
The current algorithm will visit and calculate rev 1 twice, undesired.
However, simply removing "needed" is not okay, because it will consume 10x
memory:
# without any change
% HGRCPATH= lrun ./hg annotate mercurial/commands.py -r d130a38 3>&2 [1]
MEMORY 49074176
CPUTIME 9.213
REALTIME 9.270
# with "needed" removed
MEMORY 637673472
CPUTIME 8.164
REALTIME 8.249
This patch moves "needed" (and "pcache") calculation to a separate DFS to
address the issue. It improves perf (reuses hist) while maintaining low
memory usage:
# with this patch applied
MEMORY 48164864
CPUTIME 7.871
REALTIME 7.956
[1]: lrun is a lightweight sandbox built on Linux cgroup and namespace. It's
used to measure CPU and memory usage here. Source code is available at
github.com/quark-zju/lrun.
diff --git a/mercurial/context.py b/mercurial/context.py
--- a/mercurial/context.py
+++ b/mercurial/context.py
@@ -991,13 +991,26 @@ class basefilectx(object):
# depth-first search.
+ # 1st DFS pre-calculates pcache and needed
visit = [base]
- hist = {}
pcache = {}
needed = {base: 1}
while visit:
f = visit[-1]
- pcached = f in pcache
- if not pcached:
- pcache[f] = parents(f)
+ visit.pop()
+ pl = parents(f)
+ pcache[f] = pl
+ for p in pl:
+ if p not in needed:
+ needed[p] = 1
+ else:
+ needed[p] += 1
+ if p not in pcache:
+ visit.append(p)
+
+ # 2nd DFS does the actual annotate
+ visit[:] = [base]
+ hist = {}
+ while visit:
+ f = visit[-1]
ready = True
@@ -1007,6 +1020,4 @@ class basefilectx(object):
ready = False
visit.append(p)
- if not pcached:
- needed[p] = needed.get(p, 0) + 1
if ready:
visit.pop()
More information about the Mercurial-devel
mailing list