已合并
【inductor】fix ci bug #31152
kkjocker创建于 2月28日
【inductor】fix ci bug #31152
已合并
共 1 个文件变更+2-0
| @@ -1,3 +1,4 @@ | |||
| 1 | import unittest | ||
| 1 | import torch | 2 | import torch |
| 2 | from torch.testing._internal.common_utils import run_tests, instantiate_parametrized_tests | 3 | from torch.testing._internal.common_utils import run_tests, instantiate_parametrized_tests |
| 3 | from testutils import TestUtils | 4 | from testutils import TestUtils |
| @@ -42,6 +43,7 @@ class TestMaskedSubblock(TestUtils): | |||
| 42 | slice_scatter_13 = torch.ops.aten.slice_scatter.default(slice_scatter_10, select_scatter_3, 2, 3199, -3199) | 43 | slice_scatter_13 = torch.ops.aten.slice_scatter.default(slice_scatter_10, select_scatter_3, 2, 3199, -3199) |
| 43 | return slice_scatter_13 | 44 | return slice_scatter_13 |
| 44 | 45 | ||
| 46 | |||
| 45 | def test_masked_subblock(self): | 47 | def test_masked_subblock(self): |
| 46 | arg0_1 = self._generate_tensor((1, 3200, 9598), 'float32') | 48 | arg0_1 = self._generate_tensor((1, 3200, 9598), 'float32') |
| 47 | std_ret = self.eager_function(arg0_1) | 49 | std_ret = self.eager_function(arg0_1) |