Fix all_reduce.