Skip to content

[BOO] Add split reduction and global optimization flags#1195

Merged
yzhang93 merged 4 commits into
mainfrom
users/yzhang93/split_reduction_flag
Nov 3, 2025
Merged

[BOO] Add split reduction and global optimization flags#1195
yzhang93 merged 4 commits into
mainfrom
users/yzhang93/split_reduction_flag

Conversation

@yzhang93
Copy link
Copy Markdown
Contributor

Adding two flags:

  1. For split reduction on weight backward convolutions.
  2. Global optimization to control edge reshape propagation [Global Opt] Add flag to control edge reshape propagation iree#22438.

@yzhang93 yzhang93 requested review from IanWood1 and rkayaith October 31, 2025 16:37
@yzhang93 yzhang93 force-pushed the users/yzhang93/split_reduction_flag branch from 0f1ee33 to 407fd26 Compare October 31, 2025 16:40
Copy link
Copy Markdown
Member

@rkayaith rkayaith left a comment

Choose a reason for hiding this comment

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

We'll have to bump the IREE pin before this can be merged

Comment on lines +234 to +235
flags.append("--iree-dispatch-creation-enable-split-reduction")
flags.append("--iree-global-opt-experimental-enable-edge-reshape-propagation")
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

Do we want some comments explaining why they were added? And in the case of iree-global-opt-experimental-enable-edge-reshape-propagation, saying that it should only be a temporary flag.

@yzhang93 yzhang93 force-pushed the users/yzhang93/split_reduction_flag branch from 407fd26 to 2f9f55e Compare October 31, 2025 17:17
Signed-off-by: yzhang93 <zhyuhang88@gmail.com>
@rkayaith
Copy link
Copy Markdown
Member

rkayaith commented Nov 1, 2025

The IREE bunp is merged now, once you rebase this should be good.

@yzhang93 yzhang93 merged commit fabcbad into main Nov 3, 2025
8 checks passed
@yzhang93 yzhang93 deleted the users/yzhang93/split_reduction_flag branch November 3, 2025 21:14
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.

3 participants