pytorch/test/distributed
2022-06-21 16:01:59 +00:00
..
_shard [shard] make state_dict hook be consistent 2022-06-17 22:08:06 +00:00
algorithms
bin
elastic [ci] remove IN_CI env var 2022-06-11 17:16:30 +00:00
fsdp [CheckpointWrapper] Replace generic mod prefix (#79830) 2022-06-21 16:01:59 +00:00
launcher
nn/jit
optim [CUDA graphs] Allows Adam and AdamW to be capture-safe (#77862) 2022-06-13 01:56:47 +00:00
pipeline/sync Add all bzl files per D36874458 2022-06-06 09:40:19 -07:00
rpc [ci] remove IN_CI env var 2022-06-11 17:16:30 +00:00
argparse_util_test.py
defs.bzl Add all bzl files per D36874458 2022-06-06 09:40:19 -07:00
test_c10d_common.py
test_c10d_gloo.py ROCm: unskip c10 gloo tests 2022-04-25 14:28:56 +00:00
test_c10d_nccl.py [DDP] Fix broadcast for channels-last tensors (#79060) 2022-06-08 21:52:58 +00:00
test_c10d_object_collectives.py Revert "Revert "[distributed] Handle object collectives and NCCL. (#79034)"" 2022-06-15 10:04:37 -07:00
test_c10d_spawn.py
test_c10d_spawn_gloo.py
test_c10d_spawn_nccl.py Ensure tensors are contiguous in functional all_gather. 2022-06-17 01:27:11 +00:00
test_data_parallel.py
test_distributed_spawn.py
test_launcher.py
test_nccl.py
test_pg_wrapper.py
test_store.py [Bootcamp] Set default value of TCPStore world_size to None in pybind definition (#77277) 2022-05-12 18:48:48 +00:00