We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d91295f commit ce3e8e6Copy full SHA for ce3e8e6
src/Compiler.jl
@@ -682,6 +682,7 @@ function optimization_passes(;
682
"concat_concat_axis_swap",
683
"concat_multipad",
684
"concat_concat_to_dus",
685
+ "concat_to_onedim_dusslice",
686
]
687
688
if DUS_SLICE_SIMPLIFY[]
@@ -1029,7 +1030,7 @@ end
1029
1030
1031
const optimize_comms_passes = (
1032
# rotate handler presently broken (and handled okay presently), disabling for now
- "enzyme-hlo-generate-td{patterns=lower_rotate;concat_to_onedim_dus}",
1033
+ "enzyme-hlo-generate-td{patterns=lower_rotate;concat_to_onedim_dus;concat_to_onedim_dusslice}",
1034
"transform-interpreter",
1035
"enzyme-hlo-remove-transform",
1036
"optimize-communication",
0 commit comments