-
Notifications
You must be signed in to change notification settings - Fork 25.7k
automatic_dynamic_shapes_mark_as #141415
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
automatic_dynamic_shapes_mark_as #141415
Conversation
This adds an option to cause automatic dynamic shapes to trigger unbacked SymInts rather than backed SymInts. This can potentially help if you are still seeing recompilations from 0/1 specialization but it also might just cause your program to fail with GuardOnDataDependent errors. Signed-off-by: Edward Z. Yang <ezyang@meta.com> ghstack-source-id: a3ca847 Pull Request resolved: #141415
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/141415
Note: Links to docs will display an error until the docs builds have been completed. ✅ You can merge normally! (1 Unrelated Failure)As of commit 7df218a with merge base dd2d0c6 ( UNSTABLE - The following job failed but was likely due to flakiness present on trunk and has been marked as unstable:
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
|
Can you elaborate on the tradeoffs between this and the more explicit path of having a flag like |
I'm not sure exactly what you're proposing with |
Fixes #137100 Should also add a mark_oblivious API for manual control. Signed-off-by: Edward Z. Yang <ezyang@meta.com> Pull Request resolved: #141444 Approved by: https://github.com/bobrenjc93 ghstack dependencies: #141415
Stack from ghstack (oldest at bottom):
This adds an option to cause automatic dynamic shapes to trigger
unbacked SymInts rather than backed SymInts. This can potentially
help if you are still seeing recompilations from 0/1 specialization
but it also might just cause your program to fail with
GuardOnDataDependent errors.
Signed-off-by: Edward Z. Yang ezyang@meta.com
cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @chenyang78 @kadeng @chauhang @amjames