pytorch
24c3ad78 - Move private forward grad mode helpers to torch.autograd.forward_ad (#90240)

Commit
2 years ago
Move private forward grad mode helpers to torch.autograd.forward_ad (#90240) Motivation - These were previously defined in functorch. They are not functorch-specific, so I'm moving them to torch.autograd.forward_ad and the autograd python bindings. - I need this to avoid some of my cyclic import problems. Should these be public APIs? Probably. Though this needs discussion, so punting it to the future. Test Plan: - moved the tests of these from test/functorch/test_eager_transforms.py to test/test_autograd.py Pull Request resolved: https://github.com/pytorch/pytorch/pull/90240 Approved by: https://github.com/soulitzer
Author
Committer
Parents
Loading