| .. |
|
_composable
|
[DCP][BE] Move DCP._state_dict_utils out from DCP (#115523)
|
2023-12-13 08:59:48 +00:00 |
|
_shard
|
Refactor some tests by using TEST_CUDA & TEST_MULTIGPU instead (#116083)
|
2024-01-03 08:53:59 +00:00 |
|
_spmd
|
|
|
|
_tensor
|
[dtensor] fix dtensor _to_copy op for mix precision (#116426)
|
2024-01-03 07:29:08 +00:00 |
|
_tools
|
|
|
|
algorithms
|
|
|
|
bin
|
|
|
|
checkpoint
|
[DCP][BC] Add the backward compatibility test (#116247)
|
2024-01-11 18:01:35 +00:00 |
|
elastic
|
[BE]: Enable F821 and fix bugs (#116579)
|
2024-01-01 08:40:46 +00:00 |
|
fsdp
|
[FSDP] sharded grad scaler: copy found_inf after waiting on async reduce_all (#115710)
|
2024-01-10 19:17:25 +00:00 |
|
launcher
|
|
|
|
nn/jit
|
|
|
|
optim
|
|
|
|
pipeline/sync
|
Refactor some tests by using TEST_CUDA & TEST_MULTIGPU instead (#116083)
|
2024-01-03 08:53:59 +00:00 |
|
rpc
|
[BE]: Enable F821 and fix bugs (#116579)
|
2024-01-01 08:40:46 +00:00 |
|
tensor/parallel
|
[tp] disable some assertion temporarily for torch.compile (#116573)
|
2024-01-03 23:01:19 +00:00 |
|
argparse_util_test.py
|
|
|
|
test_c10d_common.py
|
[c10d] Fix the hang issue in store.check(TIMEOUT_DUMP) (#116297)
|
2023-12-22 04:04:30 +00:00 |
|
test_c10d_functional_native.py
|
[AOTI] Add pybind for AOTIModelContainerRunnerCpu and AOTIModelContainerRunnerCuda (#116269)
|
2024-01-04 18:58:24 +00:00 |
|
test_c10d_gloo.py
|
|
|
|
test_c10d_logger.py
|
|
|
|
test_c10d_nccl.py
|
Refactor some tests by using TEST_CUDA & TEST_MULTIGPU instead (#116083)
|
2024-01-03 08:53:59 +00:00 |
|
test_c10d_object_collectives.py
|
|
|
|
test_c10d_pypg.py
|
|
|
|
test_c10d_spawn.py
|
[BE]: Update flake8 to v6.1.0 and fix lints (#116591)
|
2024-01-03 06:04:44 +00:00 |
|
test_c10d_spawn_gloo.py
|
|
|
|
test_c10d_spawn_nccl.py
|
|
|
|
test_c10d_spawn_ucc.py
|
|
|
|
test_c10d_ucc.py
|
|
|
|
test_collective_utils.py
|
|
|
|
test_compute_comm_reordering.py
|
|
|
|
test_data_parallel.py
|
|
|
|
test_device_mesh.py
|
Make input contiguous for DTensor reduce scatter to fix the incorrect numerical values (#115847)
|
2023-12-17 01:35:09 +00:00 |
|
test_distributed_spawn.py
|
|
|
|
test_dynamo_distributed.py
|
[dynamo / DDP] Add optimize_ddp_lazy_compile config to control lazy compile for DDPOptimizer (False by default) (#116292)
|
2023-12-21 22:34:24 +00:00 |
|
test_fake_pg.py
|
|
|
|
test_functional_api.py
|
[Distributed] Fix tests when CUDA not available (#117163)
|
2024-01-11 22:27:43 +00:00 |
|
test_inductor_collectives.py
|
[dynamo][collective_op] Use the value of the wrappered attribute async_op in dynamo when checking supported or not (#115921)
|
2023-12-21 03:27:57 +00:00 |
|
test_launcher.py
|
|
|
|
test_multi_threaded_pg.py
|
|
|
|
test_nccl.py
|
[BE]: Enable F821 and fix bugs (#116579)
|
2024-01-01 08:40:46 +00:00 |
|
test_pg_wrapper.py
|
|
|
|
test_store.py
|
[c10d] Add a recursive method to get the inner most store (#117074)
|
2024-01-10 20:22:55 +00:00 |
