Function log_dist

deepspeed/utils/logging.py:113–123

Log message when get_dist_msg() deems it should be logged, see its docstring for details. Args: message (str) ranks (list) level (int)

(message, ranks=None, level=logging.INFO)

def log_dist(message, ranks=None, level=logging.INFO):
    """Log message when get_dist_msg() deems it should be logged, see its docstring for details.

    Args:
        message (str)
        ranks (list)
        level (int)
    """
    final_message = get_dist_msg(message, ranks)
    if final_message is not None:
        logger.log(level, final_message)


def print_dist(message, ranks=None):