Update AllReduce interface