Skip to content

Commit 8e0958a

Browse files
committed
Git rid of new "output" uses
Useless since 9ba6d99
1 parent 79862c6 commit 8e0958a

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

pytensor/tensor/rewriting/linalg.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -509,8 +509,6 @@ def svd_uv_merge(fgraph, node):
509509
# Else, has to replace the s of this node with s of an SVD Op that compute_uv=False.
510510
# First, iterate to see if there is an SVD Op that can be reused.
511511
for cl, _ in fgraph.clients[x]:
512-
if cl == "output":
513-
continue
514512
if isinstance(cl.op, Blockwise) and isinstance(cl.op.core_op, SVD):
515513
if not cl.op.core_op.compute_uv:
516514
return {
@@ -529,8 +527,6 @@ def svd_uv_merge(fgraph, node):
529527
# We want rewrite if there is another one with compute_uv=True.
530528
# For this case, just reuse the `s` from the one with compute_uv=True.
531529
for cl, _ in fgraph.clients[x]:
532-
if cl == "output":
533-
continue
534530
if isinstance(cl.op, Blockwise) and isinstance(cl.op.core_op, SVD):
535531
if cl.op.core_op.compute_uv and (
536532
len(fgraph.clients[cl.outputs[0]]) > 0

0 commit comments

Comments
 (0)