tree 8e8cfb122bd5e9da6617473c11ab7f665ff41ec6
parent a0df40f195b3b3ad559dfa8e31ee7066013ef550
author Tianyu Liu <lty@fb.com> 1715890382 -0700
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> 1715924361 +0000

[dtensor] refactor view ops to use OpStrategy (#126011)

As titled. Some ops require adjustment of output shape argument. In rule-based sharding prop, global output shape was inferred in the rule (in `view_ops.py`). In strategy-based sharding prop, it is now obtained from propagated out_tensor_meta (in `sharding_prop.py`).

Pull Request resolved: https://github.com/pytorch/pytorch/pull/126011
Approved by: https://github.com/wanchaol, https://github.com/XilunWu