smptests/smpopenmp01/smpopenmp01.py

0001 # SPDX-License-Identifier: BSD-2-Clause
0002
0003 # Copyright (C) 2017, 2024 embedded brains GmbH & Co. KG
0004 #
0005 # Redistribution and use in source and binary forms, with or without
0006 # modification, are permitted provided that the following conditions
0007 # are met:
0008 # 1. Redistributions of source code must retain the above copyright
0009 #    notice, this list of conditions and the following disclaimer.
0010 # 2. Redistributions in binary form must reproduce the above copyright
0011 #    notice, this list of conditions and the following disclaimer in the
0012 #    documentation and/or other materials provided with the distribution.
0013 #
0014 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
0015 # AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
0016 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
0017 # ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
0018 # LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
0019 # CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
0020 # SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
0021 # INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
0022 # CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
0023 # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
0024 # POSSIBILITY OF SUCH DAMAGE.
0025
0026 import json
0027 import re
0028 import matplotlib.pyplot as plt  # type: ignore
0029 from matplotlib import ticker  # type: ignore
0030
0031
0032 def _plot(data: dict) -> None:
0033     _, axes = plt.subplots()
0034     axes.set_title("OpenMP Microbench")
0035     axes.set_xlabel("Number of Threads")
0036     axes.set_ylabel("Relative Duration")
0037     x = list(range(1, len(data) + 1))
0038     axes.xaxis.set_major_locator(ticker.FixedLocator(x))
0039     for key in [
0040             "barrier-bench", "dynamic-bench", "guided-bench", "parallel-bench",
0041             "runtime-bench", "single-bench", "static-bench"
0042     ]:
0043         d = [results[key] for results in data]
0044         y = [x / d[0] for x in d]
0045         axes.plot(x, y, label=key.replace("-bench", ""), marker="o")
0046     axes.legend(loc="best")
0047     plt.savefig("smpopenmp01.png")
0048     plt.savefig("smpopenmp01.pdf")
0049     plt.close()
0050
0051
0052 _JSON_DATA = re.compile(
0053     r"\*\*\* BEGIN OF JSON DATA \*\*\*(.*)"
0054     r"\*\*\* END OF JSON DATA \*\*\*", re.DOTALL)
0055
0056 with open("smpopenmp01.scn", "r", encoding="utf-8") as src:
0057     match = _JSON_DATA.search(src.read())
0058     data = json.loads(match.group(1))
0059
0060 _plot(data)