-
Notifications
You must be signed in to change notification settings - Fork 26.3k
Change ShardedTensor torch_function to be a classmethod. #74273
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
Change ShardedTensor torch_function to be a classmethod. #74273
Conversation
As per discussion in #63767, torch_function should be a classmethod and instance method based implementations would be deprecated soon. As a result, making this a classmethod and using the suggestion in #63767 to find the ShardedTensor instance in the args/kwargs. Differential Revision: [D34913799](https://our.internmc.facebook.com/intern/diff/D34913799/) [ghstack-poisoned]
CI Flow Status⚛️ CI FlowRuleset - Version:
|
🔗 Helpful links
💊 CI failures summary and remediationsAs of commit 637c65e (more details on the Dr. CI page):
🕵️ 2 new failures recognized by patternsThe following CI failures do not appear to be due to upstream breakages:
|
As per discussion in #63767, torch_function should be a classmethod and instance method based implementations would be deprecated soon. As a result, making this a classmethod and using the suggestion in #63767 to find the ShardedTensor instance in the args/kwargs. Differential Revision: [D34913799](https://our.internmc.facebook.com/intern/diff/D34913799/) ghstack-source-id: 151463383 Pull Request resolved: #74273
wanchaol
left a comment
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.
lgtm, thanks for fixing it!
Summary: Pull Request resolved: #74273 As per discussion in #63767, torch_function should be a classmethod and instance method based implementations would be deprecated soon. As a result, making this a classmethod and using the suggestion in #63767 to find the ShardedTensor instance in the args/kwargs. ghstack-source-id: 151463383 Test Plan: waitforbuildbot Reviewed By: wanchaol Differential Revision: D34913799 fbshipit-source-id: c2f95ace861b118ea539215188d8723316cbb066
|
Hey @pritamdamania87. |
Summary: Pull Request resolved: #74273 As per discussion in #63767, torch_function should be a classmethod and instance method based implementations would be deprecated soon. As a result, making this a classmethod and using the suggestion in #63767 to find the ShardedTensor instance in the args/kwargs. ghstack-source-id: 151463383 Test Plan: waitforbuildbot Reviewed By: wanchaol Differential Revision: D34913799 fbshipit-source-id: c2f95ace861b118ea539215188d8723316cbb066 (cherry picked from commit 1806664)
Stack from ghstack (oldest at bottom):
As per discussion in #63767,
torch_function should be a classmethod and instance method based
implementations would be deprecated soon.
As a result, making this a classmethod and using the suggestion in
#63767 to find the ShardedTensor
instance in the args/kwargs.
Differential Revision: D34913799