← All questions
System DesignMedium· Tensor Parallelism· Communication· Performance

What determines the AllReduce size in tensor parallelism?

In tensor parallelism, the communication overhead is driven by the activation tensors passed between layers. Because the size of these activations is a function of sequence length, batch size, and hidden dimension, the communication traffic scales with these factors rather than the total number of parameters in the model depth.

Code playground (Python)