pytorch/torch/distributed/_tensor/examples
Aaron Gokaslan 1dd42e42c4 [BE]: Try TCH autofixes on torch/ (#125536)
Tries TCH autofixes and see what breaks

Pull Request resolved: https://github.com/pytorch/pytorch/pull/125536
Approved by: https://github.com/ezyang
2024-05-05 23:13:59 +00:00
..
checkpoint_example.py [dtensor] change distribute_module input/output_fn to accept module (#120895) 2024-03-04 07:22:32 +00:00
convnext_example.py
torchrec_sharding_example.py [BE]: Try TCH autofixes on torch/ (#125536) 2024-05-05 23:13:59 +00:00
visualize_sharding_example.py [dtensor][debug] have visualize_sharding correctly print for sub-mesh DTensor (#121216) 2024-03-07 04:50:15 +00:00