Reland OpInfo support for forward AD (#58304)
Summary:
Try 3 to land this.
Trying ci-all label to ensure we test everything.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58304
Reviewed By: heitorschueroff
Differential Revision: D28474343
Pulled By: albanD
fbshipit-source-id: 8230fa3c0a8d3633f09999e7c2f47dbdc5fe57e9