Skip to content

Commit ce3e8e6

Browse files
committed
Update Compiler.jl
1 parent d91295f commit ce3e8e6

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/Compiler.jl

+2-1
Original file line numberDiff line numberDiff line change
@@ -682,6 +682,7 @@ function optimization_passes(;
682682
"concat_concat_axis_swap",
683683
"concat_multipad",
684684
"concat_concat_to_dus",
685+
"concat_to_onedim_dusslice",
685686
]
686687

687688
if DUS_SLICE_SIMPLIFY[]
@@ -1029,7 +1030,7 @@ end
10291030

10301031
const optimize_comms_passes = (
10311032
# rotate handler presently broken (and handled okay presently), disabling for now
1032-
"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}",
10331034
"transform-interpreter",
10341035
"enzyme-hlo-remove-transform",
10351036
"optimize-communication",

0 commit comments

Comments
 (0)