Add all_reduce #2