fix the allreduce size count