def METHOD_NAME(self, message):
    if dist.get_rank() == 0:
        element_size = torch.tensor([], dtype=self.dtype).element_size()
        swapped_GB = (self.num_elements_swapped * element_size) / (1024**3)
        logger.debug(f'{message} num_elems = {self.num_elements_swapped}, {swapped_GB:5.2f} GB')