Don't check runtime broadcast

pymc-devs · Dec 12, 2024 · 566145a · 566145a
1 parent a6b2f48
commit 566145a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytensor/link/pytorch/dispatch/elemwise.py b/pytensor/link/pytorch/dispatch/elemwise.py
@@ -32,7 +32,7 @@ def check_special_scipy(func_name):
         # torch can handle this scalar
         # broadcast, we'll let it.
         def elemwise_fn(*inputs):
-            Elemwise._check_runtime_broadcast(node, inputs)
+            # Elemwise._check_runtime_broadcast(node, inputs)
             return base_fn(*inputs)
 
     elif isinstance(scalar_op, ScalarLoop):
@@ -41,7 +41,7 @@ def elemwise_fn(*inputs):
     else:
 
         def elemwise_fn(*inputs):
-            Elemwise._check_runtime_broadcast(node, inputs)
+            # Elemwise._check_runtime_broadcast(node, inputs)
             broadcast_inputs = torch.broadcast_tensors(*inputs)
             ufunc = base_fn
             for _ in range(broadcast_inputs[0].dim()):