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
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
..
parallel [DTensor] Turn on foreach implementation of optimizer for DTensor by default (#123394) 2024-05-15 16:45:42 +00:00