pytorch
33403f48 - edge_order check in torch.gradient only applies to dim argument (#67926)

Commit
2 years ago
edge_order check in torch.gradient only applies to dim argument (#67926) Summary: Fixes https://github.com/pytorch/pytorch/issues/67919 The compatibility check on `edge_order` in `pre_check_gradient` now looks only at dim argument if it is present, otherwise it checks all dimensions. Previously, it would check all dimensions regardless of the dim argument and throw unnecessary errors. Pull Request resolved: https://github.com/pytorch/pytorch/pull/67926 Reviewed By: albanD Differential Revision: D33760621 Pulled By: mruberry fbshipit-source-id: d490cd8610c68ff3787e670fc947de3cbf2db062 (cherry picked from commit 45bc56de9e287f715186378682e22bc6ac7a6ccc)
Author
Jonathan Colen
Committer
Parents
Loading