feat: new passes (#1357)

* feat: new passes

* Update src/Compiler.jl

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.122"
+version = "0.2.123"
 
 [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.191"
+Reactant_jll = "0.0.192"
 ScopedValues = "1.3.0"
 Scratch = "1.2"
 Sockets = "1.10"

src/Compiler.jl

@@ -934,6 +934,9 @@ function optimization_passes(;
         "concat_elementwise",
         "reduce_reduce",
         "conj_real",
+        "select_broadcast_in_dim",
+        # XXX: seems to be busted
+        # "if_op_lift_common_ops",
         # TODO we want to enable but may cause an infinite compile time
         # "concat_to_onedim_dusslice",
     ]
@@ -1022,6 +1025,7 @@ function optimization_passes(;
                 "transpose_batch_norm_training",
                 "transpose_batch_norm_inference",
                 "transpose_batch_norm_grad",
+                "transpose_if",
             ],
         )
         if AGGRESSIVE_PROPAGATION[]