pytorch

6a4bf3b7 - feat(fx): `make_fx` should be aware of functions wrapped with `@fx.wrap` (#93273)

feat(fx): `make_fx` should be aware of functions wrapped with `@fx.wrap` (#93273) Fixes https://github.com/pytorch/pytorch/issues/89421 The strategy is to patch the given function wrapped with `@torch.fx.wrap` so that if a tensor tracer is active, we will `proxy_call` the function. `proxy_call` will also skip certain checks if the function to proxy call is not a torch op (checked with `isinstance(.., OpOverload)`. @IvanYashchuk @ezyang @Chillee Pull Request resolved: https://github.com/pytorch/pytorch/pull/93273 Approved by: https://github.com/ezyang