-
Notifications
You must be signed in to change notification settings - Fork 68
[release/2.1] fix test_vmapvjpvjp and skip test_profiler_experimental_tree #1444
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
@@ -829,6 +829,8 @@ def fn(inp, *args, **kwargs): | |||
{torch.float32: tol(atol=2e-03, rtol=2e-02)}), | |||
tol1('svd', | |||
{torch.float32: tol(atol=1e-03, rtol=5e-04)}), | |||
tol1('linalg.householder_product', | |||
{torch.float32: tol(atol=5e-04, rtol=5e-04)}), |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Tolerance values inspired by
pytorch/test/functorch/test_ops.py
Line 955 in e22361a
tol1('linalg.householder_product', |
@@ -249,6 +249,7 @@ def assertTreesMatch(self, actual: str, expected: str, allow_failure: bool = Fal | |||
else: | |||
raise | |||
|
|||
@skipIfRocm |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Skipping for now, proper fix to be done in https://github.com/ROCm/frameworks-internal/issues/8556
(cherry picked from commit 7e96391)
(cherry picked from commit 7e96391)
(cherry picked from commit 7e96391)
…_tree (#1460) * fix test_vmapvjpvjp and skip test_profiler_experimental_tree (#1444) (cherry picked from commit 7e96391) * remove trailing spaces --------- Co-authored-by: Ramana Cherukuri <[email protected]>
…_tree (#1460) * fix test_vmapvjpvjp and skip test_profiler_experimental_tree (#1444) (cherry picked from commit 7e96391) * remove trailing spaces --------- Co-authored-by: Ramana Cherukuri <[email protected]> (cherry picked from commit 0766b9c)
…tree (#1494) * Skip test__int_mm in 6.0 (cherry picked from commit bf4c478) * [release/2.3] fix test_vmapvjpvjp and skip test_profiler_experimental_tree (#1460) * fix test_vmapvjpvjp and skip test_profiler_experimental_tree (#1444) (cherry picked from commit 7e96391) * remove trailing spaces --------- Co-authored-by: Ramana Cherukuri <[email protected]> (cherry picked from commit 0766b9c) * Reformat test_float8_basics for current rocm support (#1415) (cherry picked from commit cb0e9ad) --------- Co-authored-by: Jack Taylor <[email protected]> Co-authored-by: Andres Lugo <[email protected]>
…tree (#1494) * Skip test__int_mm in 6.0 (cherry picked from commit bf4c478) * [release/2.3] fix test_vmapvjpvjp and skip test_profiler_experimental_tree (#1460) * fix test_vmapvjpvjp and skip test_profiler_experimental_tree (#1444) (cherry picked from commit 7e96391) * remove trailing spaces --------- Co-authored-by: Ramana Cherukuri <[email protected]> (cherry picked from commit 0766b9c) * Reformat test_float8_basics for current rocm support (#1415) (cherry picked from commit cb0e9ad) --------- Co-authored-by: Jack Taylor <[email protected]> Co-authored-by: Andres Lugo <[email protected]>
…tree (#1494) * Skip test__int_mm in 6.0 (cherry picked from commit bf4c478) * [release/2.3] fix test_vmapvjpvjp and skip test_profiler_experimental_tree (#1460) * fix test_vmapvjpvjp and skip test_profiler_experimental_tree (#1444) (cherry picked from commit 7e96391) * remove trailing spaces --------- Co-authored-by: Ramana Cherukuri <[email protected]> (cherry picked from commit 0766b9c) * Reformat test_float8_basics for current rocm support (#1415) (cherry picked from commit cb0e9ad) --------- Co-authored-by: Jack Taylor <[email protected]> Co-authored-by: Andres Lugo <[email protected]> (cherry picked from commit 9521966)
…tree (#1494) * Skip test__int_mm in 6.0 (cherry picked from commit bf4c478) * [release/2.3] fix test_vmapvjpvjp and skip test_profiler_experimental_tree (#1460) * fix test_vmapvjpvjp and skip test_profiler_experimental_tree (#1444) (cherry picked from commit 7e96391) * remove trailing spaces --------- Co-authored-by: Ramana Cherukuri <[email protected]> (cherry picked from commit 0766b9c) * Reformat test_float8_basics for current rocm support (#1415) (cherry picked from commit cb0e9ad) --------- Co-authored-by: Jack Taylor <[email protected]> Co-authored-by: Andres Lugo <[email protected]>
Skipping the following test
test/profiler/test_profiler_tree.py:test_profiler_experimental_tree
test/profiler/test_profiler_tree.py:test_profiler_experimental_tree_with_memory
Fixing
test/functorch/test_ops.py:test_vmapvjpvjp_linalg_householder_product_cuda_float32
by adjusting the TOL value.