diff --git a/src/CompileOptions.jl b/src/CompileOptions.jl index e8cac78be6..4e44aabd67 100644 --- a/src/CompileOptions.jl +++ b/src/CompileOptions.jl @@ -12,7 +12,7 @@ communication. wrap_comm::Int = 0 extend_comm::Int = 0 dus_to_pad_manual_comp_comm::Int = 0 # 2 - dus_to_pad_comm::Int = 1 + dus_to_pad_comm::Int = 0 concat_two_operands_comm::Int = 0 concat_to_pad_comm::Int = 1 extend_to_pad_comm::Int = 1 diff --git a/src/Compiler.jl b/src/Compiler.jl index 0639c142d5..59583a4076 100644 --- a/src/Compiler.jl +++ b/src/Compiler.jl @@ -1543,7 +1543,7 @@ end function get_optimize_comms_passes(options::OptimizeCommunicationOptions) options_str = String(options) res = [ - "enzyme-hlo-generate-td{patterns=lower_rotate;concat_to_onedim_dus;concat_to_onedim_dusslice;concatreshape_to_onedim_dus}", + "enzyme-hlo-generate-td{patterns=lower_rotate;concat_to_onedim_dus;concat_to_onedim_dusslice;concatreshape_to_onedim_dus;recognize_wrap;recognize_extend}", "transform-interpreter", "enzyme-hlo-remove-transform", "enzyme-hlo-generate-td{patterns=reshape_to_broadcast}",