comp-chem/cp2k
1
0
Fork 0
mirror of https://github.com/cp2k/cp2k.git synced 2026-07-28 06:05:29 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Docker: Add Cuda performance test

Browse source
This commit is contained in:
Ole Schütt 2021-03-21 21:52:21 +01:00 • committed by Ole Schütt
parent d725af4dbc
commit 34810910c1
5 changed files with 98 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

45
tools/docker/scripts/plot_performance.py
View file

@ -8,39 +8,32 @@ from collections import OrderedDict
# ======================================================================================
def main():
if len(sys.argv) != 6:
if len(sys.argv) < 4 or (len(sys.argv) - 1) % 3 != 0:
print(
"Usage: plot_performance.py <nprocs> <title> <label> <output_omp> <output_mpi>"
"Usage: plot_performance.py <title_1> <plot_1> <file_1> ... "
"<title_N> <plot_N> <file_N>"
)
sys.exit(1)
nprocs, title, label, output_omp, output_mpi = sys.argv[1:]
timings_omp = parse_timings(output_omp)
timings_mpi = parse_timings(output_mpi)
routines = list(set(list(timings_omp.keys())[:6] + list(timings_mpi.keys())[:6]))
titles, plots, timings = [], [], []
for i in range((len(sys.argv) - 1) // 3):
titles.append(sys.argv[3 * i + 1])
plots.append(sys.argv[3 * i + 2])
timings.append(parse_timings(sys.argv[3 * i + 3]))
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_omp.get(r, 0.0))
routines.sort(reverse=True, key=lambda r: timings[0].get(r, 0.0))
sum_routines_omp = sum([timings_omp.get(r, 0.0) for r in routines])
sum_routines_mpi = sum([timings_mpi.get(r, 0.0) for r in routines])
timings_omp["rest"] = timings_omp["total"] - sum_routines_omp
timings_mpi["rest"] = timings_mpi["total"] - sum_routines_mpi
for titel, plot, timing in zip(titles, plots, timings):
timing["rest"] = timing["total"] - sum([timing.get(r, 0.0) for r in routines])
full_title = f"Timings of {title} with {nprocs} OpenMP Threads"
plot = f"{label}_timings_{nprocs}omp"
print(f'Plot: name="{plot}", title="{full_title}", ylabel="time [s]"')
for r in ["rest"] + routines:
t = timings_omp.get(r, 0.0)
print(f'PlotPoint: plot="{plot}", name="{r}", label="{r}", y={t}, yerr=0.0')
print("")
full_title = f"Timings of {title} with {nprocs} MPI Ranks"
plot = f"{label}_timings_{nprocs}mpi"
print(f'Plot: name="{plot}", title="{full_title}", ylabel="time [s]"')
for r in ["rest"] + routines:
t = timings_mpi.get(r, 0.0)
print(f'PlotPoint: plot="{plot}", name="{r}", label="{r}", y={t}, yerr=0.0')
print("")
full_title = f"Timings of {titel}"
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("")
# ======================================================================================