Skip to content

[Unity][DLight] Fix Reduction Rule#15412

Merged
junrushao merged 1 commit into
apache:unityfrom
Hzfengsy:fix_dl_reduction
Jul 26, 2023
Merged

[Unity][DLight] Fix Reduction Rule#15412
junrushao merged 1 commit into
apache:unityfrom
Hzfengsy:fix_dl_reduction

Conversation

@Hzfengsy

Copy link
Copy Markdown
Member

This PR fixes a correctness issue when the epilogue length is different with the len_tx.

cc @junrushao @MasterJH5574

@tvm-bot

tvm-bot commented Jul 26, 2023

Copy link
Copy Markdown
Collaborator

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

Generated by tvm-bot

This PR fixes a correctness issue when the epilogue length is different
with the `len_tx`.

@MasterJH5574 MasterJH5574 left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

I see. Thanks for the fix. I was assuming the spatial loop is divided into bx and tx only, and was not aware that there could be another loop which is not bound to any thread (e.g., the loop with length 8 in your added test case).

@junrushao
junrushao merged commit 51d7128 into apache:unity Jul 26, 2023
junrushao pushed a commit to junrushao/tvm that referenced this pull request Jul 30, 2023
This PR fixes a correctness issue when the epilogue length is different
with the `len_tx`.
@Hzfengsy
Hzfengsy deleted the fix_dl_reduction branch July 30, 2023 05:31
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.

4 participants