Pritam Damania 9d91360b5d Cleanup APIs for pipeline parallelism. (#48630) ...

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/48630

1) Make torch.distributed.pipeline package public.
2) Make several helper methods private.
ghstack-source-id: 118820803

Test Plan: waitforbuildbot

Reviewed By: rohan-varma

Differential Revision: D25235688

fbshipit-source-id: c32833ebf090ddbd4eaf06fcb5e3f9d421623a60

2020-12-18 15:17:13 -08:00

0 lines

Python

Raw Blame History