From 5fd30089b928b16a494073e5c3144e289f6bc097 Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Sun, 21 Dec 2025 12:47:50 -0500 Subject: [PATCH 1/2] feat: new jll + new passes --- Project.toml | 2 +- src/Compiler.jl | 17 ++++++++++++++++- 2 files changed, 17 insertions(+), 2 deletions(-) diff --git a/Project.toml b/Project.toml index f9897fe9b5..69ca30fd67 100644 --- a/Project.toml +++ b/Project.toml @@ -111,7 +111,7 @@ PythonCall = "0.9.25" Random = "1.10" Random123 = "1.7" ReactantCore = "0.1.16" -Reactant_jll = "0.0.277" +Reactant_jll = "0.0.279" ScopedValues = "1.3.0" Scratch = "1.2" Sockets = "1.10" diff --git a/src/Compiler.jl b/src/Compiler.jl index 7ae53fb3ce..4ed310acce 100644 --- a/src/Compiler.jl +++ b/src/Compiler.jl @@ -921,6 +921,11 @@ function optimization_passes( "dus_dynamic_slice_simplify", "while_dus_ds_simplify", "reshape_slice_reshape", + "dot_general_remove_batch_dimensions", + "delete_dims_reduce", + "reduce_delete_dims", + "dot_general_insert_dim_contraction_simplification", + "fuse_reshape_collapse_or_expand_dims_into_reduce", ] if !is_sharded @@ -982,7 +987,10 @@ function optimization_passes( end if !compile_options.disable_loop_raising_passes - append!(transform_passes_list, ["greedy_while_loop_batch_fission"]) + append!( + transform_passes_list, + ["greedy_while_loop_batch_fission", "while_elementwise_reduction_to_reduce"], + ) end if !compile_options.disable_licm_optimization_passes @@ -1002,6 +1010,7 @@ function optimization_passes( "reduce_window_licm(0)", "reverse_licm(0)", "convolution_licm(0)", + "dynamic_slice_licm(0)", ], ) end @@ -1199,6 +1208,11 @@ function optimization_passes( "elementwise_reshape_like", ], ) + if AGGRESSIVE_PROPAGATION[] + push!(transform_passes_list, "reshape_elementwise_only_fusible(0)") + else + push!(transform_passes_list, "reshape_elementwise_only_fusible(1)") + end end if compile_options.transpose_propagate === :up @@ -1240,6 +1254,7 @@ function optimization_passes( "reorder_elementwise_and_shape_op<16>", "elementwise_all_transpose_operands_simplify", "slice_transpose", + "dynamic_slice_transpose", "einsum_transpose<1>", "slice_reshape_transpose<1>", "reduce_transpose_simplify", From 3f3bc472f6158defb5da2c85e3bfa9fd4d3f4a98 Mon Sep 17 00:00:00 2001 From: William Moses Date: Mon, 22 Dec 2025 16:51:31 -0500 Subject: [PATCH 2/2] Update Reactant_jll version to 0.0.280 --- Project.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Project.toml b/Project.toml index 69ca30fd67..43bc63d5ef 100644 --- a/Project.toml +++ b/Project.toml @@ -111,7 +111,7 @@ PythonCall = "0.9.25" Random = "1.10" Random123 = "1.7" ReactantCore = "0.1.16" -Reactant_jll = "0.0.279" +Reactant_jll = "0.0.280" ScopedValues = "1.3.0" Scratch = "1.2" Sockets = "1.10"