From 5a89cd2a22346a92f0f7bc70da498b729bf82da9 Mon Sep 17 00:00:00 2001 From: William Moses Date: Sat, 1 Nov 2025 19:56:44 -0500 Subject: [PATCH] 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}",