def METHOD_NAME(self):
    cuda_to_cpu_elem_num = 0
    cpu_to_cuda_elem_num = 0
    for cached_embedding_bag in self.cached_embedding_bag_list:
        cuda_to_cpu_elem_num += cached_embedding_bag.cache_weight_mgr._cuda_to_cpu_numel
        cpu_to_cuda_elem_num += cached_embedding_bag.cache_weight_mgr._cpu_to_cuda_numel
    print(f"CUDA->CPU num: {cuda_to_cpu_elem_num / 1e6} M elem")
    print(f"CPU->CUDA num: {cpu_to_cuda_elem_num / 1e6} M elem")