Skip to content
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

Partially handle operations with permutation factors for explicit reshards. #387

Closed
wants to merge 0 commits into from

Conversation

copybara-service[bot]
Copy link

Partially handle operations with permutation factors for explicit reshards.

  • In case there is no sharded permutation factors on any tensor, find common axes with no extra special handling.
  • In case it is a binary operation and only one side has sharded permutation factors, use the sharding of the side without sharded permutation as the common axes.
  • In case it is a binary operation and both sides have sharded permutation factors, do not handle, skip inserting explicit reshards.
  • In case it has some sharded permutation factors, and it is not a binary operation, e.g. concatenate, skip inserting explicit reshards.

It affects:

  • ConcatenateOp
  • PadOp
  • SliceOp
  • ReverseOp

@copybara-service copybara-service bot force-pushed the test_727841913 branch 2 times, most recently from 30519cf to 498c0e6 Compare February 19, 2025 14:09
@copybara-service copybara-service bot closed this Feb 19, 2025
@copybara-service copybara-service bot deleted the test_727841913 branch February 19, 2025 14:36
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.

0 participants