pytorch

a402e100 - [fx] Make wrapped_fn also work for non-mutating passes. (#84232)

[fx] Make wrapped_fn also work for non-mutating passes. (#84232) Summary: Before the change, wrapped_fn should only take mutating passes, but we don't actually have any way to detect whether a pass is mutating before running it. To make this an abstraction without involving any precondition depending on PassManager run, we could just relax the precondition to take any kind of passes, and conditionally return the original pass based on the pass result. Test Plan: eyes Reviewed By: qihqi, angelayi Differential Revision: D39086343 Pull Request resolved: https://github.com/pytorch/pytorch/pull/84232 Approved by: https://github.com/angelayi