All-Reduce

AI and Machine Learning · Distributed Training · all-reduce.yaml

A collective that sums a tensor across every participant and returns the result to all of them. Ring and tree schedules make its cost depend on interconnect bandwidth rather than device count.

Colour is the family; a dashed line is the second member of it.

Drag to pan · scroll to zoom · click a node to open it
G all-reduce All-Reduce data-parallelism Data Parallelism data-parallelism->all-reduce gradients must be summed across replicas every step

Referenced by

References