pytorch/test/distributed
Wen Zhang b8f9751f11 Add cpu/gloo tests for sharded tensor distributed checkpoint (#80997)
SharedTensor checkpoint does not depend on NCCL, replicate the GPU cases for CPU and enable development on mac

Pull Request resolved: https://github.com/pytorch/pytorch/pull/80997
Approved by: https://github.com/wanchaol
2022-07-20 23:02:52 +00:00
..
_shard Add cpu/gloo tests for sharded tensor distributed checkpoint (#80997) 2022-07-20 23:02:52 +00:00
algorithms Enable test: distributed/algorithms/quantization/test_quantization (#80097) 2022-07-01 01:32:33 +00:00
bin
elastic
fsdp Adding fsdp fp16 and bf16 hooks (#81711) 2022-07-19 23:54:51 +00:00
launcher
nn/jit
optim
pipeline/sync
rpc
argparse_util_test.py
defs.bzl
test_c10d_common.py
test_c10d_gloo.py
test_c10d_nccl.py docs: Fix a few typos (#81435) 2022-07-14 04:20:26 +00:00
test_c10d_object_collectives.py
test_c10d_pypg.py [distributed] Make DDP work with python process group (#79176) 2022-06-28 17:14:21 +00:00
test_c10d_spawn.py
test_c10d_spawn_gloo.py
test_c10d_spawn_nccl.py
test_data_parallel.py
test_distributed_spawn.py
test_launcher.py
test_nccl.py
test_pg_wrapper.py Add _reduce_scatter_base to ProcessGroupWrapper. (#79633) 2022-06-29 15:32:42 +00:00
test_store.py