Skip to content

Fix equation parsing for fallback logic#4027

Merged
steventk-g merged 1 commit intomasterfrom
einsum-fix
Sep 21, 2022
Merged

Fix equation parsing for fallback logic#4027
steventk-g merged 1 commit intomasterfrom
einsum-fix

Conversation

@steventk-g
Copy link
Copy Markdown
Collaborator

@steventk-g steventk-g commented Sep 20, 2022

  • Modify ParseEquation in torch_xla/csrc/aten_xla_type.cpp to deal with 1-argument einsum equations correctly.
  • Cleanse spaces from equation strings, which are allowed in PyTorch but not XLA.

@steventk-g steventk-g requested a review from JackCaoG September 20, 2022 22:40
@steventk-g steventk-g marked this pull request as ready for review September 20, 2022 22:40
@steventk-g steventk-g force-pushed the einsum-fix branch 2 times, most recently from 27c3d5c to b0e5a21 Compare September 21, 2022 00:09
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants