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

[BACKEND] Fix an issue with the pipeliner #4247

Merged
merged 1 commit into from
Jul 9, 2024

Conversation

htyu
Copy link
Collaborator

@htyu htyu commented Jul 2, 2024

During pipelining operations that do not depend on or being dependent by anchor operations are considered remaining ops and will be scheduled into the last stage. These ops do not present in the existing stages but can be visited by other staged ops. Fixing an ICE when looking for clusters for them.

@htyu htyu requested a review from ptillet as a code owner July 2, 2024 23:48
@Jokeren
Copy link
Contributor

Jokeren commented Jul 3, 2024

What does ICE mean?

@htyu
Copy link
Collaborator Author

htyu commented Jul 3, 2024

What does ICE mean?

I mean internal compiler error. Sorry for the confusion. I should just say a issue :)

@htyu htyu changed the title [BACKEND] Fix an ICE with the pipeliner [BACKEND] Fix an issue with the pipeliner Jul 3, 2024
@htyu htyu requested a review from manman-ren July 8, 2024 16:47
@manman-ren
Copy link
Collaborator

This looks reasonable to me. I wonder if we can further reduce the test case.

@pawelszczerbuk
Copy link
Contributor

Thanks for fixing the bug! +1 to the @manman-ren request of coming up with simpler lit reproducer.

@htyu
Copy link
Collaborator Author

htyu commented Jul 8, 2024

Test reduced.

@manman-ren
Copy link
Collaborator

Thanks for reducing the test case. LGTM

@plotfi
Copy link
Contributor

plotfi commented Jul 9, 2024

Can confirm this fixes pytorch/pytorch#129911

@pawelszczerbuk pawelszczerbuk merged commit 050b41d into triton-lang:main Jul 9, 2024
7 checks passed
@plotfi
Copy link
Contributor

plotfi commented Jul 10, 2024

This also fixes pytorch/pytorch#129625

@drisspg
Copy link

drisspg commented Aug 16, 2024

Are there any labels / tracking needed to get this cherry-picked into the release branch?

drisspg pushed a commit to drisspg/triton that referenced this pull request Aug 19, 2024
During pipelining operations that do not depend on or being dependent by
anchor operations are considered remaining ops and will be scheduled
into the last stage. These ops do not present in the existing stages but
can be visited by other staged ops. Fixing an ICE when looking for
clusters for them.
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.

None yet

6 participants