[FX] Fix retracing wrapped functions (#58061)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58061
Test Plan: Imported from OSS
Reviewed By: yuhc
Differential Revision: D28358801
Pulled By: jamesr66a
fbshipit-source-id: c7c9a8a80e5bfe1eb1f6d2cf858ac7e57153a860
