Skip to content

Commit 9f401bb

Browse files
committed
fix pytensor qr to be economic
1 parent e2553aa commit 9f401bb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

autoray/autoray.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2697,7 +2697,7 @@ def pytensor_wrap_qr_with_shapes(fn):
26972697
def qr_shaped(x, **kwargs):
26982698
*b, m, n = x.type.shape
26992699
k = min(m, n)
2700-
q, r = fn(x, **kwargs)
2700+
q, r = fn(x, mode="economic", **kwargs)
27012701
q = pt.specify_shape(q, (*b, m, k))
27022702
r = pt.specify_shape(r, (*b, k, n))
27032703
return q, r

0 commit comments

Comments
 (0)