PyTorch Developer Podcast
PyTorch Developer Podcast
Edward Yang, Team PyTorch
Compiler collectives
16 minutes Posted Aug 4, 2024 at 1:47 am.
0:00
16:33
Download MP3
Show notes
Compiler collectives are a PT2 feature where by compiler instances across multiple ranks use NCCL collectives to communicate information to other instances. This is used to ensure we consistently decide if inputs or static or dynamic across all ranks. See also PR at https://github.com/pytorch/pytorch/pull/130935