Skip to content

Commit 58de169

Browse files
committed
Change py_only database placement
1 parent 8112576 commit 58de169

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pytensor/compile/mode.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -256,13 +256,14 @@ def apply(self, fgraph):
256256
# especially constant merge
257257
optdb.register("merge2", MergeOptimizer(), "fast_run", "merge", position=49)
258258

259+
optdb.register("py_only", EquilibriumDB(), "fast_compile", position=49.1)
260+
259261
optdb.register(
260262
"add_destroy_handler", AddDestroyHandler(), "fast_run", "inplace", position=49.5
261263
)
262264

263265
# final pass just to make sure
264266
optdb.register("merge3", MergeOptimizer(), "fast_run", "merge", position=100)
265-
optdb.register("py_only", EquilibriumDB(), "fast_compile", position=100)
266267

267268
_tags: Union[Tuple[str, str], Tuple]
268269

0 commit comments

Comments
 (0)