pytorch/test/distributed
Pritam Damania 87bbcf70f7 Create torch.distributed.shard package. (#71742)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/71742

We have many sharding components currently:
torch.distributed._sharded_tensor, torch.distributed._sharding_spec,
torch.distributed._sharded_optimizer and more coming.

As a result, organizing all of this under the `torch.distributed.shard`
package. For BC reasons, I'm still keeping the old packages and have them just
reference the new package.
ghstack-source-id: 147899768

Test Plan: waitforbuildbot

Reviewed By: fduwjj, wanchaol

Differential Revision: D33755913

fbshipit-source-id: dc692b31e2607063d55dfcb3db33ec53961d5a5b
(cherry picked from commit 5b6885f3587786217f8ce143f2329ceec618404e)
2022-01-29 00:48:06 +00:00
..
algorithms [BE] move init_multigpu_helper to common_distributed (#67050) 2021-10-22 17:16:11 -07:00
bin Add test owner to distributed files starting with test_ (#66797) 2021-10-19 10:55:20 -07:00
elastic Revise the socket implementation of c10d (#68226) 2021-11-16 20:49:25 -08:00
fsdp [FSDP] Backward prefetch in recursive call (#71804) 2022-01-28 00:34:08 +00:00
launcher [torchelastic][1/n] Fix caffe2.test.distributed.launcher.api_test flaky tests (#68624) 2021-11-19 15:23:30 -08:00
nn/jit Have test classes extend from common_utils.TestCase, not unittest.TestCase (#66900) 2021-10-19 16:54:05 -07:00
optim Add the maximize flag to AdamW (#70146) 2021-12-23 09:20:29 -08:00
pipeline/sync [skip ci] set more tests with owners for distributed and elastic (#67583) 2021-11-01 12:26:03 -07:00
rpc Add test owner to distributed files starting with test_ (#66797) 2021-10-19 10:55:20 -07:00
shard Create torch.distributed.shard package. (#71742) 2022-01-29 00:48:06 +00:00
argparse_util_test.py [skip ci] set more tests with owners for distributed and elastic (#67583) 2021-11-01 12:26:03 -07:00
test_c10d_common.py [BE] rename some tests in test_c10d_common (#67828) 2021-11-18 17:14:58 -08:00
test_c10d_gloo.py [PT1.11] make static graph to be stable (#71459) 2022-01-20 19:38:41 +00:00
test_c10d_nccl.py Implement scatter primitive for ProcessGroupNCCL (#70029) 2022-01-27 19:37:55 +00:00
test_c10d_spawn.py [PyTorch][Distributed] Enable Reduce Scatter and modify all_to_all for sharded linear with more test cases. (#68786) 2021-12-06 13:38:58 -08:00
test_c10d_spawn_gloo.py [PyTorch][Distributed] Enable Reduce Scatter and modify all_to_all for sharded linear with more test cases. (#68786) 2021-12-06 13:38:58 -08:00
test_c10d_spawn_nccl.py [PyTorch][Distributed] Enable Reduce Scatter and modify all_to_all for sharded linear with more test cases. (#68786) 2021-12-06 13:38:58 -08:00
test_data_parallel.py Retire repeat_test_for_types (#71033) 2022-01-10 09:13:54 -08:00
test_distributed_spawn.py Add test owner to distributed files starting with test_ (#66797) 2021-10-19 10:55:20 -07:00
test_launcher.py Add test owner to distributed files starting with test_ (#66797) 2021-10-19 10:55:20 -07:00
test_nccl.py [NCCL] Patch bfloat16 support (#67843) 2021-11-09 13:46:13 -08:00
test_pg_wrapper.py Add test owner to distributed files starting with test_ (#66797) 2021-10-19 10:55:20 -07:00
test_store.py Add support for deleteKey for FileStore (#69953) 2022-01-07 06:20:59 -08:00