Make jit.trace_module reentrant (#29411)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/29411
Fixes https://github.com/pytorch/pytorch/issues/29367
Test Plan: Imported from OSS
Differential Revision: D18380559
Pulled By: jamesr66a
fbshipit-source-id: 5caf606ccbc5dc79dac14e3c28cc02dec19ce695