pytorch
40aaacd4 - Respect sharded dimensions when aten expaned/view consumes SymInt values (#99058)

Commit
1 year ago
Respect sharded dimensions when aten expaned/view consumes SymInt values (#99058) Currently, aten.expand always expands to the global dimension. Then, it introduces additional slice and clone ops before running compute on the expanded tensor with a local tensor. In this commit, if we detect the op consumes a SymInt size, it respects both local size and the dimension placements from where the SymInt was extracted. Pull Request resolved: https://github.com/pytorch/pytorch/pull/99058 Approved by: https://github.com/wanchaol
Author
Committer
Parents
Loading