Added all_gather+a2a option