Exclude local_careduce_fusion from Numba backend (#450)

ricardoV94 · web-flow · commit aa7c9a93239a · 2023-09-22T12:40:51.000-05:00
diff --git 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"],
+    ),
 )
 
 

Original file line number	Diff line number	Diff line change
`@@ -464,7 +464,10 @@ def clone(self, link_kwargs=None, optimizer="", **kwargs):`
`464`	`464`	`)`
`465`	`465`	`NUMBA = Mode(`
`466`	`466`	`NumbaLinker(),`
`467`		`- RewriteDatabaseQuery(include=["fast_run"], exclude=["cxx_only", "BlasOpt"]),`
	`467`	`+ RewriteDatabaseQuery(`
	`468`	`+ include=["fast_run"],`
	`469`	`+ exclude=["cxx_only", "BlasOpt", "local_careduce_fusion"],`
	`470`	`+ ),`
`468`	`471`	`)`
`469`	`472`
`470`	`473`