pytorch
52fd7e49 - Update torch.ops.aten.all ref to be symbolic-trace friendly (#85352)

Commit
3 years ago
Update torch.ops.aten.all ref to be symbolic-trace friendly (#85352) - previous impl compared the summed bool values of the tensor to its nelem, which in a symbolic world is a symint and can't be coerced back into a bool for the purpose of shoving into a result tensor - new impl adds one extra negation op but avoids the need to compare to the symbolic nelem Pull Request resolved: https://github.com/pytorch/pytorch/pull/85352 Approved by: https://github.com/ezyang, https://github.com/mruberry
Author
Committer
Parents
Loading