Skip to content

Fix epilogue loop generation#2661

Merged
naoyam merged 3 commits intomainfrom
circular_buffer_fix_epilogue
Jul 23, 2024
Merged

Fix epilogue loop generation#2661
naoyam merged 3 commits intomainfrom
circular_buffer_fix_epilogue

Conversation

@naoyam
Copy link
Collaborator

@naoyam naoyam commented Jul 22, 2024

Note that CudaCodeGenerator currently always starts a loop with 0 even if ForLoop::start_ is non-zero. I think this change is safe since only use case of non-zero start should be the epilogue loop of circular buffering. The concern of degenerate loop should not be applicable.

Note that CudaCodeGenerator currently always starts a loop with 0 even
if ForLoop::start_ is non-zero. I think this change is safe since only
use case of non-zero start should be the epilogue loop of circular
buffering. The concern of degenerate loop should not be applicable.
@naoyam naoyam requested a review from rdspring1 July 22, 2024 21:17
@naoyam
Copy link
Collaborator Author

naoyam commented Jul 22, 2024

!build --diff

@naoyam naoyam mentioned this pull request Jul 22, 2024
Copy link
Collaborator

@rdspring1 rdspring1 left a comment

Choose a reason for hiding this comment

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

LGTM

//! Get the number of circular buffer stages for the given axis,
//! the number of stages will be 2 in the case of circular buffer loop.
unsigned int getStageDepthFor(IterDomain* circular_buffered_id);
unsigned int getStageDepthFor(IterDomain* circular_buffered_id) const;
Copy link
Collaborator

Choose a reason for hiding this comment

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

nitpick: change from unsigned int to int64_t

@naoyam
Copy link
Collaborator Author

naoyam commented Jul 23, 2024

!build

1 similar comment
@naoyam
Copy link
Collaborator Author

naoyam commented Jul 23, 2024

!build

@naoyam naoyam merged commit ea43aba into main Jul 23, 2024
@naoyam naoyam deleted the circular_buffer_fix_epilogue branch July 23, 2024 14:57
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