Commits on Jul 9, 2023

1. [fx] make fx.wrap idempotent …

   Previously, if you called `torch.fx.wrap()` on the same thing twice, it
   would add two entries to `_wrapped_fns_to_patch`. Then, when tracing,
   the patcher would process them both. On the second entry, the patcher
   would double-wrap the fn (e.g. `wrap(wrap(orig_fn))`)
   
   This makes it so that wrapping is observable after the trace. While
   normally, a Patcher instance will "revert" the wrapping after tracing,
   the double wrapped function goes from `wrap(wrap(orig_fn)) -> wrap(orig_fn)`.
   
   This happens to work in normal fx stuff (after all, the wrapper function
   will behave exactly like the original function). But it upsets
   torch.package, which is not expecting to see a weird wrapper function in
   the graph.

   

   suo committed Jul 9, 2023
   Configuration menu

   • View commit details
   • Copy full SHA for 9ffe96b
   • Browse repository at this point

   Copy the full SHA

   9ffe96b View commit details

   Browse the repository at this point in the history