fullwave25 1.3.2.dev2__tar.gz → 1.3.2.dev4__tar.gz

{fullwave25-1.3.2.dev2 → fullwave25-1.3.2.dev4}/.bumpversion.toml

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

{fullwave25-1.3.2.dev2 → fullwave25-1.3.2.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: fullwave25
-Version: 1.3.2.dev2
+Version: 1.3.2.dev4
 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.3.2.dev2 → fullwave25-1.3.2.dev4}/fullwave/__init__.py

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

{fullwave25-1.3.2.dev2 → fullwave25-1.3.2.dev4}/fullwave/solver/solver.py

@@ -42,6 +42,7 @@ COMPATIBLE_CUDA_ARCHITECTURES = [
     "sm_90",  # Hopper: H100, H200
     "sm_100",  # Blackwell: RTX 50 series
     "sm_101",  # Blackwell: RTX 50 series
+    "sm_103",  # Blackwell
     "sm_120",  # Blackwell: RTX 50 series
     "sm_121",  # Blackwell: RTX 50 series
 ]
@@ -56,18 +57,6 @@ VERIFIED_CUDA_ARCHITECTURES = [
 ]
 
 
-COMPATIBLE_CUDA_VERSIONS = [
-    11.8,
-    12.4,
-    12.9,
-    13.0,
-    13.1,
-]
-
-COMPATIBLE_CUDA_RANGES = [
-    (11.8, 13.1),
-]
-
 VERIFIED_CUDA_VERSIONS = [
     12.4,
     12.9,
@@ -108,6 +97,7 @@ COMPATIBLE_CUDA_VERSIONS_ARCHITECTURES_set = {
     (12.9, "sm_90"),
     (12.9, "sm_100"),
     (12.9, "sm_101"),
+    (12.9, "sm_103"),
     (12.9, "sm_120"),
     (12.9, "sm_121"),
     # ---
@@ -149,7 +139,36 @@ def _make_cuda_arch_option(*, use_gpu: bool = True) -> str:
     return arch_option
 
 
-def _make_cuda_version_option(*, use_gpu: bool = True) -> tuple[str, float]:
+def _make_cuda_version_option(*, cuda_arch: str, use_gpu: bool = True) -> tuple[str, float]:
+    """Return the CUDA tag of the binary to run on this driver and architecture.
+
+    The driver reports the newest CUDA version it supports, and it also runs a binary
+    built with an older CUDA version. The choice is therefore the newest CUDA version
+    with a build for the architecture that is not newer than the driver. Where no such
+    build exists, the choice is the oldest build of the driver's CUDA major version,
+    which runs through CUDA minor version compatibility. A Blackwell GPU on a CUDA 12.8
+    driver takes this path, because its first build is CUDA 12.9.
+
+    Parameters
+    ----------
+    cuda_arch : str
+        The architecture option of the first GPU, such as "sm_90".
+    use_gpu : bool, optional
+        Whether the simulation runs on the GPU. Without it, the driver version is
+        returned unchecked.
+
+    Returns
+    -------
+    tuple[str, float]
+        The tag the binary name ends with, such as "cuda129", and its CUDA version.
+
+    Raises
+    ------
+    ValueError
+        If the driver version cannot be read, or the architecture has no build as
+        old as the driver and none of the driver's CUDA major version.
+
+    """
     cuda_version: float = retrieve_cuda_version()
     if use_gpu and cuda_version == -1:
         error_msg = (
@@ -159,34 +178,38 @@ def _make_cuda_version_option(*, use_gpu: bool = True) -> tuple[str, float]:
         logger.error(error_msg)
         raise ValueError(error_msg)
 
-    # range check
-    if use_gpu and not any(start <= cuda_version <= end for start, end in COMPATIBLE_CUDA_RANGES):
-        error_msg = (
-            f"CUDA version {cuda_version} is not in the compatible ranges: "
-            f"{COMPATIBLE_CUDA_RANGES}. Please install a compatible CUDA version."
+    if use_gpu:
+        built_versions = sorted(
+            version
+            for version, arch in COMPATIBLE_CUDA_VERSIONS_ARCHITECTURES_set
+            if arch == cuda_arch
         )
-        logger.error(error_msg)
-        raise ValueError(error_msg)
-
-    # find the a compatible cuda version below the system cuda
-    if use_gpu and cuda_version not in COMPATIBLE_CUDA_VERSIONS:
-        compatible_versions_below = [v for v in COMPATIBLE_CUDA_VERSIONS if v < cuda_version]
-        if compatible_versions_below:
-            closest_version = max(compatible_versions_below)
-            message = (
-                f"Warning: CUDA version {cuda_version} is not in the compatible versions: "
-                f"{COMPATIBLE_CUDA_VERSIONS}. "
-                f"Using the closest compatible version {closest_version} instead."
-            )
-            logger.warning(message)
-            cuda_version = closest_version
+        older_versions = [version for version in built_versions if version <= cuda_version]
+        same_major_versions = [
+            version for version in built_versions if int(version) == int(cuda_version)
+        ]
+        if older_versions:
+            closest_version = older_versions[-1]
+        elif same_major_versions:
+            closest_version = same_major_versions[0]
         else:
             error_msg = (
-                f"No compatible CUDA versions found below {cuda_version}. "
-                "Please install a compatible CUDA version."
+                f"No binary for architecture {cuda_arch} runs on a driver that supports "
+                f"CUDA {cuda_version}. The binaries for this architecture are built with "
+                f"CUDA {built_versions}. Please update the NVIDIA driver."
             )
             logger.error(error_msg)
             raise ValueError(error_msg)
+        if closest_version != cuda_version:
+            message = (
+                f"Warning: CUDA version {cuda_version} of the driver has no binary for "
+                f"architecture {cuda_arch}. "
+                f"Using the closest compatible version {closest_version} instead."
+            )
+            if closest_version > cuda_version:
+                message += " It runs through CUDA minor version compatibility."
+            logger.warning(message)
+            cuda_version = closest_version
 
     if use_gpu and cuda_version not in VERIFIED_CUDA_VERSIONS:
         warning_msg = (
@@ -240,7 +263,10 @@ def _retrieve_fullwave_simulation_path(
     use_isotropic_relaxation: bool = True,
 ) -> Path:
     arch_option = _make_cuda_arch_option(use_gpu=use_gpu)
-    cuda_version_option, cuda_version = _make_cuda_version_option(use_gpu=use_gpu)
+    cuda_version_option, cuda_version = _make_cuda_version_option(
+        use_gpu=use_gpu,
+        cuda_arch=arch_option,
+    )
     if use_isotropic_relaxation is False:
         error_msg = (
             "Currently, only isotropic relaxation is supported. "
@@ -251,10 +277,13 @@ def _retrieve_fullwave_simulation_path(
     # isotropic_str = "_isotropic" if use_isotropic_relaxation else ""
     isotropic_str = ""
 
-    _check_compatible_set(
+    if use_gpu and not _check_compatible_set(
         cuda_version=cuda_version,
         cuda_arch=arch_option,
-    )
+    ):
+        error_msg = f"No binary is built with CUDA {cuda_version} for architecture {arch_option}."
+        logger.error(error_msg)
+        raise ValueError(error_msg)
     if use_exponential_attenuation:
         if is_3d and use_gpu:
             path_fullwave_simulation_bin = (

{fullwave25-1.3.2.dev2 → fullwave25-1.3.2.dev4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fullwave25"
-version = "1.3.2-dev2" # Update via bump-my-version, not manually
+version = "1.3.2-dev4" # 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.3.2.dev2 → fullwave25-1.3.2.dev4}/uv.lock

@@ -761,7 +761,7 @@ wheels = [
 
 [[package]]
 name = "fullwave25"
-version = "1.3.2.dev2"
+version = "1.3.2.dev4"
 source = { editable = "." }
 dependencies = [
     { name = "cupy-cuda12x" },