已合并
fix: skip test_index_put_accumulate_large_tensor #31417
zhanw_coding创建于 3月5日
fix: skip test_index_put_accumulate_large_tensor #31417
已合并
共 2 个文件变更+4-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 | ||
| @@ -31597,5 +31597,6 @@ | |||
| 31597 | "test_fake_autocast_fft_fftshift_npu_float32 (__main__.TestFakeTensorPRIVATEUSE1)": ["", [""]], | 31597 | "test_fake_autocast_fft_fftshift_npu_float32 (__main__.TestFakeTensorPRIVATEUSE1)": ["", [""]], |
| 31598 | "test_fake_autocast_fft_ifftshift_npu_float32 (__main__.TestFakeTensorPRIVATEUSE1)": ["", [""]], | 31598 | "test_fake_autocast_fft_ifftshift_npu_float32 (__main__.TestFakeTensorPRIVATEUSE1)": ["", [""]], |
| 31599 | "test_replicate (__main__.TestDataParallel)": ["", [""]], | 31599 | "test_replicate (__main__.TestDataParallel)": ["", [""]], |
| 31600 | - "test_replicate_buffers (__main__.TestDataParallel)": ["", [""]] | 31600 | + "test_replicate_buffers (__main__.TestDataParallel)": ["", [""]], |
| 31601 | + "test_index_put_accumulate_large_tensor_npu (__main__.TestIndexingPRIVATEUSE1)": ["", ["910A"]] | ||
| 31601 | } | 31602 | } |