tree e95ec71c45c767158b9dee71c0ff3f29ef1eadcc
parent 03205ff3ba2c8392c19b26095b90e0c5d8d04e26
author Wanchao Liang <wanchaol@users.noreply.github.com> 1705995941 -0800
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> 1706072176 +0000

[dtensor] refactor partial redistribution logic (#113334)

This PR:

* Make the remaining placement transform to move from redistribute.py to
placement_types, specifically partial related logic
* redefine partial interface to make things more consistent, and add
  docs about the transformation relationships

Pull Request resolved: https://github.com/pytorch/pytorch/pull/113334
Approved by: https://github.com/tianyu-l, https://github.com/XilunWu
ghstack dependencies: #118078