PySDM-examples 2.91__py3-none-any.whl → 2.93__py3-none-any.whl

PySDM_examples/Bulenok_2023_MasterThesis/__init__.py

@@ -2,7 +2,4 @@
 """
 Box-model coalescence-breakup performance benchmark from
 [Bulenok 2023 MSc thesis](https://www.ap.uj.edu.pl/diplomas/166879)
-
-performance_comparison_Srivastava_Setup.ipynb:
-.. include:: ./performance_comparison_Srivastava_Setup.ipynb.badges.md
 """

PySDM_examples/Bulenok_2023_MasterThesis/performance_comparison_Srivastava_Setup.py

@@ -0,0 +1,225 @@
+import subprocess
+import os
+
+from datetime import datetime
+import numba  # pylint: disable=unused-import
+
+from PySDM_examples.Bulenok_2023_MasterThesis.utils import (
+    go_benchmark,
+    process_results,
+    plot_processed_results,
+    write_to_file,
+)
+from PySDM_examples.Bulenok_2023_MasterThesis.setups import (
+    setup_coalescence_only_sim,
+    setup_breakup_only_sim,
+    setup_coalescence_breakup_sim,
+)
+
+from PySDM.backends import GPU, CPU
+from PySDM.physics import si
+
+
+def main(plot: bool = True, save: str = None):
+
+    TIMESTAMP = str(datetime.now().strftime("%Y-%d-%m_%Hh-%Mm-%Ss"))
+
+    SIM_RUN_FNAME = "env_name_" + TIMESTAMP
+
+    assert not os.path.isfile(SIM_RUN_FNAME)
+
+    cmd = [
+        "bash",
+        "-c",
+        # pylint: disable=no-member
+        f"echo NUMBA_DEFAULT_NUM_THREADS: {numba.config.NUMBA_DEFAULT_NUM_THREADS}"
+        + f">> {SIM_RUN_FNAME}",
+    ]
+    subprocess.run(cmd, check=False)
+    subprocess.run(
+        [  # pylint: disable=no-member
+            "bash",
+            "-c",
+            f"echo NUMBA_NUM_THREADS: {numba.config.NUMBA_NUM_THREADS} >> {SIM_RUN_FNAME}",
+        ],
+        check=False,
+    )
+    subprocess.run(["bash", "-c", f"lscpu >> {SIM_RUN_FNAME}"], check=False)
+    subprocess.run(["bash", "-c", f"nvidia-smi >> {SIM_RUN_FNAME}"], check=False)
+    subprocess.run(["bash", "-c", f"nvidia-smi -L >> {SIM_RUN_FNAME}"], check=False)
+    subprocess.run(["bash", "-c", f"cat /proc/cpuinfo >> {SIM_RUN_FNAME}"], check=False)
+
+    CI = "CI" in os.environ
+
+    exponents = [3, 5, 8, 10, 12, 14, 16, 18, 20, 22, 24] if not CI else [3, 5]
+    n_sds = [2**i for i in exponents]
+
+    numba_n_threads = [1, 2, 4, 5, 6, 8, 10] if not CI else [1, 2]
+
+    n_realisations = 3 if not CI else 2
+    seeds = list(range(n_realisations))
+
+    n_steps_short = 100 if not CI else 3
+    n_steps_full = 2048 if not CI else 3
+
+    # # Benchmark regular setup (without scaling)
+
+    # ### Coalescence-only
+
+    res_coalescence_only = go_benchmark(
+        setup_coalescence_only_sim,
+        n_sds,
+        n_steps_short,
+        seeds,
+        numba_n_threads=numba_n_threads,
+        double_precision=True,
+        sim_run_filename=SIM_RUN_FNAME + "-coalescence",
+        backends=[CPU, GPU],
+    )
+    coalescence_only_processed = process_results(res_coalescence_only)
+    filename = f"{SIM_RUN_FNAME}-results-coalescence-double-n_steps{n_steps_short}"
+    if plot:
+        plot_processed_results(
+            coalescence_only_processed,
+            plot_title=f"coalescence-only (n_steps: {n_steps_short})",
+            plot_filename=filename + ".svg",
+        )
+    if save is not None:
+        write_to_file(filename=filename + ".txt", d=coalescence_only_processed)
+
+    # ### Breakup-only
+
+    res_breakup_only = go_benchmark(
+        setup_breakup_only_sim,
+        n_sds,
+        n_steps_short,
+        seeds,
+        numba_n_threads=numba_n_threads,
+        double_precision=True,
+        sim_run_filename=SIM_RUN_FNAME + "-breakup",
+        backends=[CPU, GPU],
+    )
+    breakup_only_processed = process_results(res_breakup_only)
+    filename = f"{SIM_RUN_FNAME}-results-breakup-double-n_steps{n_steps_short}"
+    if plot:
+        plot_processed_results(
+            breakup_only_processed,
+            plot_title=f"breakup-only  (n_steps: {n_steps_short})",
+            plot_filename=filename + ".svg",
+        )
+    if save is not None:
+        write_to_file(filename=filename + ".txt", d=breakup_only_processed)
+
+    # ### Coalescence and Breakup
+
+    res_coal_breakup = go_benchmark(
+        setup_coalescence_breakup_sim,
+        n_sds,
+        n_steps_full,
+        seeds,
+        numba_n_threads=numba_n_threads,
+        double_precision=True,
+        sim_run_filename=SIM_RUN_FNAME + "-coal-break",
+        backends=[CPU, GPU],
+    )
+    coal_breakup_processed = process_results(res_coal_breakup)
+    filename = f"{SIM_RUN_FNAME}-results-coal_with_breakup-double-n_steps{n_steps_full}"
+    if plot:
+        plot_processed_results(
+            coal_breakup_processed,
+            plot_title=f"coalescence+breakup (n_steps: {n_steps_full})",
+            plot_filename=filename + ".svg",
+        )
+    if save is not None:
+        write_to_file(filename=filename + ".txt", d=coal_breakup_processed)
+
+    # # Benchmark setup with scaling
+
+    def total_number_from_n_sd(n_sd):
+        return n_sd * 1e8
+
+    def dv_from_n_sd(n_sd):
+        return n_sd * (0.125 * si.m**3)
+
+    # ### Coalescence-only
+
+    res_coalescence_only_scaled = go_benchmark(
+        setup_coalescence_only_sim,
+        n_sds,
+        n_steps_short,
+        seeds,
+        numba_n_threads=numba_n_threads,
+        double_precision=True,
+        sim_run_filename=SIM_RUN_FNAME + "-coalescence-scaled",
+        total_number=total_number_from_n_sd,
+        dv=dv_from_n_sd,
+        backends=[CPU, GPU],
+    )
+    coalescence_only_processed_scaled = process_results(res_coalescence_only_scaled)
+    filename = (
+        f"{SIM_RUN_FNAME}-results-scaled-coalescence-double-n_steps{n_steps_short}"
+    )
+    if plot:
+        plot_processed_results(
+            coalescence_only_processed_scaled,
+            plot_title=f"coalescence-only with scaling (n_steps: {n_steps_short})",
+            plot_filename=filename + ".svg",
+        )
+    if save is not None:
+        write_to_file(filename=filename + ".txt", d=coalescence_only_processed_scaled)
+
+    # ### Breakup-only
+
+    res_breakup_only_scaled = go_benchmark(
+        setup_breakup_only_sim,
+        n_sds,
+        n_steps_short,
+        seeds,
+        numba_n_threads=numba_n_threads,
+        double_precision=True,
+        sim_run_filename=SIM_RUN_FNAME + "-breakup-scaled",
+        total_number=total_number_from_n_sd,
+        dv=dv_from_n_sd,
+        backends=[CPU, GPU],
+    )
+    breakup_only_processed_scaled = process_results(res_breakup_only_scaled)
+    filename = f"{SIM_RUN_FNAME}-results-scaled-breakup-double-n_steps{n_steps_short}"
+    if plot:
+        plot_processed_results(
+            breakup_only_processed_scaled,
+            plot_title=f"breakup-only with scaling (n_steps: {n_steps_short})",
+            plot_filename=filename + ".svg",
+        )
+    if save is not None:
+        write_to_file(filename=filename + ".txt", d=breakup_only_processed_scaled)
+
+    # ### Coalescence and Breakup
+
+    res_coal_breakup_scaled = go_benchmark(
+        setup_coalescence_breakup_sim,
+        n_sds,
+        n_steps_full,
+        seeds,
+        numba_n_threads=numba_n_threads,
+        double_precision=True,
+        sim_run_filename=SIM_RUN_FNAME + "-coal-break-scaled",
+        total_number=total_number_from_n_sd,
+        dv=dv_from_n_sd,
+        backends=[CPU, GPU],
+    )
+    coal_breakup_processed_scaled = process_results(res_coal_breakup_scaled)
+    filename = (
+        f"{SIM_RUN_FNAME}-results-scaled-coal_with_breakup-double-n_steps{n_steps_full}"
+    )
+    if plot:
+        plot_processed_results(
+            coal_breakup_processed_scaled,
+            plot_title=f"coalescence+breakup with scaling (n_steps: {n_steps_full})",
+            plot_filename=filename + ".svg",
+        )
+    if save is not None:
+        write_to_file(filename=filename + ".txt", d=coal_breakup_processed_scaled)
+
+
+if __name__ == "__main__":
+    main(plot="CI" not in os.environ, save=".")

