Why it matters

Collective overlap hides comms latency. Standard in training frameworks.

Advertisement

The architecture

Async streams.

Bucket gradients.

Overlap comms + compute.

Collective overlapAsync streamsCUDA parallelBucket gradientsfor larger msgsOverlapcomms + computePyTorch DDP + gradient bucketing overlaps by default; big speedup
Collective overlap.
Advertisement

How it works end to end

PyTorch DDP bucketing.

DeepSpeed overlap.

Megatron optimize.