已合并
[sync] PR-39006: [test]skip_test_jacfwd #39336
ascend-robot创建于 6月25日
[sync] PR-39006: [test]skip_test_jacfwd #39336
已合并
共 1 个文件变更+6-0
| @@ -1,4 +1,9 @@ | |||
| 1 | # Owner(s): ["module: functorch"] | 1 | # Owner(s): ["module: functorch"] |
| 2 | +# Copyright (c) Facebook, Inc. and its affiliates. | ||
| 3 | +# All rights reserved. | ||
| 4 | +# | ||
| 5 | +# This source code is licensed under the BSD-style license found in the | ||
| 6 | +# LICENSE file in the root directory of this source tree. | ||
| 2 | 7 | ||
| 3 | import copy | 8 | import copy |
| 4 | import math | 9 | import math |
| @@ -2474,6 +2479,7 @@ class TestHessian(TestCase): | |||
| 2474 | y = torch.randn(3, device=device) | 2479 | y = torch.randn(3, device=device) |
| 2475 | self._test_against_reference(f, (x, y)) | 2480 | self._test_against_reference(f, (x, y)) |
| 2476 | 2481 | ||
| 2482 | + | ||
| 2477 | def test_jacfwd_different_levels(self, device): | 2483 | def test_jacfwd_different_levels(self, device): |
| 2478 | # Test case from: | 2484 | # Test case from: |
| 2479 | # pytorch functorch issues 597 | 2485 | # pytorch functorch issues 597 |