pytorch
161f7c0b - [spmd expansion] support torch.ops.aten.sym_numel (#98388)

Commit
1 year ago
[spmd expansion] support torch.ops.aten.sym_numel (#98388) The current logic assumes non-overload ops takes two arguments however torch.ops.aten.sym_numel takes one. Pull Request resolved: https://github.com/pytorch/pytorch/pull/98388 Approved by: https://github.com/mrshenli
Author
Committer
Parents
Loading