pytorch

af62a70f - [export] Fix nn_module_stack in retracing (#121423)

[export] Fix nn_module_stack in retracing (#121423) Fixes https://fb.workplace.com/groups/1075192433118967/permalink/1391916691446538/ Pull Request resolved: https://github.com/pytorch/pytorch/pull/121423 Approved by: https://github.com/zhxchen17