-
Notifications
You must be signed in to change notification settings - Fork 25.2k
Add complex_to_float option in ReductionOpInfo #78423
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
As per title [ghstack-poisoned]
🔗 Helpful links
❌ 1 New FailuresAs of commit 7ab07c0 (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
|
As per title [ghstack-poisoned]
What's up with test failures? They don't seem related, but also trunk doesn't look broken. |
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 looks good, but what's up with test failures?
kwargs=kwargs)) | ||
|
||
return inputs | ||
yield SampleInput(t.detach().requires_grad_(requires_grad), args=args, kwargs=kwargs) |
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.
wow previous invocation seems buggy, bringing unneeded history.
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.
I do not understand how the previous implementation worked tbh... I should look further into that.
This little fix needed some extra fixes here and there (hence the test failures). I needed to convert quite a few other functions into generators after this one. Let's see if I managed to not miss any one.
@@ -1381,6 +1378,7 @@ def __init__( | |||
# These are mutually exclusive options | |||
assert not (result_dtype and promotes_int_to_float) | |||
assert not (result_dtype and promotes_int_to_int64) | |||
assert not (result_dtype and complex_to_real) |
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.
but vector_norm
accepts dtype
arg?
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.
result_dtype
is used to specify when a function always returns a given dtype (e.g. eq
).
As per title [ghstack-poisoned]
As per title [ghstack-poisoned]
Summary: As per title Pull Request resolved: #78423 Approved by: https://github.com/ngimel Test Plan: contbuild & OSS CI, see https://hud.pytorch.org/commit/pytorch/pytorch/9b97d5d625b02ed27a22d9e3c981caa4b1dea5c7 Reviewed By: seemethere Differential Revision: D36815705 Pulled By: seemethere fbshipit-source-id: d9e59c329bc63e56719ae39c07c71d8d656b198b
Stack from ghstack:
As per title