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

[Stream][NFC] Move topological sort out of loop #19794

Merged
merged 1 commit into from
Jan 23, 2025

Conversation

IanWood1
Copy link
Contributor

There doesn't appear to be a need to re-sort for after each transformation with each partition builder (for 405b: ~5,000 iters for several thousand ops). Also, I moved this after erasing dead ops since there is no reason to sort ops that will just be deleted.

On 405b sharded, this reduces compile time from 4.4min -> 8sec for this pass.

Copy link
Collaborator

@benvanik benvanik left a comment

Choose a reason for hiding this comment

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

wow, impressive!

if everything seems to pass then lgtm!

@IanWood1 IanWood1 marked this pull request as ready for review January 23, 2025 22:03
@IanWood1 IanWood1 merged commit dd31890 into iree-org:main Jan 23, 2025
40 checks passed
@MaheshRavishankar
Copy link
Contributor

Thanks for chasing this down!

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