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

Add strip assertions flag #19014

Merged
merged 5 commits into from
Nov 14, 2024
Merged

Add strip assertions flag #19014

merged 5 commits into from
Nov 14, 2024

Conversation

IanWood1
Copy link
Contributor

@IanWood1 IanWood1 commented Nov 4, 2024

This change temporarily adds iree-opt-strip-assertions to SDXL CI which should fix the regression discussed in #19002. Assertions within linalg.generic ops can mess with dispatch creation and lead to poorly performing dispatches, despite the fact that they get stripped in later pipelines.

Part 2 (future PR or maybe this one): remove flag usage and make this default.

@IanWood1 IanWood1 force-pushed the strip_assertions branch 2 times, most recently from dfe6e48 to 5da931f Compare November 5, 2024 21:52
@IanWood1 IanWood1 marked this pull request as ready for review November 5, 2024 22:37
Part 1 (this pr): add `iree-opt-strip-assertions` to CI.

Part 2: remove flag usage and make this default.

Issue: iree-org#1900

Signed-off-by: Ian Wood <ianwood2024@u.northwestern.edu>
Signed-off-by: Ian Wood <ianwood2024@u.northwestern.edu>
Signed-off-by: Ian Wood <ianwood2024@u.northwestern.edu>
@IanWood1
Copy link
Contributor Author

Could I get a review on this? This might also be affecting Llama performance, there were ~100 dispatches that could be CSEed if the asserts were removed.

Copy link
Contributor

@MaheshRavishankar MaheshRavishankar left a comment

Choose a reason for hiding this comment

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

So we dont want to turn this on by default?

@IanWood1
Copy link
Contributor Author

The idea was to to fix sdxl regressions first by just using the flag, then make it default after because there may be some side effects of stripping all assertions like this by default. But I can change this to be default if that seems better

@MaheshRavishankar
Copy link
Contributor

The idea was to to fix sdxl regressions first by just using the flag, then make it default after because there may be some side effects of stripping all assertions like this by default. But I can change this to be default if that seems better

Makes sense. Lets keep to the original plan.

@IanWood1 IanWood1 merged commit bf711a1 into iree-org:main Nov 14, 2024
36 checks passed
Groverkss pushed a commit to Groverkss/iree that referenced this pull request Dec 1, 2024
This change _temporarily_ adds `iree-opt-strip-assertions` to SDXL CI
which should fix the regression discussed in
iree-org#19002. Assertions within
`linalg.generic` ops can mess with dispatch creation and lead to poorly
performing dispatches, despite the fact that they get stripped in later
pipelines.

---------

Signed-off-by: Ian Wood <ianwood2024@u.northwestern.edu>
giacs-epic pushed a commit to giacs-epic/iree that referenced this pull request Dec 4, 2024
This change _temporarily_ adds `iree-opt-strip-assertions` to SDXL CI
which should fix the regression discussed in
iree-org#19002. Assertions within
`linalg.generic` ops can mess with dispatch creation and lead to poorly
performing dispatches, despite the fact that they get stripped in later
pipelines.

---------

Signed-off-by: Ian Wood <ianwood2024@u.northwestern.edu>
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.

3 participants