Readd bcast check

pymc-devs · Jan 26, 2025 · b627684 · b627684
1 parent cb43d9a
commit b627684
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()):