Skip to content

Conversation

avikchaudhuri
Copy link
Contributor

@avikchaudhuri avikchaudhuri commented Dec 20, 2024

Stack from ghstack (oldest at bottom):

When we unflatten, the submodules we generate (InterpreterModule or InterpreterModuleDispatcher) are not related by type to the original submodules N. This makes isinstance(mod, N) checks fail. Since we do not have the original types after export, the best we can do is expose a type_name() method that carries the original type name, which we do carry in nn_module_stack entries.

Differential Revision: D67526542

cc @H-Huang @awgu @kwen2501 @wanchaol @fegin @fduwjj @wz337 @wconstab @d4l3k @c-p-i-o

Differential Revision: [D67526542](https://our.internmc.facebook.com/intern/diff/D67526542/)

[ghstack-poisoned]
Copy link

pytorch-bot bot commented Dec 20, 2024

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/143664

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit d88d39d with merge base 2c48af5 (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D67526542

avikchaudhuri added a commit that referenced this pull request Dec 20, 2024
Pull Request resolved: #143664

Differential Revision: [D67526542](https://our.internmc.facebook.com/intern/diff/D67526542/)
ghstack-source-id: 259149796
@pytorch-bot pytorch-bot bot added the oncall: distributed Add this issue/PR to distributed oncall triage queue label Dec 20, 2024
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D67526542

@pytorch-bot pytorch-bot bot added the ciflow/trunk Trigger trunk jobs on your pull request label Dec 20, 2024
@facebook-github-bot
Copy link
Contributor

@pytorchbot merge

(Initiating merge automatically since Phabricator Diff has merged)

@pytorchmergebot
Copy link
Collaborator

Merge started

Your change will be merged once all checks pass (ETA 0-4 Hours).

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging
Check the merge workflow status
here

pytorchmergebot pushed a commit that referenced this pull request Dec 21, 2024
Retracing while preserving module call signatures used to be a problem because graph modules don't have submodules at given paths. This led to a number of failing retracebility tests. By not trying to wrap modules with export tracepoints we can pass most of these tests; the only exception is where you do module swapping on retraced programs, which is still not possible.

Differential Revision: [D67539304](https://our.internmc.facebook.com/intern/diff/D67539304/)
Pull Request resolved: #143676
Approved by: https://github.com/zhxchen17, https://github.com/tugsbayasgalan
ghstack dependencies: #143664
@github-actions github-actions bot deleted the gh/avikchaudhuri/42/head branch January 20, 2025 02:06
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/trunk Trigger trunk jobs on your pull request fb-exported Merged oncall: distributed Add this issue/PR to distributed oncall triage queue release notes: export

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants