Skip to content

Commit 566145a

Browse files
committed
Don't check runtime broadcast
1 parent a6b2f48 commit 566145a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pytensor/link/pytorch/dispatch/elemwise.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def check_special_scipy(func_name):
3232
# torch can handle this scalar
3333
# broadcast, we'll let it.
3434
def elemwise_fn(*inputs):
35-
Elemwise._check_runtime_broadcast(node, inputs)
35+
# Elemwise._check_runtime_broadcast(node, inputs)
3636
return base_fn(*inputs)
3737

3838
elif isinstance(scalar_op, ScalarLoop):
@@ -41,7 +41,7 @@ def elemwise_fn(*inputs):
4141
else:
4242

4343
def elemwise_fn(*inputs):
44-
Elemwise._check_runtime_broadcast(node, inputs)
44+
# Elemwise._check_runtime_broadcast(node, inputs)
4545
broadcast_inputs = torch.broadcast_tensors(*inputs)
4646
ufunc = base_fn
4747
for _ in range(broadcast_inputs[0].dim()):

0 commit comments

Comments
 (0)