3 Commits

Author	SHA1	Message	Date
Shangdi Yu	83d88d128d	[reland] Make export._trace._WrapperModule work in strict mode (#146919) (#151264) ... Summary: as title `export._trace._WrapperModule` is used to wrap functions into a Module so we can export the function. We add `export._wrapper_utils` to `dynamo`'s `MOD_INLINELIST` so dynamo traces into `_WrapperModule` Fixes https://github.com/pytorch/pytorch/issues/146867 Test Plan: ``` buck run fbcode//mode/dev-nosan //caffe2/test:test_export -- -r wrapper_module ``` Differential Revision: D72986826 Pull Request resolved: https://github.com/pytorch/pytorch/pull/151264 Approved by: https://github.com/angelayi	2025-04-15 18:35:34 +00:00
PyTorch MergeBot	2f899f07aa	Revert "Make export._trace._WrapperModule work in strict mode (#146919)" ... This reverts commit dad5e5e2622c82ca272290225abe16ee461d9ac9. Reverted https://github.com/pytorch/pytorch/pull/146919 on behalf of https://github.com/malfet due to Broke lint, see https://github.com/pytorch/pytorch/actions/runs/14415686353/job/40431799827 ([comment](https://github.com/pytorch/pytorch/pull/146919#issuecomment-2798446930))	2025-04-12 04:12:36 +00:00
Shangdi Yu	dad5e5e262	Make export._trace._WrapperModule work in strict mode (#146919) ... Summary: as title `export._trace._WrapperModule` is used to wrap functions into a Module so we can export the function. We add `export._wrapper_utils` to `dynamo`'s `MOD_INLINELIST` so dynamo traces into `_WrapperModule` Fixes https://github.com/pytorch/pytorch/issues/146867 Test Plan: ``` buck run fbcode//mode/dev-nosan //caffe2/test:test_export -- -r wrapper_module ``` Differential Revision: D69434316 Pull Request resolved: https://github.com/pytorch/pytorch/pull/146919 Approved by: https://github.com/angelayi	2025-04-12 03:22:08 +00:00