Skip to content

Commit

Permalink
Update base for Update on "Revert "Skip test_memory_format_nn_BatchNo…
Browse files Browse the repository at this point in the history
…rm2d in inductor (#125970)""


This reverts commit 0a9c6e9.


enable the test since it's fixed.

[ghstack-poisoned]
  • Loading branch information
shunting314 committed May 21, 2024
1 parent cb03e29 commit 81123f5
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions test/functorch/test_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -2130,6 +2130,10 @@ def test_extremal_numerics_nll_loss(self, device):
cotangents = torch.randn_like(result, device=device)
self._compare_jacobians_of_vjp(fn, (cotangents, input))

@unittest.skip(
"The test fails in CI but does not repro on dev machines. "
"Check https://github.com/pytorch/pytorch/pull/126586 for more details"
)
def test_extremal_numerics_l1_loss(self, device):
N, C, H, W = 3, 4, 5, 6
shapes = ((N, C), (N, C, H), (N, C, H, W))
Expand Down

0 comments on commit 81123f5

Please sign in to comment.