mirror of
https://github.com/cp2k/cp2k.git
synced 2026-07-22 06:55:21 -04:00
75 lines
2.6 KiB
Python
Executable file
75 lines
2.6 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
# author: Ole Schuett
|
|
|
|
import re
|
|
import sys
|
|
from typing import Dict
|
|
from collections import OrderedDict
|
|
|
|
|
|
# ======================================================================================
|
|
def main() -> None:
|
|
if len(sys.argv) < 5 or (len(sys.argv) - 1) % 4 != 0:
|
|
print(
|
|
"Usage: plot_performance.py <title_1> <prefix_1> <postfix_1> <file_1> ... "
|
|
"<title_N> <prefix_N> <postfix_N> <file_N>"
|
|
)
|
|
sys.exit(1)
|
|
|
|
titles, prefixes, postfixes, timings = [], [], [], []
|
|
for i in range((len(sys.argv) - 1) // 4):
|
|
titles.append(sys.argv[4 * i + 1])
|
|
prefixes.append(sys.argv[4 * i + 2])
|
|
postfixes.append(sys.argv[4 * i + 3])
|
|
timings.append(parse_timings(sys.argv[4 * i + 4]))
|
|
|
|
routines = list(set(r for t in timings for r in list(t.keys())[:6]))
|
|
routines.remove("total")
|
|
routines.sort(reverse=True, key=lambda r: timings[0].get(r, 0.0))
|
|
|
|
for title, prefix, postfix, timing in zip(titles, prefixes, postfixes, timings):
|
|
print(
|
|
f'PlotPoint: plot="total_timings_{postfix}", name="{prefix}", label="{prefix}", y={timing["total"]}, yerr=0.0'
|
|
)
|
|
print("")
|
|
|
|
for title, prefix, postfix, timing in zip(titles, prefixes, postfixes, timings):
|
|
timing["rest"] = timing["total"] - sum([timing.get(r, 0.0) for r in routines])
|
|
plot = f"{prefix}_timings_{postfix}"
|
|
full_title = f"Timings of {title}"
|
|
print(f'Plot: name="{plot}", title="{full_title}", ylabel="time [s]"')
|
|
for r in ["rest"] + routines:
|
|
t = timing.get(r, 0.0)
|
|
print(f'PlotPoint: plot="{plot}", name="{r}", label="{r}", y={t}, yerr=0.0')
|
|
print("")
|
|
|
|
|
|
# ======================================================================================
|
|
def parse_timings(out_fn: str) -> Dict[str, float]:
|
|
output = open(out_fn, encoding="utf8").read()
|
|
|
|
pattern = r"\n( -+\n - +-\n - +T I M I N G +-\n([^\n]*\n){4}.*? -+)\n"
|
|
match = re.search(pattern, output, re.DOTALL)
|
|
assert match
|
|
report_lines = match.group(1).split("\n")[7:-1]
|
|
print("\nFrom {}:\n{}\n".format(out_fn, match.group(0).strip()))
|
|
|
|
# Extract average self time.
|
|
timings = {}
|
|
for line in report_lines:
|
|
parts = line.split()
|
|
timings[parts[0]] = float(parts[3])
|
|
|
|
# Add total runtime.
|
|
assert report_lines[0].split()[0] == "CP2K"
|
|
timings["total"] = float(report_lines[0].split()[5])
|
|
|
|
# Sort by time, longest first.
|
|
return OrderedDict(sorted(timings.items(), reverse=True, key=lambda kv: kv[1]))
|
|
|
|
|
|
# ======================================================================================
|
|
main()
|
|
|
|
# EOF
|