Skip to content

Commit

Permalink
Remove expected failure in test_eager_transforms.py (#125883)
Browse files Browse the repository at this point in the history
Seems to be supported now

CC @tinglvv @nWEIdia @Aidyn-A

Pull Request resolved: #125883
Approved by: https://github.com/Chillee, https://github.com/Aidyn-A
  • Loading branch information
eqy authored and pytorchmergebot committed May 15, 2024
1 parent 0ca8bf4 commit 5af4b49
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions test/functorch/test_eager_transforms.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,9 +67,7 @@
from torch.testing._internal.common_utils import (
freeze_rng_state,
instantiate_parametrized_tests,
IS_ARM64,
IS_FBCODE,
IS_MACOS,
IS_WINDOWS,
markDynamoStrictTest,
parametrize,
Expand Down Expand Up @@ -5082,9 +5080,7 @@ class TestCompileTransforms(TestCase):
@skipIfRocm(msg="test leaks memory on ROCm")
# torch.compile is not supported on Windows
# Triton only supports GPU with SM70 or later.
@expectedFailureIf(
(IS_ARM64 and not IS_MACOS) or IS_WINDOWS or (TEST_CUDA and not SM70OrLater)
)
@expectedFailureIf(IS_WINDOWS or (TEST_CUDA and not SM70OrLater))
def test_compile_vmap_hessian(self, device):
# The model and inputs are a smaller version
# of code at benchmark repo:
Expand Down

0 comments on commit 5af4b49

Please sign in to comment.