From abb646f466a8c2b588e8ee72acefee2130da0173 Mon Sep 17 00:00:00 2001 From: Ricardo Vieira Date: Fri, 3 May 2024 14:41:47 +0200 Subject: [PATCH] Remove restrictive check in `local_blockwise_alloc` --- pytensor/tensor/rewriting/blockwise.py | 1 - 1 file changed, 1 deletion(-) diff --git a/pytensor/tensor/rewriting/blockwise.py b/pytensor/tensor/rewriting/blockwise.py index 22f3f9a490..0bed304c29 100644 --- a/pytensor/tensor/rewriting/blockwise.py +++ b/pytensor/tensor/rewriting/blockwise.py @@ -188,6 +188,5 @@ def local_blockwise_alloc(fgraph, node): ) for new_out in new_outs ] - assert new_outs[0].type.broadcastable == old_out_type.broadcastable copy_stack_trace(node.outputs, new_outs) return new_outs