Skip to content

Reland: mul: convert inputs to result type.#7602

Merged
ysiraichi merged 3 commits intomasterfrom
ysiraichi/reland-fix-mul-dtype-promotion
Jul 2, 2024
Merged

Reland: mul: convert inputs to result type.#7602
ysiraichi merged 3 commits intomasterfrom
ysiraichi/reland-fix-mul-dtype-promotion

Conversation

@ysiraichi
Copy link
Copy Markdown
Collaborator

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

return bridge::TryGetXlaTensor(tensor);
});
XLA_CHECK(it != inputs_.end());
XLA_CHECK(it != inputs.end());
Copy link
Copy Markdown
Collaborator Author

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.

Copy link
Copy Markdown
Collaborator

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

Copy link
Copy Markdown
Collaborator Author

@ysiraichi ysiraichi Jul 1, 2024

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.

Copy link
Copy Markdown
Collaborator

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.

Copy link
Copy Markdown
Collaborator Author

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Error: Check Failed: it != inputs_.end() When Using torch and torch_xla Nightly Version (Post-20240527) with SPMD

2 participants