Skip to content

Push shardy outliner up past RemoveSubAxesInInputOutputShardings pass.#1185

Open
copybara-service[bot] wants to merge 1 commit intomainfrom
test_900127894
Open

Push shardy outliner up past RemoveSubAxesInInputOutputShardings pass.#1185
copybara-service[bot] wants to merge 1 commit intomainfrom
test_900127894

Conversation

@copybara-service
Copy link
Copy Markdown

Push shardy outliner up past RemoveSubAxesInInputOutputShardings pass.

RemoveSubAxesInInputOutputShardingsPass runs for all functions. Before this change, it runs only for the main func because the module is an inlined one. Hence, in order to keep the behavior the same, on this change, it should explicitly run it on the main func.

This CL should not change the behaviour.

RemoveSubAxesInInputOutputShardingsPass runs for all functions. Before this change, it runs only for the main func because the module is an inlined one. Hence, in order to keep the behavior the same, on this change, it should explicitly run it on the main func.

This CL should not change the behaviour.

PiperOrigin-RevId: 900127894
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.

1 participant