4 GPUs, 2x speed: why PyTorch DDP stalled on a PCIe-only 4x RTX A6000, and two flags that took it to 3.6x
All-reduce at 4.3 GB/s through the host bridge halved our scaling. bf16 gradient compression plus gradient accumulation got most of it back.
All-reduce at 4.3 GB/s through the host bridge halved our scaling. bf16 gradient compression plus gradient accumulation got most of it back.
Same API, same image, same script: 186 seconds on one GPU type, 707 on another. Where the time goes, and why the newest PyTorch broke on a V100.
Same model, same benchmark, three rented GPUs: $0.12 to $0.22 per 1M output tokens when the batch is full, $4.86 to $9.47 when it is not.
527 chunks, the same model both ways, real bills. The API won for our docs; the GPU wins past about 630k tokens.