diff --git a/pytensor/ifelse.py b/pytensor/ifelse.py index 4f8935ab64..1cd236f8f3 100644 --- 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, )