Pritam Damania 0d6fa1adc5 Introduce ChunkShardingSpec as a model sharding specification. (#55728) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55728 Full design: https://github.com/pytorch/pytorch/issues/55207 This PR introduces ChunkShardingSpec (SingleShardingSpec in the design). Used the name ChunkShardingSpec since it is very similar to `torch.chunk` in terms of how a Tensor is split up and feels more clear compared to SingleShardingSpec. ghstack-source-id: 129603318 Test Plan: waitforbuildbot Reviewed By: SciPioneer Differential Revision: D27694108 fbshipit-source-id: c8764abe6a4d5fc56d023fda29b74b5af2a73b49		2021-05-23 16:04:57 -07:00
..
__init__.py
remote_module.py	Introduce ChunkShardingSpec as a model sharding specification. (#55728)	2021-05-23 16:04:57 -07:00