-
Notifications
You must be signed in to change notification settings - Fork 25.7k
Description
distributed_c10d has a nice doc page here, and the source code of that live in here. We should add a similar page for rpc and distributed autograd to describe the API.
@rohan-varma brought up a point that whether we want to add the new docs into the existing distributed.rst file as distributed/distributed_c10d.py, distributed/rpc/ and distributed/autograd all live in the same distributed/ folder, or whether we want to keep them in separate files. I would vote for keeping them separate from the existing one, as the existing one mostly focus on c10d, and the new ones are on rpc/autograd, which are quite different topics. The following items would need more discussion and let's track them here:
- The title of the existing
distributed.rstpage says Distributed communication package - torch.distributed, do we want to rename that to Distributed collective communication package - torch.distributed to be more precise? - Should rpc and distributed autograd api docs live in the same file?
cc @pietern @mrshenli @pritamdamania87 @zhaojuanmao @satgera @gqchen @aazzolini @rohan-varma @xush6528