已合并
fix: skip test_index_put_accumulate_large_tensor #31416
zhanw_coding创建于 3月5日
fix: skip test_index_put_accumulate_large_tensor #31416
已合并
共 2 个文件变更+3-1
| @@ -1,3 +1,4 @@ | |||
| 1 | +# Copyright (c) Meta Platforms, Inc. and affiliates | ||
| 1 | # Owner(s): ["module: tests"] | 2 | # Owner(s): ["module: tests"] |
| 2 | import warnings | 3 | import warnings |
| 3 | import unittest | 4 | import unittest |
| @@ -1651,6 +1652,7 @@ class NumpyTests(TestCase): | |||
| 1651 | torch.diagonal(kernel2).copy_(torch.square(col_max.view(4))) | 1652 | torch.diagonal(kernel2).copy_(torch.square(col_max.view(4))) |
| 1652 | self.assertEqual(kernel, kernel2) | 1653 | self.assertEqual(kernel, kernel2) |
| 1653 | 1654 | ||
| 1655 | + | ||
| 1654 | instantiate_device_type_tests(TestIndexing, globals(), only_for='privateuse1') | 1656 | instantiate_device_type_tests(TestIndexing, globals(), only_for='privateuse1') |
| 1655 | instantiate_device_type_tests(NumpyTests, globals(), only_for='privateuse1') | 1657 | instantiate_device_type_tests(NumpyTests, globals(), only_for='privateuse1') |
| 1656 | 1658 | ||
| @@ -31649,5 +31649,5 @@ | |||
| 31649 | "test_use_pinned_memory_for_d2h (__main__.TestSerialization)": ["", [""]], | 31649 | "test_use_pinned_memory_for_d2h (__main__.TestSerialization)": ["", [""]], |
| 31650 | "test_fake_autocast_fft_fftshift_npu_float32 (__main__.TestFakeTensorPRIVATEUSE1)": ["", [""]], | 31650 | "test_fake_autocast_fft_fftshift_npu_float32 (__main__.TestFakeTensorPRIVATEUSE1)": ["", [""]], |
| 31651 | "test_fake_autocast_fft_ifftshift_npu_float32 (__main__.TestFakeTensorPRIVATEUSE1)": ["", [""]], | 31651 | "test_fake_autocast_fft_ifftshift_npu_float32 (__main__.TestFakeTensorPRIVATEUSE1)": ["", [""]], |
| 31652 | - "test_index_put_accumulate_large_tensor (__main__.TestIndexing)": ["", ["910A"]] | 31652 | + "test_index_put_accumulate_large_tensor_npu (__main__.TestIndexingPRIVATEUSE1)": ["", ["910A"]] |
| 31653 | } | 31653 | } |