-
Notifications
You must be signed in to change notification settings - Fork 454
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
Reland: mul
: convert inputs to result type.
#7602
Conversation
@@ -158,7 +158,7 @@ class OpConfig { | |||
inputs.begin(), inputs.end(), [](const at::Tensor& tensor) { | |||
return bridge::TryGetXlaTensor(tensor); | |||
}); | |||
XLA_CHECK(it != inputs_.end()); | |||
XLA_CHECK(it != inputs.end()); |
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.
Previously, incorrectly checking against inputs_
instead of inputs
.
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.
man! I should catch this during reviewing haha
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.
Must admit that it was NOT (wow, what a typo) the best naming choice, though.
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.
nice catch 🤣 , I will check the TPU CI after this pr is merged.
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.
Again, sorry for the typo there. That was really bad.
Fix: #7266
This PR tries to re-land #7130, which was reverted due to regressions on XLA:TPU. On top of that, it also fixes a subtle bug that went under the radar. Although we couldn't find a minimal reproducer, internal CI for TPU was reportedly broken due to this PR.
Therefore, the idea is to have this PR merged, and check the internal CI for TPU results.
cc @miladm @vanbasten23 @JackCaoG