From 0e15f4af49e4e81ff72a6ba5bc0a1d570ed7ff50 Mon Sep 17 00:00:00 2001 From: William Moses Date: Sat, 1 Nov 2025 13:20:22 -0500 Subject: [PATCH 1/2] Change default value of dus_to_pad_comm to 0 --- src/CompileOptions.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 From 9c9f91934f8228d2e84275da0a7538dbfe8a3e51 Mon Sep 17 00:00:00 2001 From: William Moses Date: Sat, 1 Nov 2025 19:52:07 -0500 Subject: [PATCH 2/2] Update optimization patterns in get_optimize_comms_passes --- src/Compiler.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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}",