pytorch

9ee31202 - [Composite compliance testing] Refactor check_forward_ad_formula to accept Callable (#81239)

[Composite compliance testing] Refactor check_forward_ad_formula to accept Callable (#81239) Like https://github.com/pytorch/pytorch/pull/81059; this PR addresses the review comments. Test Plan: - run tests Pull Request resolved: https://github.com/pytorch/pytorch/pull/81239 Approved by: https://github.com/ezyang