feat: new passes (#1415)

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.137"
+version = "0.2.138"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
@@ -90,7 +90,7 @@ PythonCall = "0.9"
 Random = "1.10"
 Random123 = "1.7"
 ReactantCore = "0.1.15"
-Reactant_jll = "0.0.209"
+Reactant_jll = "0.0.210"
 ScopedValues = "1.3.0"
 Scratch = "1.2"
 Sockets = "1.10"

src/Compiler.jl

@@ -881,6 +881,12 @@ function optimization_passes(
         "split_convolution_into_reverse_convolution",
         # TODO we want to enable but may cause an infinite compile time
         # "concat_to_onedim_dusslice",
+        # TODO expose an option to enable this
+        # "chained_multiply_to_power",
+        "power_multiply_to_power",
+        "log_simplify",
+        "neg_mul_const_simplify",
+        "neg_div_const_simplify",
     ]
 
     if !compile_options.disable_scatter_gather_optimization_passes
@@ -1129,7 +1135,15 @@ function optimization_passes(
     end
 
     if compile_options.all_finite
-        push!(transform_passes_list, "all_finite")
+        append!(
+            transform_passes_list,
+            [
+                "all_finite_is_finite",
+                "all_finite_is_inf",
+                "all_finite_is_pos_inf",
+                "all_finite_is_neg_inf",
+            ],
+        )
     end
 
     lower_transform_passes = copy(transform_passes_list)