feat: new constant propagation passes (#1369)

Author: avik-pal

Project.toml

@@ -1,7 +1,7 @@
 name = "Reactant"
 uuid = "3c362404-f566-11ee-1572-e11a4b42c853"
 authors = ["William Moses <[email protected]>", "Valentin Churavy <[email protected]>", "Sergio Sánchez Ramírez <[email protected]>", "Paul Berg <[email protected]>", "Avik Pal <[email protected]>", "Mosè Giordano <[email protected]>"]
-version = "0.2.126"
+version = "0.2.127"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
@@ -90,7 +90,7 @@ PythonCall = "0.9"
 Random = "1.10"
 Random123 = "1.7"
 ReactantCore = "0.1.11"
-Reactant_jll = "0.0.195"
+Reactant_jll = "0.0.196"
 ScopedValues = "1.3.0"
 Scratch = "1.2"
 Sockets = "1.10"

src/Compiler.jl

@@ -750,17 +750,13 @@ function optimization_passes(;
         "min_simplify<16>",
         "or_simplify<16>",
         "xor_simplify<16>",
-        "negate_simplify<16>",
         "mul_simplify<16>",
         "div_simplify<16>",
         "rem_simplify<16>",
         "pow_simplify<16>",
         "simplify_extend<16>",
         "simplify_wrap<16>",
         "simplify_rotate<16>",
-        "sqrt_simplify<16>",
-        "cos_simplify<16>",
-        "sin_simplify<16>",
         "noop_slice<16>",
         "noop_reverse<16>",
         "slice_slice<16>",
@@ -769,8 +765,6 @@ function optimization_passes(;
         "select_pad_to_dus<1>",
         "and_pad_pad<1>",
         "negative_pad_to_slice<16>",
-        "tanh_simplify<16>",
-        "exp_simplify<16>",
         "slice_simplify<16>",
         "convert_simplify<16>",
         "dynamic_slice_to_static<16>",
@@ -938,17 +932,14 @@ function optimization_passes(;
     append!(
         transform_passes_list,
         [
+            # unary constant propagation
             "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",
@@ -959,6 +950,34 @@ function optimization_passes(;
             "tanh_const_prop",
             "logistic_const_prop",
             "conj_const_prop",
+            "ceil_const_prop",
+            "cbrt_const_prop",
+            "real_const_prop",
+            "imag_const_prop",
+            "round_const_prop",
+            "round_nearest_even_const_prop",
+            "sign_const_prop",
+            "floor_const_prop",
+            "tan_const_prop",
+            # binary constant propagation
+            "add_const_prop",
+            "and_const_prop",
+            "atan2_const_prop",
+            "complex_const_prop",
+            "div_const_prop",
+            "max_const_prop",
+            "min_const_prop",
+            "mul_const_prop",
+            "or_const_prop",
+            "pow_const_prop",
+            "rem_const_prop",
+            "sub_const_prop",
+            "xor_const_prop",
+            # other constant propagations
+            "const_prop_through_barrier<16>",
+            "concat_const_prop<1>($max_constant_threshold)",
+            "dynamic_update_slice_const_prop($max_constant_threshold)",
+            "scatter_update_computation_const_prop",
         ],
     )