Skip to content

Conversation

@wrongtest-intellif
Copy link
Contributor

The MR try fix a corner case in software pipeline injection.

When the loop num is too few to fill the pipeline main body, the setup phase and draining phase's loop indices should not overlap. Currently the decomposition is

  • prologue: [0, num_stage - 1)
  • body: [num_stage - 1, loop_num)
  • epilogue: [loop_num, loop_num+ num_stage - 1)

Thus we should also ensure that i >= max(loop_num, num_stage - 1) for epilogue part.

@wrongtest-intellif
Copy link
Contributor Author

@tvm-bot re-run

@wrongtest-intellif wrongtest-intellif changed the title [TIR][SW] ensure non-overlap pipeline epilogue with prologue [TIR][SoftwarePipeline] Ensure pipeline epilogue and prologue do not overlap Feb 8, 2025
@wrongtest-intellif
Copy link
Contributor Author

@tvm-bot re-run

@tqchen tqchen merged commit 4a64be2 into apache:main Feb 8, 2025
20 checks passed
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.

2 participants