diff --git a/pytensor/compile/mode.py b/pytensor/compile/mode.py index dd81a81ded..6dd5496505 100644 --- a/pytensor/compile/mode.py +++ b/pytensor/compile/mode.py @@ -464,7 +464,10 @@ def clone(self, link_kwargs=None, optimizer="", **kwargs): ) NUMBA = Mode( NumbaLinker(), - RewriteDatabaseQuery(include=["fast_run"], exclude=["cxx_only", "BlasOpt"]), + RewriteDatabaseQuery( + include=["fast_run"], + exclude=["cxx_only", "BlasOpt", "local_careduce_fusion"], + ), )