Skip to content

Commit d8df6f5

Browse files
committed
feat: pass in new options to optimize-comm
[skip ci]
1 parent ce3e626 commit d8df6f5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Diff for: src/Compiler.jl

+2-2
Original file line numberDiff line numberDiff line change
@@ -1034,11 +1034,11 @@ const optimize_comms_passes = (
10341034
"enzyme-hlo-generate-td{patterns=lower_rotate;concat_to_onedim_dus;concat_to_onedim_dusslice}",
10351035
"transform-interpreter",
10361036
"enzyme-hlo-remove-transform",
1037-
"optimize-communication",
1037+
"optimize-communication{rotate_to_pad_comm=1 dus_to_pad_comm=1 concat_to_pad_comm=1 extend_to_pad_comm=1 wrap_to_pad_comm=1}",
10381038
"enzyme-hlo-generate-td{patterns=lower_rotate;lower_wrap;lower_extend}",
10391039
"transform-interpreter",
10401040
"enzyme-hlo-remove-transform",
1041-
"optimize-communication",
1041+
"optimize-communication{rotate_to_pad_comm=1 dus_to_pad_comm=1 concat_to_pad_comm=1 extend_to_pad_comm=1 wrap_to_pad_comm=1}",
10421042
)
10431043

10441044
function compile_mlir!(

0 commit comments

Comments
 (0)