-
Notifications
You must be signed in to change notification settings - Fork 21.3k
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
[JIT] fix resolving of functions in torch/functional. fix compilation of torch.stft #33504
Conversation
[ghstack-poisoned]
[ghstack-poisoned]
ghstack-source-id: ec658b69fab9feab4b836ad76586c869e441aee2 Pull Request resolved: #33504
💊 CircleCI build failures summary and remediationsAs of commit 442016a: Commit 442016a was recently pushed. Waiting for builds... This comment was automatically generated by Dr. CI (expand for details).Follow this link to opt-out of these comments for your Pull Requests.Please report bugs/suggestions on the GitHub issue tracker. This comment has been revised 31 times. |
…compilation of torch.stft" Fix resolution fo functions that are bound onto torch in torch/functional.py. This does not fix compilation of all of those functions, those will be done in follow ups. Does torch.stft as a start. Fix for #21478 [ghstack-poisoned]
ghstack-source-id: fe824a5a6e1ff7b3f980ddadfa152dcd5e77a427 Pull Request resolved: #33504
…compilation of torch.stft" Fix resolution fo functions that are bound onto torch in torch/functional.py. This does not fix compilation of all of those functions, those will be done in follow ups. Does torch.stft as a start. Fix for #21478 [ghstack-poisoned]
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.
Looks good to go after a couple mechanical changes
torch/jit/_builtins.py
Outdated
@@ -83,8 +83,25 @@ | |||
(torch.nn.init._no_grad_zero_, "aten::_no_grad_zero_"), | |||
(torch._C._get_tracing_state, "aten::_get_tracing_state"), | |||
(warnings.warn, "aten::warn"), | |||
(torch._C._VariableFunctions.stft, "aten::stft") |
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.
if you do the above you shouldn't need this here, instances in nn/functional.py
resolve to their aten::
equivalent already
pytorch/torch/nn/functional.py
Lines 805 to 806 in cfb4862
return (_VF.dropout_(input, p, training) | |
if inplace |
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.
For some reason I had to add _VF.stft here, not sure why. But I moved _VF so that i can just use _VF.stft
inline.
…compilation of torch.stft" Fix resolution fo functions that are bound onto torch in torch/functional.py. This does not fix compilation of all of those functions, those will be done in follow ups. Does torch.stft as a start. Fixes #21478 [ghstack-poisoned]
…compilation of torch.stft" Fix resolution fo functions that are bound onto torch in torch/functional.py. This does not fix compilation of all of those functions, those will be done in follow ups. Does torch.stft as a start. Fixes #21478 Differential Revision: [D20014591](https://our.internmc.facebook.com/intern/diff/D20014591) [ghstack-poisoned]
ghstack-source-id: 23bd174a965c5c38e9cc9d18cc212c9872e8cba3 Pull Request resolved: #33504
…compilation of torch.stft" Fix resolution fo functions that are bound onto torch in torch/functional.py. This does not fix compilation of all of those functions, those will be done in follow ups. Does torch.stft as a start. Fixes #21478 Differential Revision: [D20014591](https://our.internmc.facebook.com/intern/diff/D20014591) [ghstack-poisoned]
ghstack-source-id: bfc3f3eb0c57b9d6f894531f5c927caf01eb7558 Pull Request resolved: #33504
…compilation of torch.stft" Fix resolution fo functions that are bound onto torch in torch/functional.py. This does not fix compilation of all of those functions, those will be done in follow ups. Does torch.stft as a start. Fixes #21478 Differential Revision: [D20014591](https://our.internmc.facebook.com/intern/diff/D20014591) [ghstack-poisoned]
…compilation of torch.stft" Fix resolution fo functions that are bound onto torch in torch/functional.py. This does not fix compilation of all of those functions, those will be done in follow ups. Does torch.stft as a start. Fixes #21478 Differential Revision: [D20014591](https://our.internmc.facebook.com/intern/diff/D20014591) [ghstack-poisoned]
…compilation of torch.stft" Fix resolution fo functions that are bound onto torch in torch/functional.py. This does not fix compilation of all of those functions, those will be done in follow ups. Does torch.stft as a start. Fixes #21478 Differential Revision: [D20014591](https://our.internmc.facebook.com/intern/diff/D20014591) [ghstack-poisoned]
…compilation of torch.stft" Fix resolution fo functions that are bound onto torch in torch/functional.py. This does not fix compilation of all of those functions, those will be done in follow ups. Does torch.stft as a start. Fixes #21478 Differential Revision: [D20014591](https://our.internmc.facebook.com/intern/diff/D20014591) [ghstack-poisoned]
…compilation of torch.stft" Fix resolution fo functions that are bound onto torch in torch/functional.py. This does not fix compilation of all of those functions, those will be done in follow ups. Does torch.stft as a start. Fixes #21478 Differential Revision: [D20014591](https://our.internmc.facebook.com/intern/diff/D20014591) [ghstack-poisoned]
… of torch.stft (#33504) Summary: Pull Request resolved: #33504 Fix resolution fo functions that are bound onto torch in torch/functional.py. This does not fix compilation of all of those functions, those will be done in follow ups. Does torch.stft as a start. Fixes #21478 Test Plan: Imported from OSS Differential Revision: D20014591 Pulled By: eellison fbshipit-source-id: bb362f1b5479adbb890e72a54111ef716679d127
… of torch.stft (pytorch#33504) Summary: Pull Request resolved: pytorch#33504 Fix resolution fo functions that are bound onto torch in torch/functional.py. This does not fix compilation of all of those functions, those will be done in follow ups. Does torch.stft as a start. Fixes pytorch#21478 Test Plan: Imported from OSS Differential Revision: D20014591 Pulled By: eellison fbshipit-source-id: bb362f1b5479adbb890e72a54111ef716679d127
Stack from ghstack:
Fix resolution fo functions that are bound onto torch in torch/functional.py. This does not fix compilation of all of those functions, those will be done in follow ups. Does torch.stft as a start.
Fixes #21478
Differential Revision: D20014591