compiled autograd: default fakeify backward inputs with static shapes instead of duck sizing#133581
compiled autograd: default fakeify backward inputs with static shapes instead of duck sizing#133581bdhirsh wants to merge 1 commit intogh/bdhirsh/607/basefrom
Conversation
… instead of duck sizing [ghstack-poisoned]
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/133581
Note: Links to docs will display an error until the docs builds have been completed. ❌ 40 New Failures, 1 Cancelled Job, 1 Unrelated FailureAs of commit 64d034e with merge base 454713f ( NEW FAILURES - The following jobs have failed:
BROKEN TRUNK - The following job failed but were present on the merge base:👉 Rebase onto the `viable/strict` branch to avoid these failures
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
|
Still don't fully understand the change, changing the fakification logic shouldn't do anything:
|
|
@bdhirsh I have actually completely forgotten what exactly we talked about in our 1:1, so you had better it write it down here :P |
|
The compiled autograd failures are both due to the GmWrapper inputs not being unflattened properly in the previous PR. Tests passed for me if you do something like If the traced graph is always the same between tracing with static FakeTensor vs dynamic ones (is this is true?), then this approach should be okay |
|
Looks like this PR hasn't been updated in a while so we're going to go ahead and mark this as |
This fixes the problem described here, by tweaking compiled autograd to default to fakifying with static shapes instead of defaulting to duck sizing.
I think there are some (independent) dynamic shape + compiled autograd issues though, detailed here #133575 (cc @ezyang @penguinwu @bobrenjc93 @voznesenskym @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @chenyang78 @kadeng @chauhang @amjames @xmfan @yf225 @rec )
Stack from ghstack (oldest at bottom):
cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @chenyang78 @kadeng @chauhang @amjames