Fix linalg smoke tests (#1563)

pytorch · Oct 2, 2023 · ab2443d · ab2443d
1 parent aa6df6f
commit ab2443d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
@@ -179,7 +179,7 @@ def smoke_test_linalg() -> None:
     torch.dist(A, U @ torch.diag(S) @ Vh)
 
     U, S, Vh = torch.linalg.svd(A)
-    assert U.shape == A.shape and S.shape == torch.Size([3]) and Vh.shape == torch.Size([3, 3])
+    assert U.shape == torch.Size([5, 5]) and S.shape == torch.Size([3]) and Vh.shape == torch.Size([3, 3])
     torch.dist(A, U[:, :3] @ torch.diag(S) @ Vh)
 
     A = torch.randn(7, 5, 3)