def METHOD_NAME(self):
    # TODO
    try:
        tp_world_size = mpu.get_tensor_model_parallel_world_size()
        if tp_world_size == 1:
            return ''
        mp_rank = mpu.get_tensor_model_parallel_rank()
        return '_mp_rank_{:02d}'.format(mp_rank)
    except (ImportError, AssertionError):
        return ''