Skip to content

Commit

Permalink
Removes linalg.eigvalsh from skiplist and fixes errors that occur usi…
Browse files Browse the repository at this point in the history
…ng custom atol/rtol. (#8076)
  • Loading branch information
anishfish2 committed Sep 26, 2024
1 parent 3cd4677 commit 686f908
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions experimental/torch_xla2/test/test_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,6 @@
"linalg.cholesky_ex",
"linalg.cond",
"linalg.det",
"linalg.eigvalsh",
"linalg.householder_product",
"linalg.inv",
"linalg.inv_ex",
Expand Down Expand Up @@ -182,7 +181,7 @@
'nn.functional.feature_alpha_dropout',
}

atol_dict = {"matrix_exp": (2e-1, 2e-4), "linalg.pinv": (8e-1, 2e0), "linalg.eig": (2e0, 3e0), "linalg.eigh": (5e1, 3e0)}
atol_dict = {"matrix_exp": (2e-1, 2e-4), "linalg.pinv": (8e-1, 2e0), "linalg.eig": (2e0, 3e0), "linalg.eigh": (5e1, 3e0), "linalg.eigvalsh": (5e1, 3e0)}

def diff_output(testcase, output1, output2, rtol, atol, equal_nan=True, check_output=True):
if isinstance(output1, torch.Tensor):
Expand Down

0 comments on commit 686f908

Please sign in to comment.