Source code for scimba_torch.utils.verbosity

"""Verbosity utilities for scimba_torch."""

import torch

from scimba_common.version import __version__

SCIMBA_IS_VERBOSE = False




def print_torch_setting() -> None:
    """Print torch device and default floating point format."""
    print(f"torch device: {torch.get_default_device()}")
    print(f"torch floating point format: {torch.get_default_dtype()}")

    if torch.cuda.is_available():
        print(f"cuda devices:        {torch.cuda.device_count()}")
        print(f"cuda current device: {torch.cuda.current_device()}")
        print(f"cuda device name:    {torch.cuda.get_device_name(0)}")





def get_verbosity() -> bool:
    """Get the verbosity level of scimba.

    Returns:
        the current verbosity.
    """
    return SCIMBA_IS_VERBOSE





def set_verbosity(verbose: bool) -> None:
    """Set the verbosity level of scimba.

    Args:
        verbose: the wanted verbosity
    """
    global SCIMBA_IS_VERBOSE
    SCIMBA_IS_VERBOSE = verbose

    # DEPRECATED: this was to avoid a circular import which does no happen
    # now that __version__ has been moved in scimba_common/version
    # init_path = Path(__file__).parent.parent.parent / "scimba_common/version.py"
    # version = _get_version_from_init(init_path)

    if SCIMBA_IS_VERBOSE:
        print(f"\n/////////////// Scimba {__version__} ////////////////")
        print_torch_setting()
        print("\n")