[SD] Add flag to collapse reduction dims pre dispatch formation (#1999)

This commit is contained in:
Quinn Dawkins
2023-11-30 00:09:17 -05:00
committed by GitHub
parent 78c607e1d3
commit d31d28b082

View File

@@ -563,6 +563,9 @@ def get_opt_flags(model, precision="fp16"):
iree_flags += opt_flags[model][is_tuned][precision][
"specified_compilation_flags"
][device]
# Due to lack of support for multi-reduce, we always collapse reduction
# dims before dispatch formation right now.
iree_flags += ["--iree-flow-collapse-reduction-dims"]
return iree_flags