Accept Blockwise in ifelse lift rewrite

ricardoV94 · ricardoV94 · commit 7c58661bfab2 · 2023-09-24T13:43:02.000+02:00
diff --git a/pytensor/ifelse.py b/pytensor/ifelse.py
@@ -482,6 +482,7 @@ def cond_make_inplace(fgraph, node):
     at.basic.Alloc,
     at.elemwise.Elemwise,
     at.elemwise.DimShuffle,
+    at.blockwise.Blockwise,
 )
 
 

Original file line number	Diff line number	Diff line change
`@@ -482,6 +482,7 @@ def cond_make_inplace(fgraph, node):`
`482`	`482`	`at.basic.Alloc,`
`483`	`483`	`at.elemwise.Elemwise,`
`484`	`484`	`at.elemwise.DimShuffle,`
	`485`	`+ at.blockwise.Blockwise,`
`485`	`486`	`)`
`486`	`487`
`487`	`488`