Accept Blockwise in ifelse lift rewrite

pymc-devs · Sep 24, 2023 · 7c58661 · 7c58661
1 parent aa7c9a9
commit 7c58661
Showing 1 changed file with 1 addition and 0 deletions.
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,
 )