Move push_elemwise_constants to post_fusion pass

Author: aseyboldt

pytensor/compile/mode.py

@@ -262,7 +262,15 @@ def apply(self, fgraph):
     "fast_run",
     "fusion",
     "local_elemwise_fusion",
-    position=49,
+    position=48.7,
+)
+
+optdb.register(
+    "post_fusion",
+    EquilibriumDB(),
+    "fast_run",
+    "fast_compile",
+    position=48.8,
 )
 
 # especially constant merge

pytensor/tensor/rewriting/elemwise.py

@@ -411,7 +411,6 @@ def push_elemwise_constants(fgraph, node):
     contained scalar op.
     """
     op = node.op
-
     if not isinstance(op, Elemwise):
         return False
 
@@ -465,7 +464,7 @@ def is_constant_scalar(x):
     )
 
 
-compile.optdb["specialize"].register(
+compile.optdb["post_fusion"].register(
     "push_elemwise_constants",
     push_elemwise_constants,
     "fast_run_numba",