OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 37 Packages Projects Releases Wiki Activity
edcd968b51
pytorch/test/distributed
History
Wanchao Liang 05addd5658 [tp] add kwargs support to prepare_module_input (#124114) 
as titled, this PR adds kwargs support to PrepareModuleInput style,
where there might be modules who have only kwargs inputs but no
positional args, so we should support this

Pull Request resolved: https://github.com/pytorch/pytorch/pull/124114
Approved by: https://github.com/XilunWu
2024-04-22 21:46:31 +00:00
..
_composable [BE]: FURB142 - Remove set mutations. Use set update (#124551) 2024-04-21 14:12:33 +00:00
_shard Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
_spmd Update to TorchFix 0.4.0 (#119424) 2024-02-12 23:30:12 +00:00
_tensor distributed: templated ring attention (#124215) 2024-04-19 00:57:08 +00:00
_tools Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
algorithms Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
bin
checkpoint [BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261) 2024-04-17 19:29:34 +00:00
elastic [BE]: TRY002 - Ban raising vanilla exceptions (#124570) 2024-04-21 22:26:40 +00:00
fsdp [BE]: FURB142 - Remove set mutations. Use set update (#124551) 2024-04-21 14:12:33 +00:00
launcher Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
nn/jit Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
optim Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
pipeline/sync [BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261) 2024-04-17 19:29:34 +00:00
rpc Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
tensor/parallel [tp] add kwargs support to prepare_module_input (#124114) 2024-04-22 21:46:31 +00:00
argparse_util_test.py
test_c10d_common.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_c10d_functional_native.py [functional collective] change the Python APIs to only use the native funcol ops (#123777) 2024-04-13 03:08:36 +00:00
test_c10d_gloo.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_c10d_logger.py [c10d] add nccl version to c10d logger (#111215) 2023-10-16 18:47:49 +00:00
test_c10d_nccl.py [NCCL][TEST] Synchronize proper devices (#124517) 2024-04-20 23:42:32 +00:00
test_c10d_object_collectives.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_c10d_pypg.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_c10d_spawn_gloo.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_c10d_spawn_nccl.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_c10d_spawn_ucc.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_c10d_spawn.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_c10d_ucc.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_collective_utils.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_compute_comm_reordering.py [reland] Fix estimate_nccl_collective_runtime (#118986) 2024-02-12 18:48:06 +00:00
test_data_parallel.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_device_mesh.py [DeviceMesh][Test] Add 3d unit test for get_local_rank() (#124142) 2024-04-18 23:19:17 +00:00
test_distributed_spawn.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_dynamo_distributed.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_fake_pg.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_functional_api.py distributed: templated ring attention (#124215) 2024-04-19 00:57:08 +00:00
test_inductor_collectives.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_launcher.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_multi_threaded_pg.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_nccl.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_pg_wrapper.py [torch/distributed] Check gloo availability when doing isinstance(pg,… (#124233) 2024-04-19 04:07:00 +00:00
test_store.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00