From 90fc9f9dc65cfcb3628f6e55b3b35b3d3b38bee8 Mon Sep 17 00:00:00 2001 From: Ricardo Vieira Date: Fri, 22 Sep 2023 19:01:56 +0200 Subject: [PATCH] Exclude local_careduce_fusion from Numba backend --- pytensor/compile/mode.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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"], + ), )