saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test/distributed/tensor/parallel
History
wz337 c1dc8bb858 [DTensor] Turn on foreach implementation of optimizer for DTensor by default (#123394) 
Append DTensor to the optimizer `_foreach_supported_types` and turn on foreach implementation of optimizer for DTensor if not specified by the users.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/123394
Approved by: https://github.com/wanchaol
2024-05-15 16:45:42 +00:00
..
__init__.py
test_ddp_2d_parallel.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_fsdp_2d_parallel.py [Test][2D] Turn on 2D state_dict tests for uneven sharding (#124255) 2024-04-17 20:45:34 +00:00
test_parallelize_api.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_tp_examples.py [DTensor] Turn on foreach implementation of optimizer for DTensor by default (#123394) 2024-05-15 16:45:42 +00:00
test_tp_random_state.py Enable UFMT on all of test/distributed (#123539) 2024-04-17 06:46:02 +00:00
test_tp_style.py [tp] add some test for shard output layouts for rowwise parallel (#125713) 2024-05-08 03:45:34 +00:00