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

[DispatchCreation] Add CSE before canonicalization of flow.dispatch.workgroups #19178

Merged

Conversation

MaheshRavishankar
Copy link
Contributor

The real change in this PR is to add CSE before canonicalization so
that operands of flow.dispatch.workgroups can be better de-duped.

Since this needed a pipeline test, noticed the pipeline test existed
in Flow and called the global optimization, dispatch creation and flow
pipelines. This PR moves the test to dispatch creation and only tests
that pipeline.

Signed-off-by: MaheshRavishankar mahesh.ravishankar@gmail.com

of contractions by default due to regressions. This PR fixes the
regressions and turns them on again.

Fixes are:

1) If the original code had multiple dynamic dimensions, where one or
   more of those are not blocked, those need to be accounted for in
   the `expand_shape` generated.

Signed-off-by: MaheshRavishankar <mahesh.ravishankar@gmail.com>
@MaheshRavishankar
Copy link
Contributor Author

Depends on #19171 . Only review the last commit.

…workgroups`.

The real change in this PR is to add CSE before canonicalization so
that operands of `flow.dispatch.workgroups` can be better de-duped.

Since this needed a pipeline test, noticed the pipeline test existed
in Flow and called the global optimization, dispatch creation and flow
pipelines. This PR moves the test to dispatch creation and only tests
that pipeline.

Signed-off-by: MaheshRavishankar <mahesh.ravishankar@gmail.com>
Copy link
Contributor

@hanhanW hanhanW left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

out of curiosity: should we have a pipeline_tests for flow?

@MaheshRavishankar
Copy link
Contributor Author

out of curiosity: should we have a pipeline_tests for flow?

Potentially, but Flow does not actually form dispatches now.. So I was mostly looking for dispatch formation pipeline tests.

@MaheshRavishankar MaheshRavishankar merged commit 50a001f into iree-org:main Nov 18, 2024
36 checks passed
Groverkss pushed a commit to Groverkss/iree that referenced this pull request Dec 1, 2024
…workgroups` (iree-org#19178)

The real change in this PR is to add CSE before canonicalization so
that operands of `flow.dispatch.workgroups` can be better de-duped.

Since this needed a pipeline test, noticed the pipeline test existed
in Flow and called the global optimization, dispatch creation and flow
pipelines. This PR moves the test to dispatch creation and only tests
that pipeline.

Signed-off-by: MaheshRavishankar <mahesh.ravishankar@gmail.com>
giacs-epic pushed a commit to giacs-epic/iree that referenced this pull request Dec 4, 2024
…workgroups` (iree-org#19178)

The real change in this PR is to add CSE before canonicalization so
that operands of `flow.dispatch.workgroups` can be better de-duped.

Since this needed a pipeline test, noticed the pipeline test existed
in Flow and called the global optimization, dispatch creation and flow
pipelines. This PR moves the test to dispatch creation and only tests
that pipeline.

Signed-off-by: MaheshRavishankar <mahesh.ravishankar@gmail.com>
Signed-off-by: Giacomo Serafini <179146510+giacs-epic@users.noreply.github.com>
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