def METHOD_NAME():
    if paddle.distributed.get_world_size() == 1:
        return 1
    hcg = get_hcg()
    dp_size = hcg.get_data_parallel_world_size()
    sharding_size = hcg.get_sharding_parallel_world_size()
    return dp_size * sharding_size