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

[CodePart] fix results of specialized IfOp #21

Merged
merged 2 commits into from
Jan 14, 2025
Merged

[CodePart] fix results of specialized IfOp #21

merged 2 commits into from
Jan 14, 2025

Conversation

manman-ren
Copy link
Contributor

Summary: When running hstu with multi-target, we have: IfOp
thenBlock:
opA (taskId 1)
opB (taskId 2)
Yield resOpA resOpB
elseBlock:
Op uses output of IfOp

In this case, when specializing IfOp for taskId 1, we should remove some of the results
IfOp (taskId 1)
thenBlock:
opA (taskId 1)
Yield resOpA
elseBlock:

@manman-ren manman-ren requested a review from htyu January 12, 2025 02:14
@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Meta Open Source bot. label Jan 12, 2025
Copy link
Contributor

@htyu htyu left a comment

Choose a reason for hiding this comment

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

Good catch!

lib/Dialect/TritonGPU/Transforms/WSCodePartition.cpp Outdated Show resolved Hide resolved
lib/Dialect/TritonGPU/Transforms/WSCodePartition.cpp Outdated Show resolved Hide resolved
@manman-ren
Copy link
Contributor Author

Thanks for reviewing! Updated the patch.

Copy link
Contributor

@htyu htyu left a comment

Choose a reason for hiding this comment

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

LGTM!

Summary: When running hstu with multi-target, we have:
IfOp
  thenBlock:
    opA (taskId 1)
    opB (taskId 2)
    Yield resOpA resOpB
  elseBlock:
Op uses output of IfOp

In this case, when specializing IfOp for taskId 1, we should remove some
of the results
IfOp (taskId 1)
  thenBlock:
    opA (taskId 1)
    Yield resOpA
  elseBlock:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:
Summary:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:
@manman-ren manman-ren merged commit c44915f into ws Jan 14, 2025
2 of 6 checks passed
@manman-ren manman-ren deleted the mren/ws-fix-dp branch January 16, 2025 04:14
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Meta Open Source bot.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants