pytorch
b651e060 - Add Pointwise Tag from pointwise set in DTensor, use in aot_autograd partitioner (#90029)

Commit
3 years ago
Add Pointwise Tag from pointwise set in DTensor, use in aot_autograd partitioner (#90029) Takes the pointwise op list from [DTensor](https://github.com/pytorch/pytorch/blob/master/torch/distributed/_tensor/ops/pointwise_ops.py#L36) as an initially starting point for pointwise ops, and feeds them to the aot autograd partitioner. Pull Request resolved: https://github.com/pytorch/pytorch/pull/90029 Approved by: https://github.com/ezyang
Author
Committer
Parents
Loading