pytorch
9378a2ce - [dtensor] support aten.where and enable implicit scalar promotion (#110584)

Commit
1 year ago
[dtensor] support aten.where and enable implicit scalar promotion (#110584) This PR adds support for aten.where and support implicit scalar promotion, basically when we meet scalar tensors in dispatching logic, we implicitly convert it those to replicated dtensor The latter also enables bunch of ops in op db to pass Pull Request resolved: https://github.com/pytorch/pytorch/pull/110584 Approved by: https://github.com/fduwjj
Author
Committer
Parents
Loading