-
Notifications
You must be signed in to change notification settings - Fork 24.4k
Added {logical_not, trace} refs, moved logical ops to use method overloads #79000
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
Conversation
…loads [ghstack-poisoned]
🔗 Helpful links
❌ 1 New FailuresAs of commit 003d457 (more details on the Dr. CI page): Expand to see more
🕵️ 1 new failure recognized by patternsThe following CI failures do not appear to be due to upstream breakages
|
…method overloads" [ghstack-poisoned]
One thing I did in this PR that's questionable is porting |
…method overloads" [ghstack-poisoned]
…method overloads" [ghstack-poisoned]
…method overloads" [ghstack-poisoned]
torch/_refs/__init__.py
Outdated
# populate the decomp table | ||
@register_decomposition(torch.ops.aten.trace) | ||
def trace(self: TensorLikeType) -> TensorLikeType: | ||
return torch.sum(torch.diag(self)) |
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.
This is incorrect because diag accepts vectors and matrices but trace only accepts matrices
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.
(what about adding an error input for it?)
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.
Added an error input for it (although it's unfortunately not tested now for ref implementations since diag isn't there).
Hey @Chillee! Looks good overall -- a few small comments for your review inline |
Yeah don't worry about it |
…method overloads" [ghstack-poisoned]
…method overloads" [ghstack-poisoned]
…method overloads" [ghstack-poisoned]
…method overloads" [ghstack-poisoned]
…method overloads" [ghstack-poisoned]
Unfortunately test failures look related |
…method overloads" [ghstack-poisoned]
…method overloads" [ghstack-poisoned]
…method overloads" [ghstack-poisoned]
…method overloads" [ghstack-poisoned]
…method overloads" [ghstack-poisoned]
…method overloads" [ghstack-poisoned]
fyi @zasdfgbnm |
@pytorchbot revert -m "Introduces test failure, see https://hud.pytorch.org/pr/79000" -c ignoredsignal |
@pytorchbot successfully started a revert job. Check the current status here |
…hod overloads" This reverts commit 64b6bd8. Reverted #79000 on behalf of https://github.com/malfet due to Introduces test failure, see https://hud.pytorch.org/pr/79000
…loads (#79000) Summary: Pull Request resolved: #79000 Approved by: https://github.com/ezyang Test Plan: contbuild & OSS CI, see https://hud.pytorch.org/commit/pytorch/pytorch/64b6bd8c1e73abb039c111bb7a9ca37a5baff71e Reviewed By: osalpekar Differential Revision: D37058880 Pulled By: osalpekar fbshipit-source-id: 3b8e1d7343815422f08b17ba79727fac0592d709
…hod overloads" Summary: This reverts commit 64b6bd8. Reverted #79000 on behalf of https://github.com/malfet due to Introduces test failure, see https://hud.pytorch.org/pr/79000 Test Plan: contbuild & OSS CI, see https://hud.pytorch.org/commit/pytorch/pytorch/d67309aefb435930c77d45935181b3238cf5772e Reviewed By: osalpekar Differential Revision: D37059062 Pulled By: osalpekar fbshipit-source-id: 0b6d659b1d8601ee65ffd93381559ca7c6e2ba83
Stack from ghstack (oldest at bottom):