We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a6b2f48 commit 566145aCopy full SHA for 566145a
pytensor/link/pytorch/dispatch/elemwise.py
@@ -32,7 +32,7 @@ def check_special_scipy(func_name):
32
# torch can handle this scalar
33
# broadcast, we'll let it.
34
def elemwise_fn(*inputs):
35
- Elemwise._check_runtime_broadcast(node, inputs)
+ # Elemwise._check_runtime_broadcast(node, inputs)
36
return base_fn(*inputs)
37
38
elif isinstance(scalar_op, ScalarLoop):
@@ -41,7 +41,7 @@ def elemwise_fn(*inputs):
41
else:
42
43
44
45
broadcast_inputs = torch.broadcast_tensors(*inputs)
46
ufunc = base_fn
47
for _ in range(broadcast_inputs[0].dim()):
0 commit comments