From ab2443d2a8e666bfc589bc625990d4d9b092a6d4 Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Mon, 2 Oct 2023 15:50:34 -0400 Subject: [PATCH] Fix linalg smoke tests (#1563) --- test/smoke_test/smoke_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py index 8ae1d1c51..3d1b6af64 100644 --- 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)