From ece09903bd9b55017ea81adb9effbba51da17dfd Mon Sep 17 00:00:00 2001
From: "Peter W. Draper" <p.w.draper@durham.ac.uk>
Date: Thu, 14 Mar 2019 15:51:27 +0000
Subject: [PATCH] Improvements to postprocessing script
---
tools/process_memuse.py | 20 ++++++++++++++++----
1 file changed, 16 insertions(+), 4 deletions(-)
diff --git a/tools/process_memuse.py b/tools/process_memuse.py
index dccd38c58d..87e83ead31 100755
--- a/tools/process_memuse.py
+++ b/tools/process_memuse.py
@@ -35,12 +35,15 @@ if len(sys.argv) != 2:
memuse = OrderedDict()
labels = {}
totalmem = 0
+process_use = ""
+peak = 0.0
with open(sys.argv[1]) as infile:
+ print "# tic label MB"
for line in infile:
if line[0] == "#":
if "# Current use:" in line:
- print line
+ process_use = line[14:-1]
else:
tic , adr, rank, step, allocated, label, size = line.split()
rank = int(rank)
@@ -70,6 +73,8 @@ with open(sys.argv[1]) as infile:
# Unmatched free, skip for now.
doprint = False
if doprint:
+ if totalmem > peak:
+ peak = totalmem
print tic, label, totalmem/(1048576.0)
totals = {}
@@ -81,8 +86,15 @@ for adr in labels:
else:
totals[labels[adr]] = memuse[adr][0]
-print "# Memory use by label"
+print "# Aligned memory use by label"
+total = 0.0
for label in sorted(totals):
- print "## ", label, totals[label]/(1048576.0)
-
+ mem = totals[label]/(1048576.0)
+ total = total + mem
+ print "## ", label, '{:.3f}'.format(mem)
+print "# Total aligned memory still in use : ", '{:.3f}'.format(total), " (MB)"
+print "# Peak aligned memory usage : ", '{:.3f}'.format(peak/1048576.0), " (MB)"
+if process_use != "":
+ print "#"
+ print "# Memory use by process (all/system):", process_use
sys.exit(0)
--
GitLab