fullwave25 1.2.6.dev6__tar.gz → 1.2.6.dev7__tar.gz

{fullwave25-1.2.6.dev6 → fullwave25-1.2.6.dev7}/.bumpversion.toml

@@ -1,5 +1,5 @@
 [tool.bumpversion]
-current_version = "1.2.6-dev6"
+current_version = "1.2.6-dev7"
 parse = """(?x)
     (?P<major>0|[1-9]\\d*)\\.
     (?P<minor>0|[1-9]\\d*)\\.

{fullwave25-1.2.6.dev6 → fullwave25-1.2.6.dev7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fullwave25
-Version: 1.2.6.dev6
+Version: 1.2.6.dev7
 Summary: Fullwave 2.5: Ultrasound wave propagation simulation with heterogeneous power law attenuation modelling capabilities
 Author: Masashi Sode, Gianmarco Pinton
 Maintainer: Masashi Sode

{fullwave25-1.2.6.dev6 → fullwave25-1.2.6.dev7}/fullwave/__init__.py

@@ -60,7 +60,7 @@ try:
     __version__ = version("fullwave")
 except PackageNotFoundError:
     # Update via bump-my-version, not manually
-    __version__ = "1.2.6-dev6"
+    __version__ = "1.2.6-dev7"
 
 VERSION = __version__  # for convenience
 logger.info("Fullwave version: %s", __version__)

{fullwave25-1.2.6.dev6 → fullwave25-1.2.6.dev7}/fullwave/solver/solver.py

@@ -1,5 +1,6 @@
 """solver module."""
 
+import gc
 import logging
 import time
 from pathlib import Path
@@ -594,6 +595,24 @@ class Solver:
                 use_gpu=use_gpu_pml,
             )
 
+    @staticmethod
+    def _release_gpu_memory_pools() -> None:
+        """Release all CuPy GPU memory pool blocks back to CUDA.
+
+        Call ``gc.collect()`` first so that Python releases references to
+        CuPy arrays, then drain both the device and pinned memory pools.
+        This prevents stale allocations from causing memory pressure when
+        subsequent operations allocate large GPU arrays.
+        """
+        gc.collect()
+        try:
+            import cupy as cp  # noqa: PLC0415
+
+            cp.get_default_memory_pool().free_all_blocks()
+            cp.get_default_pinned_memory_pool().free_all_blocks()
+        except ImportError:
+            pass
+
     @staticmethod
     def _check_input(
         grid: fullwave.Grid,
@@ -966,6 +985,7 @@ class Solver:
                 "Input data generation completed in %s. Skipping simulation execution.",
                 simulation_dir,
             )
+            self._release_gpu_memory_pools()
             return simulation_dir
 
         sim_result = self.fullwave_launcher.run(

{fullwave25-1.2.6.dev6 → fullwave25-1.2.6.dev7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fullwave25"
-version = "1.2.6-dev6" # Update via bump-my-version, not manually
+version = "1.2.6-dev7" # Update via bump-my-version, not manually
 description = "Fullwave 2.5: Ultrasound wave propagation simulation with heterogeneous power law attenuation modelling capabilities"
 readme = "README.md"
 requires-python = ">=3.10"

{fullwave25-1.2.6.dev6 → fullwave25-1.2.6.dev7}/uv.lock

@@ -735,7 +735,7 @@ wheels = [
 
 [[package]]
 name = "fullwave25"
-version = "1.2.6.dev6"
+version = "1.2.6.dev7"
 source = { editable = "." }
 dependencies = [
     { name = "matplotlib" },