feat: new constant prop passes (#1363)

* feat: new constant prop passes [skip ci]

* chore: bump jll

Author: avik-pal

Project.toml

@@ -90,7 +90,7 @@ PythonCall = "0.9"
 Random = "1.10"
 Random123 = "1.7"
 ReactantCore = "0.1.11"
-Reactant_jll = "0.0.193"
+Reactant_jll = "0.0.194"
 ScopedValues = "1.3.0"
 Scratch = "1.2"
 Sockets = "1.10"

src/Compiler.jl

@@ -725,8 +725,6 @@ function optimization_passes(;
         "merge_consecutive_reshapes<16>",
         "transpose_is_reshape<16>",
         "zero_extent_tensor_canon<16>",
-        "chlo_inf_const_prop<16>",
-        "gamma_const_prop<16>",
         "cse_broadcast_in_dim<16>",
         "cse_slice<16>",
         "cse_transpose<16>",
@@ -752,7 +750,6 @@ function optimization_passes(;
         "min_simplify<16>",
         "or_simplify<16>",
         "xor_simplify<16>",
-        "abs_const_prop<16>",
         "negate_simplify<16>",
         "mul_simplify<16>",
         "div_simplify<16>",
@@ -766,7 +763,6 @@ function optimization_passes(;
         "sin_simplify<16>",
         "noop_slice<16>",
         "noop_reverse<16>",
-        "const_prop_through_barrier<16>",
         "slice_slice<16>",
         "shift_right_logical_simplify<16>",
         "pad_simplify<16>($max_constant_threshold)",
@@ -792,7 +788,6 @@ function optimization_passes(;
         "slice_elementwise<1>",
         "slice_pad<1>",
         "dot_reshape_dot<1>",
-        "concat_const_prop<1>($max_constant_threshold)",
         "concat_fuse<1>",
         "pad_reshape_pad<1>",
         "pad_pad<1>",
@@ -845,20 +840,14 @@ function optimization_passes(;
         "slice_dot_general<1>",
         "if_inline<1>",
         "if_to_select<1>",
-        "dynamic_update_slice_const_prop($max_constant_threshold)",
         "dynamic_gather_op_is_not_dynamic<16>",
         "divide_sqrt_to_multiply_rsqrt<16>",
         "associative_binary_op_reordering<1>",
         "transpose_broadcast_in_dim_to_broadcast_in_dim<16>",
         "scatter_indices_are_unique",
         "replace_neg_add_with_subtract",
-        "log_const_prop<1>",
-        "log_plus_one_const_prop<1>",
         "binop_const_simplify",
-        "is_finite_const_prop",
-        "not_const_prop",
         "not_select_simplify",
-        "scatter_update_computation_const_prop",
         "common_compare_expression_rewrite",
         "compare_select_simplify",
         "while_simplify<1>(1)",
@@ -945,6 +934,34 @@ function optimization_passes(;
         # "concat_to_onedim_dusslice",
     ]
 
+    # constant prop patterns
+    append!(
+        transform_passes_list,
+        [
+            "chlo_inf_const_prop<16>",
+            "gamma_const_prop<16>",
+            "abs_const_prop<16>",
+            "const_prop_through_barrier<16>",
+            "concat_const_prop<1>($max_constant_threshold)",
+            "dynamic_update_slice_const_prop($max_constant_threshold)",
+            "log_const_prop<1>",
+            "log_plus_one_const_prop<1>",
+            "is_finite_const_prop",
+            "not_const_prop",
+            "scatter_update_computation_const_prop",
+            "neg_const_prop",
+            "sqrt_const_prop",
+            "rsqrt_const_prop",
+            "cos_const_prop",
+            "sin_const_prop",
+            "exp_const_prop",
+            "expm1_const_prop",
+            "tanh_const_prop",
+            "logistic_const_prop",
+            "conj_const_prop",
+        ],
+    )
+
     if DUS_SLICE_SIMPLIFY[]
         push!(transform_passes_list, "dus_slice_simplify")
     end