PySDM_examples/Bulenok_2023_MasterThesis/utils.py

@@ -6,7 +6,6 @@ import time
 import numba
 import numpy as np
 from matplotlib import pyplot
-from open_atmos_jupyter_utils import show_plot
 
 from PySDM.backends import CPU, GPU
 
@@ -208,11 +207,11 @@ class PlottingHelpers:
 
 def plot_processed_results(
     processed_d,
-    show=True,
+    *,
     plot_label="",
     plot_title=None,
     metric="min",
-    plot_filename=None,
+    plot_filename,
     markers=None,
     colors=None,
 ):
@@ -257,33 +256,27 @@ def plot_processed_results(
     if plot_title:
         pyplot.title(plot_title)
 
-    if show:
-        if plot_filename:
-            show_plot(filename=plot_filename)
-        else:
-            pyplot.show()
+    pyplot.savefig(plot_filename)
 
 
 def plot_processed_on_same_plot(coal_d, break_d, coal_break_d):
-    plot_processed_results(coal_d, plot_label="-c", show=False)
-    plot_processed_results(break_d, plot_label="-b", show=False)
-    plot_processed_results(coal_break_d, plot_label="-cb", show=False)
-
-    show_plot()
+    filename = "same_plot.svg"
+    plot_processed_results(coal_d, plot_label="-c", plot_filename=filename)
+    plot_processed_results(break_d, plot_label="-b", plot_filename=filename)
+    plot_processed_results(coal_break_d, plot_label="-cb", plot_filename=filename)
 
 
 def plot_time_per_step(
     processed_d,
     n_sd,
-    show=True,
+    *,
     plot_label="",
     plot_title=None,
     metric="mean",
-    plot_filename=None,
+    plot_filename,
     step_from_to=None,
 ):
     backends = PlottingHelpers.get_sorted_backend_list(processed_d)
-
     markers = PlottingHelpers.get_backend_markers(backends)
 
     for backend in backends:
@@ -305,8 +298,4 @@ def plot_time_per_step(
     if plot_title:
         pyplot.title(plot_title + f"(n_sd: {n_sd})")
 
-    if show:
-        if plot_filename:
-            show_plot(filename=plot_filename)
-        else:
-            pyplot.show()
+    pyplot.savefig(plot_filename)

{PySDM_examples-2.91.dist-info → PySDM_examples-2.93.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PySDM-examples
-Version: 2.91
+Version: 2.93
 Summary: PySDM usage examples reproducing results from literature and depicting how to use PySDM from Python Jupyter notebooks
 Home-page: https://github.com/open-atmos/PySDM
 Author: https://github.com/open-atmos/PySDM/graphs/contributors

{PySDM_examples-2.91.dist-info → PySDM_examples-2.93.dist-info}/RECORD

@@ -43,9 +43,10 @@ PySDM_examples/Bieli_et_al_2022/settings.py,sha256=qJxoGChzqta-c_O2yv4E3wEM4yoEC
 PySDM_examples/Bieli_et_al_2022/simulation.py,sha256=4L0AxnpGjfWqIUgVdZdH1PgXzrx4yXEeJ8_WaWsksy8,1324
 PySDM_examples/Bolin_1958/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 PySDM_examples/Bolot_et_al_2013/__init__.py,sha256=wFQIlJaBVXaB5ji97EYjoFIv8fQFBXcPQYp2UDW0uEA,187
-PySDM_examples/Bulenok_2023_MasterThesis/__init__.py,sha256=gwWrz77ShGpXh0P1IKYin-U6vndeJtFkmYrRgdNLa0k,283
+PySDM_examples/Bulenok_2023_MasterThesis/__init__.py,sha256=GPHWvxMGulWIaSH7CQlESd2jNTRJeMcXRqjzzJ-AAf8,164
+PySDM_examples/Bulenok_2023_MasterThesis/performance_comparison_Srivastava_Setup.py,sha256=uPWdzK-9gYIr3LxMj-AIFB7PP4-zru3VnfJX2iSA4uo,7272
 PySDM_examples/Bulenok_2023_MasterThesis/setups.py,sha256=UVTp0oBMblcR46J29aylfZ6g6-8jlhXCb90ODdcfaSE,3424
-PySDM_examples/Bulenok_2023_MasterThesis/utils.py,sha256=Y0OXKd8D2vJ0vAstl_f68St39MRY89sMqaK1ASo-h1w,8019
+PySDM_examples/Bulenok_2023_MasterThesis/utils.py,sha256=T4UfPCPAaSal8paE4i1PTB4YSMEccqHMq-Z6ky9mABI,7813
 PySDM_examples/Ervens_and_Feingold_2012/__init__.py,sha256=F9Q-77npT6TyIK1vdNS4RPeBb8g4mFBUYz5qg4pj2wo,159
 PySDM_examples/Ervens_and_Feingold_2012/settings.py,sha256=DtkcDqkcO2WAkH63P1RQ19cejn0oAamdsZefN3y0Gy8,442
 PySDM_examples/Gedzelman_and_Arnold_1994/__init__.py,sha256=P721EPYBc_zDk2Jb2pPGvTaT76uGzvX6vbpG93hReMw,303
@@ -149,7 +150,7 @@ PySDM_examples/utils/read_vtk_1d.py,sha256=7_3sagEbX_FxqFuGkFWdwQapd1-9echFghU8T
 PySDM_examples/utils/widgets/__init__.py,sha256=U67WZ2YDcXSocMraI7pzcMxCfcxzXNqehDzoG6ogsh4,519
 PySDM_examples/utils/widgets/freezer.py,sha256=tFlZWShXUzA2X0cxgeE44N1g9kD33q3KQWZBqkgBP0A,407
 PySDM_examples/utils/widgets/progbar_updater.py,sha256=3uhG32QkAiSM3dTV7DoROJl7fAUNA6ZsJEIKSg3tdmk,267
-PySDM_examples-2.91.dist-info/METADATA,sha256=Q9ecoMOzinWG88eiZNVqhG6xdSEftvDGTvbZ5noExFM,12075
-PySDM_examples-2.91.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-PySDM_examples-2.91.dist-info/top_level.txt,sha256=axw2inuuYKPWkQhzBYtkZJTPDlo5LrpGLOA0iSAf9v0,15
-PySDM_examples-2.91.dist-info/RECORD,,
+PySDM_examples-2.93.dist-info/METADATA,sha256=BPeWDgz87uj448RWN9hN1j0M6JyAx0nRe_UmZ8t5g6E,12075
+PySDM_examples-2.93.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+PySDM_examples-2.93.dist-info/top_level.txt,sha256=axw2inuuYKPWkQhzBYtkZJTPDlo5LrpGLOA0iSAf9v0,15
+PySDM_examples-2.93.dist-info/RECORD,,