-
Notifications
You must be signed in to change notification settings - Fork 21.7k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Fix issue 112919 #124746
Fix issue 112919 #124746
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hi! Thanks for working on this! Could you please add a test to test/dynamo/test_export.py to show that your fix works?
Looks like this PR hasn't been updated in a while so we're going to go ahead and mark this as |
This pull request modifies the export function in torch/_dynamo/eval_frame.py to handle the @torch._dynamo.assume_constant_result decorator. The changes allow for the correct handling of top-level annotated functions when using export with the assume_constant_result decorator.
Changes Made:
Added support for assume_constant_result decorator in the export function.
Additional Notes:
This modification was done for a course project as a computer science student. It's my first request :)
cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @chenyang78 @kadeng @chauhang