Pritam Damania 96cedefd8e [Pipe] Refactor convert_to_balance under non-test package. (#50860) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/50860 Since fairscale.nn.Pipe still uses 'balance' and 'devices' parameters, other frameworks like fairseq still use these parameters. As a result, the `convert_to_balance` method is a nice utility to use for migrating to PyTorch Pipe without changing a lot of code in other frameworks. In addition to this I've renamed the method to be more illustrative of what it does and also allowed an optional devices parameter. ghstack-source-id: 120430775 Test Plan: 1) waitforbuildbot 2) Tested with fairseq Reviewed By: SciPioneer Differential Revision: D25987273 fbshipit-source-id: dccd42cf1a74b08c876090d3a10a94911cc46dd8		2021-01-28 12:10:21 -08:00
..
benchmark_dataset.py	Add benchmark for torch.distributed.pipeline.sync.Pipe (#49577)	2020-12-18 18:33:47 -08:00
pipe.py	[Pipe] Refactor convert_to_balance under non-test package. (#50860)	2021-01-28 12:10:21 -08:00