Star843
1302
代码
代码
Issues1384
Pull Requests2263
流水线
项目讨论
Wiki
分析
Star843
1302
已合并
Skip failing torch official TestCase tests and disable unsupported test cases #32139
haiyan8创建于 3月21日
Skip failing torch official TestCase tests and disable unsupported test cases #32139
已合并
haiyan8创建于 3月21日
从yuhaiyan/pytorchcherry-pick-mr-32138-1774073465134-auto合入到Ascend/pytorchv2.10.0
共 19 个文件变更+586-40
Mtest/contrib/test_bidirectional_lstm.py+1-0
@@ -20,6 +20,7 @@ class TestBidirectionalLstm(TestCase):
20 input_grad = input1.grad.cpu()20 input_grad = input1.grad.cpu()
21 return output.detach().cpu(), input_grad.cpu()21 return output.detach().cpu(), input_grad.cpu()
22 22 
23+ @unittest.skip("Temporarily skipping")
23 def test_bidirectional_lstm(self):24 def test_bidirectional_lstm(self):
24 np.random.seed(123)25 np.random.seed(123)
25 data1 = np.random.randn(2, 2, 8)26 data1 = np.random.randn(2, 2, 8)
Mtest/custom_ops/test_float_status.py+28-26
@@ -1,26 +1,28 @@
1-import torch1+import unittest
2-import torch_npu2+import torch
3- 3+import torch_npu
4-from torch_npu.testing.testcase import TestCase, run_tests4+ 
5- 5+from torch_npu.testing.testcase import TestCase, run_tests
6- 6+ 
7-class TestFloatStatus(TestCase):7+ 
8- 8+class TestFloatStatus(TestCase):
9- def test_float_status(self, device="npu"):9+ 
10- float_tensor = torch.tensor([40000.0], dtype=torch.float16).npu()10+ @unittest.skip("Temporarily skipping")
11- float_tensor = float_tensor + float_tensor11+ def test_float_status(self, device="npu"):
12- 12+ float_tensor = torch.tensor([40000.0], dtype=torch.float16).npu()
13- input1 = torch.zeros(8).npu()13+ float_tensor = float_tensor + float_tensor
14- float_status = torch_npu.npu_alloc_float_status(input1)14+ 
15- local_float_status = torch_npu.npu_get_float_status(float_status)15+ input1 = torch.zeros(8).npu()
16- 16+ float_status = torch_npu.npu_alloc_float_status(input1)
17- self.assertTrue(local_float_status.cpu()[0] != 0)17+ local_float_status = torch_npu.npu_get_float_status(float_status)
18- cleared_float_status = torch_npu.npu_clear_float_status(local_float_status)18+ 
19- input1 = torch.zeros(8).npu()19+ self.assertTrue(local_float_status.cpu()[0] != 0)
20- float_status = torch_npu.npu_alloc_float_status(input1)20+ cleared_float_status = torch_npu.npu_clear_float_status(local_float_status)
21- local_float_status = torch_npu.npu_get_float_status(float_status)21+ input1 = torch.zeros(8).npu()
22- self.assertTrue(local_float_status.cpu()[0] == 0)22+ float_status = torch_npu.npu_alloc_float_status(input1)
23- 23+ local_float_status = torch_npu.npu_get_float_status(float_status)
24- 24+ self.assertTrue(local_float_status.cpu()[0] == 0)
25-if __name__ == "__main__":25+ 
26- run_tests()26+ 
27+if __name__ == "__main__":
28+ run_tests()
Mtest/custom_ops/test_functionalization.py+5-0
@@ -1,3 +1,4 @@
1+import unittest
1from contextlib import nullcontext2from contextlib import nullcontext
2import numpy as np3import numpy as np
3import torch4import torch
@@ -98,6 +99,7 @@ class TestFunctionalization(TestCase):
98 self.assertEqual(out_ref_, out_functional_)99 self.assertEqual(out_ref_, out_functional_)
99 self.assertEqual(out_ref_, out_reinplace_)100 self.assertEqual(out_ref_, out_reinplace_)
100 101 
102+ @unittest.skip("Temporarily skipping")
101 def test_scatter_update(self):103 def test_scatter_update(self):
102 def f(iself, indices, updates):104 def f(iself, indices, updates):
103 return torch.ops.npu.scatter_update_(iself, indices, updates, -2)105 return torch.ops.npu.scatter_update_(iself, indices, updates, -2)
@@ -119,6 +121,7 @@ def forward(self, arg0_1, arg1_1, arg2_1):
119 self.assert_functionalization(f, in_self, in_indices, in_updates)121 self.assert_functionalization(f, in_self, in_indices, in_updates)
120 122 
121 @SupportedDevices(['Ascend910B'])123 @SupportedDevices(['Ascend910B'])
124+ @unittest.skip("Temporarily skipping")
122 def test_npu_quant_scatter(self):125 def test_npu_quant_scatter(self):
123 def f(fake_var, fake_indices, fake_updates, fake_quant_scales):126 def f(fake_var, fake_indices, fake_updates, fake_quant_scales):
124 return torch.ops.npu.npu_quant_scatter_(fake_var, fake_indices, fake_updates, fake_quant_scales,127 return torch.ops.npu.npu_quant_scatter_(fake_var, fake_indices, fake_updates, fake_quant_scales,
@@ -146,6 +149,7 @@ def forward(self, arg0_1, arg1_1, arg2_1, arg3_1):
146 149 
147 self.assert_functionalization(f, in_var, in_indices, in_updates, in_quant_scales)150 self.assert_functionalization(f, in_var, in_indices, in_updates, in_quant_scales)
148 151 
152+ @unittest.skip("Temporarily skipping")
149 def test_npu_scatter_nd_update(self):153 def test_npu_scatter_nd_update(self):
150 def f(var, indices, updates):154 def f(var, indices, updates):
151 return torch_npu.npu_scatter_nd_update_(var, indices, updates)155 return torch_npu.npu_scatter_nd_update_(var, indices, updates)
@@ -170,6 +174,7 @@ def forward(self, arg0_1, arg1_1, arg2_1):
170 174 
171 self.assert_functionalization(f, var, indices, updates)175 self.assert_functionalization(f, var, indices, updates)
172 176 
177+ @unittest.skip("Temporarily skipping")
173 def test_npu_silu_functionalize(self):178 def test_npu_silu_functionalize(self):
174 @impl(m, "npu_silu")179 @impl(m, "npu_silu")
175 def npu_silu(self_):180 def npu_silu(self_):
Mtest/distributed/_tensor/test_random_ops.py+3-0
@@ -1,3 +1,4 @@
1+import unittest
1import itertools2import itertools
2from unittest import skip3from unittest import skip
3 4 
@@ -78,6 +79,7 @@ class DistTensorRandomInitTest(DTensorTestBase):
78class DistTensorRandomOpTest(DTensorTestBase):79class DistTensorRandomOpTest(DTensorTestBase):
79 @skipIfUnsupportMultiNPU(4)80 @skipIfUnsupportMultiNPU(4)
80 @with_comms81 @with_comms
82+ @unittest.skip("Temporarily skipping")
81 def test_rng_tracker_init(self):83 def test_rng_tracker_init(self):
82 torch.npu.manual_seed(self.rank)84 torch.npu.manual_seed(self.rank)
83 object_list = [torch.npu.initial_seed()]85 object_list = [torch.npu.initial_seed()]
@@ -93,6 +95,7 @@ class DistTensorRandomOpTest(DTensorTestBase):
93 95 
94 @skipIfUnsupportMultiNPU(4)96 @skipIfUnsupportMultiNPU(4)
95 @with_comms97 @with_comms
98+ @unittest.skip("Temporarily skipping")
96 def test_manual_seed(self):99 def test_manual_seed(self):
97 device_mesh = DeviceMesh(self.device_type, torch.arange(self.world_size))100 device_mesh = DeviceMesh(self.device_type, torch.arange(self.world_size))
98 manual_seed(1234, device_mesh)101 manual_seed(1234, device_mesh)
Mtest/distributed/tensor/test_math_ops.py+2-0
@@ -1,3 +1,4 @@
1+import unittest
1import itertools2import itertools
2 3 
3import torch4import torch
@@ -91,6 +92,7 @@ class TestMathOps(NPUDTensorTestBase):
91 @SupportedDevices(['Ascend910B'])92 @SupportedDevices(['Ascend910B'])
92 @skipIfUnsupportMultiNPU(2)93 @skipIfUnsupportMultiNPU(2)
93 @with_comms94 @with_comms
95+ @unittest.skip("Temporarily skipping")
94 def test_npu_add_rms_norm_forward(self):96 def test_npu_add_rms_norm_forward(self):
95 device_mesh = self.build_device_mesh()97 device_mesh = self.build_device_mesh()
96 98 
Mtest/distributed/tensor/test_matrix_ops.py+5-0
@@ -1,3 +1,4 @@
1+import unittest
1import itertools2import itertools
2import operator3import operator
3from functools import reduce4from functools import reduce
@@ -840,6 +841,7 @@ class TestMatmul(NPUDTensorTestBase):
840 @SupportedDevices(['Ascend910B'])841 @SupportedDevices(['Ascend910B'])
841 @skipIfUnsupportMultiNPU(2)842 @skipIfUnsupportMultiNPU(2)
842 @with_comms843 @with_comms
844+ @unittest.skip("Temporarily skipping")
843 def test_matmul_sharding_strategy_n_n(self):845 def test_matmul_sharding_strategy_n_n(self):
844 device_mesh = self.build_device_mesh()846 device_mesh = self.build_device_mesh()
845 shape1 = (8,)847 shape1 = (8,)
@@ -851,6 +853,7 @@ class TestMatmul(NPUDTensorTestBase):
851 @SupportedDevices(['Ascend910B'])853 @SupportedDevices(['Ascend910B'])
852 @skipIfUnsupportMultiNPU(2)854 @skipIfUnsupportMultiNPU(2)
853 @with_comms855 @with_comms
856+ @unittest.skip("Temporarily skipping")
854 def test_matmul_sharding_strategy_nm_m1(self):857 def test_matmul_sharding_strategy_nm_m1(self):
855 device_mesh = self.build_device_mesh()858 device_mesh = self.build_device_mesh()
856 shape1 = (8, 4)859 shape1 = (8, 4)
@@ -888,6 +891,7 @@ class TestMatmul(NPUDTensorTestBase):
888 @SupportedDevices(['Ascend910B'])891 @SupportedDevices(['Ascend910B'])
889 @skipIfUnsupportMultiNPU(2)892 @skipIfUnsupportMultiNPU(2)
890 @with_comms893 @with_comms
894+ @unittest.skip("Temporarily skipping")
891 def test_matmul_sharding_strategy_xnm_mk(self):895 def test_matmul_sharding_strategy_xnm_mk(self):
892 device_mesh = self.build_device_mesh()896 device_mesh = self.build_device_mesh()
893 shape1 = (8, 4, 8)897 shape1 = (8, 4, 8)
@@ -908,6 +912,7 @@ class TestMatmul(NPUDTensorTestBase):
908 @SupportedDevices(['Ascend910B'])912 @SupportedDevices(['Ascend910B'])
909 @skipIfUnsupportMultiNPU(2)913 @skipIfUnsupportMultiNPU(2)
910 @with_comms914 @with_comms
915+ @unittest.skip("Temporarily skipping")
911 def test_matmul_sharding_strategy_nm_xmk(self):916 def test_matmul_sharding_strategy_nm_xmk(self):
912 device_mesh = self.build_device_mesh()917 device_mesh = self.build_device_mesh()
913 shape1 = (4, 8)918 shape1 = (4, 8)
Mtest/distributed/test_all_to_all_single.py+2-0
@@ -226,12 +226,14 @@ class HcclAlltoAllSingleTest(TestCase):
226 p.join(4)226 p.join(4)
227 227 
228 @skipIfUnsupportMultiNPU(4)228 @skipIfUnsupportMultiNPU(4)
229+ @unittest.skip("Temporarily skipping")
229 def test_alltoall_single_4p_dist(self):230 def test_alltoall_single_4p_dist(self):
230 self._test_multiprocess_4p(231 self._test_multiprocess_4p(
231 HcclAlltoAllSingleTest._test_alltoall_single_4p,232 HcclAlltoAllSingleTest._test_alltoall_single_4p,
232 HcclAlltoAllSingleTest._init_dist_hccl)233 HcclAlltoAllSingleTest._init_dist_hccl)
233 234 
234 @skipIfUnsupportMultiNPU(4)235 @skipIfUnsupportMultiNPU(4)
236+ @unittest.skip("Temporarily skipping")
235 def test_alltoall_single_4p_size_dist(self):237 def test_alltoall_single_4p_size_dist(self):
236 self._test_multiprocess_4p(238 self._test_multiprocess_4p(
237 HcclAlltoAllSingleTest._test_alltoall_single_4p_size,239 HcclAlltoAllSingleTest._test_alltoall_single_4p_size,
Mtest/distributed/test_reduce_scatter_tensor.py+1-0
@@ -156,6 +156,7 @@ class HcclReduceScatterTensorTest(HcclReduceScatterTestBase):
156 HcclReduceScatterTensorTest._init_dist_hccl, input_list, world_size)156 HcclReduceScatterTensorTest._init_dist_hccl, input_list, world_size)
157 157 
158 @skipIfUnsupportMultiNPU(2)158 @skipIfUnsupportMultiNPU(2)
159+ @unittest.skip("Temporarily skipping")
159 def test_reduce_scatter_tensor_uneven_avg(self):160 def test_reduce_scatter_tensor_uneven_avg(self):
160 ranks = [2]161 ranks = [2]
161 dtype_list = [np.float32, np.float16]162 dtype_list = [np.float32, np.float16]
Mtest/distributed/test_send_recv.py+4-0
@@ -214,6 +214,7 @@ class HcclSendRecvDistTest(TestCase):
214 p.join()214 p.join()
215 self.assertEqual(p.exitcode, 0, "subprocess exit with abnormal code.")215 self.assertEqual(p.exitcode, 0, "subprocess exit with abnormal code.")
216 216 
217+ @unittest.skip("Temporarily disabled")
217 @skipIfUnsupportMultiNPU(2)218 @skipIfUnsupportMultiNPU(2)
218 def test_send_recv_hccl_dist(self):219 def test_send_recv_hccl_dist(self):
219 self._test_multiprocess(220 self._test_multiprocess(
@@ -228,6 +229,7 @@ class HcclSendRecvDistTest(TestCase):
228 torch.randn(31, 31),229 torch.randn(31, 31),
229 HcclSendRecvDistTest._init_dist_hccl)230 HcclSendRecvDistTest._init_dist_hccl)
230 231
232+ @unittest.skip("Temporarily disabled")
231 @skipIfUnsupportMultiNPU(4)233 @skipIfUnsupportMultiNPU(4)
232 def test_send_recv_hccl_dist_with_p2p(self):234 def test_send_recv_hccl_dist_with_p2p(self):
233 with patch.dict(os.environ, {"P2P_HCCL_BUFFSIZE": "20"}):235 with patch.dict(os.environ, {"P2P_HCCL_BUFFSIZE": "20"}):
@@ -250,6 +252,7 @@ class HcclSendRecvDistTest(TestCase):
250 torch.randn(2, 2),252 torch.randn(2, 2),
251 HcclSendRecvDistTest._init_dist_hccl)253 HcclSendRecvDistTest._init_dist_hccl)
252 254 
255+ @unittest.skip("Temporarily disabled")
253 @skipIfUnsupportMultiNPU(2)256 @skipIfUnsupportMultiNPU(2)
254 def test_send_recv_hccl_group(self):257 def test_send_recv_hccl_group(self):
255 self._test_multiprocess(258 self._test_multiprocess(
@@ -269,6 +272,7 @@ class HcclSendRecvDistTest(TestCase):
269 torch.randn(2, 2),272 torch.randn(2, 2),
270 HcclSendRecvDistTest._init_pg_hccl)273 HcclSendRecvDistTest._init_pg_hccl)
271 274 
275+ @unittest.skip("Temporarily disabled")
272 @skipIfUnsupportMultiNPU(2)276 @skipIfUnsupportMultiNPU(2)
273 def test_send_recv_hccl_bool(self):277 def test_send_recv_hccl_bool(self):
274 self._test_multiprocess(278 self._test_multiprocess(
Mtest/nn/test_dropout_layers.py+2-0
@@ -1,3 +1,4 @@
1+import unittest
1import torch2import torch
2import torch.nn as nn3import torch.nn as nn
3import torch_npu4import torch_npu
@@ -30,6 +31,7 @@ class TestDropoutLayers(TestCase):
30 output = m(input1)31 output = m(input1)
31 self.assertEqual(output is not None, True)32 self.assertEqual(output is not None, True)
32 33 
34+ @unittest.skip("Temporarily skipping")
33 def test_native_dropout(self):35 def test_native_dropout(self):
34 for train in [True, False]:36 for train in [True, False]:
35 for p in [0.0, 1.0, 0.5]:37 for p in [0.0, 1.0, 0.5]:
Mtest/nn/test_loss_functions.py+16-0
@@ -37,6 +37,7 @@ class TestLossFunctions(TestCase):
37 output.backward()37 output.backward()
38 self.assertEqual(input1.grad is not None, True)38 self.assertEqual(input1.grad is not None, True)
39 39 
40+ @unittest.skip("Temporarily skipping")
40 def test_CTCLoss(self):41 def test_CTCLoss(self):
41 T = 50 # Input sequence length42 T = 50 # Input sequence length
42 C = 20 # Number of classes (including blank)43 C = 20 # Number of classes (including blank)
@@ -67,6 +68,7 @@ class TestLossFunctions(TestCase):
67 self.assertRtolEqual(loss.detach().cpu().numpy(), closs.detach().numpy())68 self.assertRtolEqual(loss.detach().cpu().numpy(), closs.detach().numpy())
68 self.assertRtolEqual(input1.grad.cpu().numpy(), cinput.grad.numpy())69 self.assertRtolEqual(input1.grad.cpu().numpy(), cinput.grad.numpy())
69 70 
71+ @unittest.skip("Temporarily skipping")
70 def test_NLLLoss(self):72 def test_NLLLoss(self):
71 m = nn.LogSoftmax(dim=1)73 m = nn.LogSoftmax(dim=1)
72 # input1 is of size N x C = 3 x 574 # input1 is of size N x C = 3 x 5
@@ -120,6 +122,7 @@ class TestLossFunctions(TestCase):
120 output.backward()122 output.backward()
121 self.assertEqual(input1.grad is not None, True)123 self.assertEqual(input1.grad is not None, True)
122 124 
125+ @unittest.skip("Temporarily skipping")
123 def test_BCEWithLogitsLoss(self):126 def test_BCEWithLogitsLoss(self):
124 loss = nn.BCEWithLogitsLoss().npu()127 loss = nn.BCEWithLogitsLoss().npu()
125 input1 = torch.randn(3).npu()128 input1 = torch.randn(3).npu()
@@ -198,6 +201,7 @@ class TestLossFunctions(TestCase):
198 output.backward()201 output.backward()
199 self.assertEqual(anchor.grad is not None, True)202 self.assertEqual(anchor.grad is not None, True)
200 203 
204+ @unittest.skip("Temporarily skipping")
201 def test_binary_cross_entropy(self):205 def test_binary_cross_entropy(self):
202 input1 = torch.rand(2, 3)206 input1 = torch.rand(2, 3)
203 target = torch.rand(2, 3)207 target = torch.rand(2, 3)
@@ -209,6 +213,7 @@ class TestLossFunctions(TestCase):
209 213 
210 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())214 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())
211 215 
216+ @unittest.skip("Temporarily skipping")
212 def test_binary_cross_entropy_with_logits(self):217 def test_binary_cross_entropy_with_logits(self):
213 input1 = torch.rand(2, 3)218 input1 = torch.rand(2, 3)
214 target = torch.rand(2, 3)219 target = torch.rand(2, 3)
@@ -220,6 +225,7 @@ class TestLossFunctions(TestCase):
220 225 
221 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())226 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())
222 227 
228+ @unittest.skip("Temporarily skipping")
223 def test_poisson_nll_loss(self):229 def test_poisson_nll_loss(self):
224 input1 = torch.rand(2, 3)230 input1 = torch.rand(2, 3)
225 target = torch.rand(2, 3)231 target = torch.rand(2, 3)
@@ -231,6 +237,7 @@ class TestLossFunctions(TestCase):
231 237 
232 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())238 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())
233 239 
240+ @unittest.skip("Temporarily skipping")
234 def test_cosine_embedding_loss(self):241 def test_cosine_embedding_loss(self):
235 input1 = torch.randn(2, 3)242 input1 = torch.randn(2, 3)
236 input2 = torch.randn(2, 3)243 input2 = torch.randn(2, 3)
@@ -260,6 +267,7 @@ class TestLossFunctions(TestCase):
260 267 
261 self.assertRtolEqual(cpu_output.numpy(), npu_output.cpu().numpy())268 self.assertRtolEqual(cpu_output.numpy(), npu_output.cpu().numpy())
262 269 
270+ @unittest.skip("Temporarily skipping")
263 def test_hinge_embedding_loss(self):271 def test_hinge_embedding_loss(self):
264 input1 = torch.randn(5, 3)272 input1 = torch.randn(5, 3)
265 targets = torch.randint(1, 20, (5, 3), dtype=torch.long)273 targets = torch.randint(1, 20, (5, 3), dtype=torch.long)
@@ -285,6 +293,7 @@ class TestLossFunctions(TestCase):
285 293 
286 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())294 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())
287 295 
296+ @unittest.skip("Temporarily skipping")
288 def test_l1_loss(self):297 def test_l1_loss(self):
289 input1 = torch.randn(5, 3)298 input1 = torch.randn(5, 3)
290 targets = torch.randn(5, 3)299 targets = torch.randn(5, 3)
@@ -297,6 +306,7 @@ class TestLossFunctions(TestCase):
297 306 
298 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())307 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())
299 308 
309+ @unittest.skip("Temporarily skipping")
300 def test_mse_loss(self):310 def test_mse_loss(self):
301 input1 = torch.randn(5, 3)311 input1 = torch.randn(5, 3)
302 targets = torch.randn(5, 3)312 targets = torch.randn(5, 3)
@@ -309,6 +319,7 @@ class TestLossFunctions(TestCase):
309 319 
310 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())320 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())
311 321 
322+ @unittest.skip("Temporarily skipping")
312 def test_margin_ranking_loss(self):323 def test_margin_ranking_loss(self):
313 input1 = torch.randn(3)324 input1 = torch.randn(3)
314 input2 = torch.randn(3)325 input2 = torch.randn(3)
@@ -323,6 +334,7 @@ class TestLossFunctions(TestCase):
323 334 
324 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())335 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())
325 336 
337+ @unittest.skip("Temporarily skipping")
326 def test_multilabel_margin_loss(self):338 def test_multilabel_margin_loss(self):
327 input1 = torch.Tensor([[0.1, 0.2, 0.4, 0.8], [0.1, 0.2, 0.4, 0.8]]).to(torch.float32)339 input1 = torch.Tensor([[0.1, 0.2, 0.4, 0.8], [0.1, 0.2, 0.4, 0.8]]).to(torch.float32)
328 targets = torch.Tensor([[3, 0, -1, 1], [0, 1, 3, -1]]).to(torch.int64)340 targets = torch.Tensor([[3, 0, -1, 1], [0, 1, 3, -1]]).to(torch.int64)
@@ -335,6 +347,7 @@ class TestLossFunctions(TestCase):
335 347 
336 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())348 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())
337 349 
350+ @unittest.skip("Temporarily skipping")
338 def test_nll_loss(self):351 def test_nll_loss(self):
339 input1 = torch.Tensor([[0.1, 0.2, 0.4, 0.8], [0.1, 0.2, 0.4, 0.8]]).to(torch.float32)352 input1 = torch.Tensor([[0.1, 0.2, 0.4, 0.8], [0.1, 0.2, 0.4, 0.8]]).to(torch.float32)
340 targets = torch.Tensor([3, 2]).to(torch.int64)353 targets = torch.Tensor([3, 2]).to(torch.int64)
@@ -347,6 +360,7 @@ class TestLossFunctions(TestCase):
347 360 
348 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())361 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())
349 362 
363+ @unittest.skip("Temporarily skipping")
350 def test_smooth_l1_loss(self):364 def test_smooth_l1_loss(self):
351 input1 = torch.Tensor([[0.1, 0.2, 0.4, 0.8], [0.1, 0.2, 0.4, 0.8]]).to(torch.float32)365 input1 = torch.Tensor([[0.1, 0.2, 0.4, 0.8], [0.1, 0.2, 0.4, 0.8]]).to(torch.float32)
352 targets = torch.Tensor([[3, 0, -1, 1], [0, 1, 3, -1]]).to(torch.float32)366 targets = torch.Tensor([[3, 0, -1, 1], [0, 1, 3, -1]]).to(torch.float32)
@@ -359,6 +373,7 @@ class TestLossFunctions(TestCase):
359 373 
360 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())374 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())
361 375 
376+ @unittest.skip("Temporarily skipping")
362 def test_soft_margin_loss(self):377 def test_soft_margin_loss(self):
363 input1 = torch.Tensor([[0.1, 0.2, 0.4, 0.8], [0.1, 0.2, 0.4, 0.8]]).to(torch.float32)378 input1 = torch.Tensor([[0.1, 0.2, 0.4, 0.8], [0.1, 0.2, 0.4, 0.8]]).to(torch.float32)
364 targets = torch.Tensor([[3, 0, -1, 1], [0, 1, 3, -1]]).to(torch.float32)379 targets = torch.Tensor([[3, 0, -1, 1], [0, 1, 3, -1]]).to(torch.float32)
@@ -371,6 +386,7 @@ class TestLossFunctions(TestCase):
371 386 
372 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())387 self.assertRtolEqual(cpu_output.detach().numpy(), npu_output.detach().cpu().numpy())
373 388 
389+ @unittest.skip("Temporarily skipping")
374 def test_triplet_margin_loss(self):390 def test_triplet_margin_loss(self):
375 input1 = torch.randn(5, 10)391 input1 = torch.randn(5, 10)
376 input2 = torch.randn(5, 10)392 input2 = torch.randn(5, 10)
Mtest/nn/test_recurrent_layers.py+7-1
@@ -1,3 +1,4 @@
1+import unittest
1import torch2import torch
2import torch.nn as nn3import torch.nn as nn
3import torch_npu4import torch_npu
@@ -6,13 +7,15 @@ from torch_npu.testing.testcase import TestCase, run_tests
6 7 
7 8 
8class TestRecurrentLayers(TestCase):9class TestRecurrentLayers(TestCase):
10+ @unittest.skip("Temporarily skipping")
9 def test_RNN(self):11 def test_RNN(self):
10 input1 = torch.randn(5, 3, 10).npu()12 input1 = torch.randn(5, 3, 10).npu()
11 h0 = torch.randn(2, 3, 20).npu()13 h0 = torch.randn(2, 3, 20).npu()
12 rnn = nn.RNN(10, 20, 2).npu()14 rnn = nn.RNN(10, 20, 2).npu()
13 output, hn = rnn(input1, h0)15 output, hn = rnn(input1, h0)
14 self.assertEqual(output is not None, True)16 self.assertEqual(output is not None, True)
15- 17+
18+ @unittest.skip("Temporarily skipping")
16 def test_LSTM(self):19 def test_LSTM(self):
17 input1 = torch.randn(5, 3, 10).npu()20 input1 = torch.randn(5, 3, 10).npu()
18 h0 = torch.randn(2, 3, 20).npu()21 h0 = torch.randn(2, 3, 20).npu()
@@ -28,6 +31,7 @@ class TestRecurrentLayers(TestCase):
28 output, hn = rnn(input1, h0)31 output, hn = rnn(input1, h0)
29 self.assertEqual(output is not None, True)32 self.assertEqual(output is not None, True)
30 33 
34+ @unittest.skip("Temporarily skipping")
31 def test_RNNCell(self):35 def test_RNNCell(self):
32 input1 = torch.randn(6, 3, 10).npu()36 input1 = torch.randn(6, 3, 10).npu()
33 hx = torch.randn(3, 20).npu()37 hx = torch.randn(3, 20).npu()
@@ -37,6 +41,7 @@ class TestRecurrentLayers(TestCase):
37 hx = rnn(input1[i], hx)41 hx = rnn(input1[i], hx)
38 output.append(hx)42 output.append(hx)
39 43 
44+ @unittest.skip("Temporarily skipping")
40 def test_LSTMCell(self):45 def test_LSTMCell(self):
41 input1 = torch.randn(2, 3, 10).npu()46 input1 = torch.randn(2, 3, 10).npu()
42 hx = torch.randn(3, 20).npu()47 hx = torch.randn(3, 20).npu()
@@ -49,6 +54,7 @@ class TestRecurrentLayers(TestCase):
49 output = torch.stack(output, dim=0)54 output = torch.stack(output, dim=0)
50 self.assertEqual(output is not None, True)55 self.assertEqual(output is not None, True)
51 56 
57+ @unittest.skip("Temporarily skipping")
52 def test_GRUCell(self):58 def test_GRUCell(self):
53 input1 = torch.randn(6, 3, 10).npu()59 input1 = torch.randn(6, 3, 10).npu()
54 hx = torch.randn(3, 20).npu()60 hx = torch.randn(3, 20).npu()
Mtest/onnx/test_wrapper_onnx_ops.py+1-0
@@ -1148,6 +1148,7 @@ class TestOnnxOps(TestCase):
1148 1148 
1149 @unittest.skip("skip now")1149 @unittest.skip("skip now")
1150 @SupportedDevices(['Ascend910B'])1150 @SupportedDevices(['Ascend910B'])
1151+ @unittest.skip("Temporarily skipping")
1151 def test_wrapper_npu_moe_compute_expert_tokens(self):1152 def test_wrapper_npu_moe_compute_expert_tokens(self):
1152 class Model(torch.nn.Module):1153 class Model(torch.nn.Module):
1153 def __init__(self):1154 def __init__(self):
Mtest/optim/test_fused_optimizers.py+4-0
@@ -122,6 +122,7 @@ class TestFusedOptim(TestCase):
122 if p.grad is not None:122 if p.grad is not None:
123 self.assertRtolEqual(p, p_clone, prec=1e-3)123 self.assertRtolEqual(p, p_clone, prec=1e-3)
124 124 
125+ @unittest.skip("Temporarily skipping")
125 def test_step_3rd_optims(self):126 def test_step_3rd_optims(self):
126 optim_cases = self._create_optimizer_cases(all_cases=True)127 optim_cases = self._create_optimizer_cases(all_cases=True)
127 num_iters = 10128 num_iters = 10
@@ -137,6 +138,7 @@ class TestFusedOptim(TestCase):
137 if p.grad is not None:138 if p.grad is not None:
138 self.assertRtolEqual(p.sum().item(), self.third_optim_baseline[fused_opt_obj][i])139 self.assertRtolEqual(p.sum().item(), self.third_optim_baseline[fused_opt_obj][i])
139 140 
141+ @unittest.skip("Temporarily skipping")
140 def test_unscale(self):142 def test_unscale(self):
141 model = self._create_simple_model()143 model = self._create_simple_model()
142 input_tensor = torch.rand(3, 1, 24, 24).to("npu:0")144 input_tensor = torch.rand(3, 1, 24, 24).to("npu:0")
@@ -159,6 +161,7 @@ class TestFusedOptim(TestCase):
159 if p.grad is not None:161 if p.grad is not None:
160 self.assertEqual(grads_before_unscale[p] / 128, p.grad)162 self.assertEqual(grads_before_unscale[p] / 128, p.grad)
161 163
164+ @unittest.skip("Temporarily skipping")
162 def test_simple_model_train_dynamic(self):165 def test_simple_model_train_dynamic(self):
163 model = self._create_simple_model()166 model = self._create_simple_model()
164 optim_cases = self._create_optimizer_cases()167 optim_cases = self._create_optimizer_cases()
@@ -189,6 +192,7 @@ class TestFusedOptim(TestCase):
189 scaler_fused.update()192 scaler_fused.update()
190 self.assertRtolEqual(loss, loss_fused)193 self.assertRtolEqual(loss, loss_fused)
191 194
195+ @unittest.skip("Temporarily skipping")
192 def test_simple_model_train_static(self):196 def test_simple_model_train_static(self):
193 model = self._create_simple_model()197 model = self._create_simple_model()
194 optim_cases = self._create_optimizer_cases()198 optim_cases = self._create_optimizer_cases()
Mtest/profiler/analysis/prof_bean/test_memory_usage_bean.py+2-0
@@ -1,3 +1,4 @@
1+import unittest
1import struct2import struct
2import random3import random
3 4 
@@ -45,6 +46,7 @@ class TestMemoryUsageBean(TestCase):
45 46 
46 return sample47 return sample
47 48 
49+ @unittest.skip("Temporarily skipping")
48 def test_property(self):50 def test_property(self):
49 for sample in self.test_cases:51 for sample in self.test_cases:
50 memory_usage_bean = MemoryUseBean(sample)52 memory_usage_bean = MemoryUseBean(sample)
Mtest/test_sparse_csr.py+2-0
@@ -1,3 +1,4 @@
1+import unittest
1import torch2import torch
2from torch_npu.testing.common_utils import SupportedDevices3from torch_npu.testing.common_utils import SupportedDevices
3from torch_npu.testing.testcase import TestCase, run_tests4from torch_npu.testing.testcase import TestCase, run_tests
@@ -102,6 +103,7 @@ class TestSparseCsr(TestCase):
102 self.assertRtolEqual(sparse_csr_npu.values(), res_cpu.values())103 self.assertRtolEqual(sparse_csr_npu.values(), res_cpu.values())
103 104 
104 @SupportedDevices(['Ascend910B'])105 @SupportedDevices(['Ascend910B'])
106+ @unittest.skip("Temporarily skipping")
105 def test_sparse_angle(self):107 def test_sparse_angle(self):
106 for dtype in self.float_dtypes:108 for dtype in self.float_dtypes:
107 sparse_csr_npu, sparse_csr_cpu = self._create_sparse_csr_tensor(dtype)109 sparse_csr_npu, sparse_csr_cpu = self._create_sparse_csr_tensor(dtype)
Mtest/trans_contiguous/test_single_reshape_copy_to_contiguous.py+2-0
@@ -1,3 +1,4 @@
1+import unittest
1import os2import os
2import torch3import torch
3import numpy as np4import numpy as np
@@ -172,6 +173,7 @@ class SingleViewCopyToContiguous(TestCase):
172 cpu_out2 = cpu_input[1:10, :, :].clone()173 cpu_out2 = cpu_input[1:10, :, :].clone()
173 self.assertRtolEqual(npu_out2.to("cpu").numpy(), cpu_out2.numpy())174 self.assertRtolEqual(npu_out2.to("cpu").numpy(), cpu_out2.numpy())
174 175 
176+ @unittest.skip("Temporarily skipping")
175 def test_select_at_first_axis_to_single_element_tensor_copy(self, device="npu"):177 def test_select_at_first_axis_to_single_element_tensor_copy(self, device="npu"):
176 dtype_list5 = [torch.float32]178 dtype_list5 = [torch.float32]
177 format_list5 = [2, 3, 29]179 format_list5 = [2, 3, 29]
Mtest/unsupported_test_cases/.pytorch-disabled-tests.json+496-13
@@ -31184,8 +31184,8 @@
31184 "test_nan_policy_propagate__refs_prod_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31184 "test_nan_policy_propagate__refs_prod_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31185 "test_nan_policy_propagate__refs_sum_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31185 "test_nan_policy_propagate__refs_sum_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31186 "test_nan_policy_propagate__refs_sum_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31186 "test_nan_policy_propagate__refs_sum_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31187- "test_nan_policy_propagate_linalg_vector_norm_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31187+ "test_nan_policy_propagate_linalg_vector_norm_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31188- "test_nan_policy_propagate_linalg_vector_norm_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31188+ "test_nan_policy_propagate_linalg_vector_norm_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31189 "test_nan_policy_propagate_masked_mean_npu_float64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31189 "test_nan_policy_propagate_masked_mean_npu_float64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31190 "test_nan_policy_propagate_masked_prod_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31190 "test_nan_policy_propagate_masked_prod_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31191 "test_nan_policy_propagate_masked_prod_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31191 "test_nan_policy_propagate_masked_prod_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
@@ -31197,36 +31197,36 @@
31197 "test_nan_policy_propagate_prod_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31197 "test_nan_policy_propagate_prod_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31198 "test_nan_policy_propagate_sum_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31198 "test_nan_policy_propagate_sum_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31199 "test_nan_policy_propagate_sum_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31199 "test_nan_policy_propagate_sum_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31200- "test_noncontiguous_all__refs_all_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31200+ "test_noncontiguous_all__refs_all_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31201- "test_noncontiguous_expanded__refs_all_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31201+ "test_noncontiguous_expanded__refs_all_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31202 "test_noncontiguous_expanded__refs_any_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31202 "test_noncontiguous_expanded__refs_any_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31203 "test_noncontiguous_expanded_count_nonzero_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31203 "test_noncontiguous_expanded_count_nonzero_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31204- "test_noncontiguous_innermost__refs_all_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31204+ "test_noncontiguous_innermost__refs_all_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31205 "test_noncontiguous_innermost__refs_any_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31205 "test_noncontiguous_innermost__refs_any_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31206 "test_noncontiguous_innermost__refs_count_nonzero_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31206 "test_noncontiguous_innermost__refs_count_nonzero_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31207 "test_noncontiguous_innermost_count_nonzero_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31207 "test_noncontiguous_innermost_count_nonzero_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31208- "test_noncontiguous_outermost__refs_all_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31208+ "test_noncontiguous_outermost__refs_all_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31209- "test_noncontiguous_outermost__refs_all_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31209+ "test_noncontiguous_outermost__refs_all_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31210 "test_noncontiguous_outermost__refs_any_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31210 "test_noncontiguous_outermost__refs_any_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31211 "test_noncontiguous_outermost__refs_any_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31211 "test_noncontiguous_outermost__refs_any_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31212 "test_noncontiguous_outermost__refs_count_nonzero_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31212 "test_noncontiguous_outermost__refs_count_nonzero_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31213 "test_noncontiguous_outermost__refs_count_nonzero_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31213 "test_noncontiguous_outermost__refs_count_nonzero_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31214 "test_noncontiguous_outermost_count_nonzero_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31214 "test_noncontiguous_outermost_count_nonzero_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31215 "test_noncontiguous_outermost_count_nonzero_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31215 "test_noncontiguous_outermost_count_nonzero_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31216- "test_noncontiguous_transposed__refs_all_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31216+ "test_noncontiguous_transposed__refs_all_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31217- "test_noncontiguous_transposed__refs_all_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31217+ "test_noncontiguous_transposed__refs_all_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31218 "test_noncontiguous_transposed__refs_any_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31218 "test_noncontiguous_transposed__refs_any_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31219 "test_noncontiguous_transposed__refs_any_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31219 "test_noncontiguous_transposed__refs_any_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31220 "test_noncontiguous_transposed_count_nonzero_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31220 "test_noncontiguous_transposed_count_nonzero_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31221 "test_noncontiguous_transposed_count_nonzero_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31221 "test_noncontiguous_transposed_count_nonzero_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31222- "test_ref_scalar_input__refs_all_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31222+ "test_ref_scalar_input__refs_all_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31223- "test_ref_scalar_input__refs_all_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31223+ "test_ref_scalar_input__refs_all_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31224 "test_ref_scalar_input__refs_any_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31224 "test_ref_scalar_input__refs_any_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31225 "test_ref_scalar_input__refs_any_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31225 "test_ref_scalar_input__refs_any_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31226 "test_ref_scalar_input_count_nonzero_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31226 "test_ref_scalar_input_count_nonzero_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31227 "test_ref_scalar_input_count_nonzero_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31227 "test_ref_scalar_input_count_nonzero_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31228- "test_result_dtype__refs_all_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31228+ "test_result_dtype__refs_all_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31229- "test_result_dtype__refs_all_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31229+ "test_result_dtype__refs_all_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31230 "test_result_dtype__refs_any_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31230 "test_result_dtype__refs_any_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31231 "test_result_dtype__refs_any_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31231 "test_result_dtype__refs_any_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
31232 "test_result_dtype__refs_prod_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],31232 "test_result_dtype__refs_prod_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", ["910A"]],
@@ -31616,6 +31616,489 @@
31616 "test_replicate (__main__.TestDataParallel)": ["", [""]],31616 "test_replicate (__main__.TestDataParallel)": ["", [""]],
31617 "test_replicate_buffers (__main__.TestDataParallel)": ["", [""]],31617 "test_replicate_buffers (__main__.TestDataParallel)": ["", [""]],
31618 "test_index_put_accumulate_large_tensor_npu (__main__.TestIndexingPRIVATEUSE1)": ["", ["910A"]],31618 "test_index_put_accumulate_large_tensor_npu (__main__.TestIndexingPRIVATEUSE1)": ["", ["910A"]],
31619+ "test_serialization_array_with_empty (__main__.TestNpu)": ["", [""]],
31620+ "test_fn_fwgrad_bwgrad__unsafe_masked_index_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31621+ "test_fn_fwgrad_bwgrad__unsafe_masked_index_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31622+ "test_fn_fwgrad_bwgrad__unsafe_masked_index_put_accumulate_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31623+ "test_fn_fwgrad_bwgrad__unsafe_masked_index_put_accumulate_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31624+ "test_fn_fwgrad_bwgrad_abs_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31625+ "test_fn_fwgrad_bwgrad_alias_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31626+ "test_fn_fwgrad_bwgrad_alias_copy_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31627+ "test_fn_fwgrad_bwgrad_expand_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31628+ "test_fn_fwgrad_bwgrad_expand_copy_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31629+ "test_fn_fwgrad_bwgrad_logsumexp_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31630+ "test_fn_fwgrad_bwgrad_masked_logsumexp_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31631+ "test_fn_fwgrad_bwgrad_nn_functional_channel_shuffle_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31632+ "test_fn_fwgrad_bwgrad_nn_functional_channel_shuffle_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31633+ "test_fn_fwgrad_bwgrad_permute_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31634+ "test_fn_fwgrad_bwgrad_permute_copy_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31635+ "test_fn_fwgrad_bwgrad_squeeze_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31636+ "test_fn_fwgrad_bwgrad_squeeze_copy_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31637+ "test_fn_fwgrad_bwgrad_t_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31638+ "test_fn_fwgrad_bwgrad_t_copy_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31639+ "test_fn_fwgrad_bwgrad_torch_ops_aten__safe_softmax_default_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31640+ "test_fn_fwgrad_bwgrad_transpose_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31641+ "test_fn_fwgrad_bwgrad_transpose_copy_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31642+ "test_fn_fwgrad_bwgrad_unbind_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31643+ "test_fn_fwgrad_bwgrad_unbind_copy_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31644+ "test_fn_fwgrad_bwgrad_unsqueeze_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31645+ "test_fn_fwgrad_bwgrad_unsqueeze_copy_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31646+ "test_fn_fwgrad_bwgrad_view_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31647+ "test_forward_mode_AD__unsafe_masked_index_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31648+ "test_forward_mode_AD__unsafe_masked_index_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31649+ "test_forward_mode_AD__unsafe_masked_index_put_accumulate_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31650+ "test_forward_mode_AD__unsafe_masked_index_put_accumulate_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31651+ "test_forward_mode_AD_alias_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31652+ "test_forward_mode_AD_alias_copy_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31653+ "test_forward_mode_AD_expand_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31654+ "test_forward_mode_AD_expand_copy_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31655+ "test_forward_mode_AD_logsumexp_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31656+ "test_forward_mode_AD_masked_logsumexp_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31657+ "test_forward_mode_AD_nn_functional_channel_shuffle_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31658+ "test_forward_mode_AD_nn_functional_channel_shuffle_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31659+ "test_forward_mode_AD_pca_lowrank_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31660+ "test_forward_mode_AD_permute_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31661+ "test_forward_mode_AD_permute_copy_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31662+ "test_forward_mode_AD_squeeze_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31663+ "test_forward_mode_AD_squeeze_copy_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31664+ "test_forward_mode_AD_svd_lowrank_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31665+ "test_forward_mode_AD_t_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31666+ "test_forward_mode_AD_t_copy_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31667+ "test_forward_mode_AD_torch_ops_aten__safe_softmax_default_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31668+ "test_forward_mode_AD_transpose_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31669+ "test_forward_mode_AD_transpose_copy_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31670+ "test_forward_mode_AD_unbind_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31671+ "test_forward_mode_AD_unbind_copy_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31672+ "test_forward_mode_AD_unsqueeze_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31673+ "test_forward_mode_AD_unsqueeze_copy_npu_float64 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31674+ "test_forward_mode_AD_view_copy_npu_complex128 (__main__.TestFwdGradientsPRIVATEUSE1)": ["", [""]],
31675+ "test_dim_default_keepdim_linalg_vector_norm_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31676+ "test_dim_default_linalg_vector_norm_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31677+ "test_dim_multi_keepdim_linalg_vector_norm_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31678+ "test_dim_multi_linalg_vector_norm_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31679+ "test_dim_multi_unsorted_keepdim_linalg_vector_norm_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31680+ "test_dim_multi_unsorted_linalg_vector_norm_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31681+ "test_dim_none_keepdim_linalg_vector_norm_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31682+ "test_dim_none_linalg_vector_norm_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31683+ "test_dim_offbounds__refs_all_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31684+ "test_dim_offbounds__refs_any_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31685+ "test_dim_single_keepdim_linalg_vector_norm_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31686+ "test_dim_single_linalg_vector_norm_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31687+ "test_empty_tensor_empty_slice_linalg_vector_norm_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31688+ "test_empty_tensor_nonempty_slice_linalg_vector_norm_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31689+ "test_identity_linalg_vector_norm_npu_float16 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31690+ "test_identity_linalg_vector_norm_npu_float32 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31691+ "test_nan_policy_propagate_linalg_vector_norm_npu_float32 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31692+ "test_nan_policy_propagate_linalg_vector_norm_npu_float64 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31693+ "test_nan_policy_propagate_masked_logsumexp_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31694+ "test_nan_policy_propagate_masked_logsumexp_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31695+ "test_noncontiguous_all_linalg_vector_norm_npu_float16 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31696+ "test_noncontiguous_all_masked_logsumexp_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31697+ "test_noncontiguous_all_masked_logsumexp_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31698+ "test_noncontiguous_expanded_linalg_vector_norm_npu_float16 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31699+ "test_noncontiguous_expanded_linalg_vector_norm_npu_float32 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31700+ "test_noncontiguous_expanded_masked_logsumexp_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31701+ "test_noncontiguous_expanded_masked_logsumexp_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31702+ "test_noncontiguous_innermost_linalg_vector_norm_npu_float16 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31703+ "test_noncontiguous_innermost_linalg_vector_norm_npu_float32 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31704+ "test_noncontiguous_innermost_masked_logsumexp_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31705+ "test_noncontiguous_innermost_masked_logsumexp_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31706+ "test_noncontiguous_outermost_linalg_vector_norm_npu_float16 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31707+ "test_noncontiguous_outermost_linalg_vector_norm_npu_float32 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31708+ "test_noncontiguous_outermost_masked_logsumexp_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31709+ "test_noncontiguous_outermost_masked_logsumexp_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31710+ "test_noncontiguous_transposed_linalg_vector_norm_npu_float16 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31711+ "test_noncontiguous_transposed_linalg_vector_norm_npu_float32 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31712+ "test_noncontiguous_transposed_masked_logsumexp_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31713+ "test_noncontiguous_transposed_masked_logsumexp_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31714+ "test_result_dtype_linalg_vector_norm_npu_bfloat16 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31715+ "test_result_dtype_linalg_vector_norm_npu_float16 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31716+ "test_result_dtype_linalg_vector_norm_npu_float32 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31717+ "test_result_dtype_linalg_vector_norm_npu_float64 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31718+ "test_result_dtype_masked_logsumexp_npu_complex128 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31719+ "test_result_dtype_masked_logsumexp_npu_complex64 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31720+ "test_cat_out_fast_path_dim0_dim1_npu_uint16 (__main__.TestTensorCreationPRIVATEUSE1)": ["", [""]],
31721+ "test_cat_out_fast_path_dim0_dim1_npu_uint32 (__main__.TestTensorCreationPRIVATEUSE1)": ["", [""]],
31722+ "test_cat_out_fast_path_dim0_dim1_npu_uint64 (__main__.TestTensorCreationPRIVATEUSE1)": ["", [""]],
31723+ "test_clamp_type_promotion_npu_float32_bool_float64 (__main__.TestTypePromotionPRIVATEUSE1)": ["", [""]],
31724+ "test_clamp_type_promotion_npu_float32_float32_float64 (__main__.TestTypePromotionPRIVATEUSE1)": ["", [""]],
31725+ "test_clamp_type_promotion_npu_float32_int32_float64 (__main__.TestTypePromotionPRIVATEUSE1)": ["", [""]],
31726+ "test_fn_grad__unsafe_masked_index_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31727+ "test_fn_grad__unsafe_masked_index_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31728+ "test_fn_grad__unsafe_masked_index_put_accumulate_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31729+ "test_fn_grad__unsafe_masked_index_put_accumulate_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31730+ "test_fn_grad_alias_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31731+ "test_fn_grad_alias_copy_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31732+ "test_fn_grad_cond_simple_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31733+ "test_fn_grad_expand_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31734+ "test_fn_grad_expand_copy_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31735+ "test_fn_grad_invoke_quant_packed_simple_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31736+ "test_fn_grad_invoke_quant_simple_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31737+ "test_fn_grad_invoke_subgraph_simple_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31738+ "test_fn_grad_logsumexp_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31739+ "test_fn_grad_masked_logsumexp_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31740+ "test_fn_grad_nn_functional_channel_shuffle_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31741+ "test_fn_grad_nn_functional_channel_shuffle_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31742+ "test_fn_grad_pca_lowrank_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31743+ "test_fn_grad_permute_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31744+ "test_fn_grad_permute_copy_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31745+ "test_fn_grad_squeeze_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31746+ "test_fn_grad_squeeze_copy_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31747+ "test_fn_grad_svd_lowrank_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31748+ "test_fn_grad_t_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31749+ "test_fn_grad_t_copy_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31750+ "test_fn_grad_torch_ops_aten__safe_softmax_default_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31751+ "test_fn_grad_transpose_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31752+ "test_fn_grad_transpose_copy_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31753+ "test_fn_grad_unbind_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31754+ "test_fn_grad_unbind_copy_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31755+ "test_fn_grad_unsqueeze_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31756+ "test_fn_grad_unsqueeze_copy_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31757+ "test_fn_grad_view_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31758+ "test_fn_gradgrad__unsafe_masked_index_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31759+ "test_fn_gradgrad__unsafe_masked_index_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31760+ "test_fn_gradgrad__unsafe_masked_index_put_accumulate_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31761+ "test_fn_gradgrad__unsafe_masked_index_put_accumulate_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31762+ "test_fn_gradgrad_alias_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31763+ "test_fn_gradgrad_alias_copy_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31764+ "test_fn_gradgrad_expand_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31765+ "test_fn_gradgrad_expand_copy_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31766+ "test_fn_gradgrad_logsumexp_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31767+ "test_fn_gradgrad_masked_logsumexp_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31768+ "test_fn_gradgrad_nn_functional_channel_shuffle_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31769+ "test_fn_gradgrad_nn_functional_channel_shuffle_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31770+ "test_fn_gradgrad_pca_lowrank_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31771+ "test_fn_gradgrad_permute_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31772+ "test_fn_gradgrad_permute_copy_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31773+ "test_fn_gradgrad_squeeze_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31774+ "test_fn_gradgrad_squeeze_copy_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31775+ "test_fn_gradgrad_svd_lowrank_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31776+ "test_fn_gradgrad_t_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31777+ "test_fn_gradgrad_t_copy_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31778+ "test_fn_gradgrad_torch_ops_aten__safe_softmax_default_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31779+ "test_fn_gradgrad_transpose_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31780+ "test_fn_gradgrad_transpose_copy_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31781+ "test_fn_gradgrad_unbind_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31782+ "test_fn_gradgrad_unbind_copy_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31783+ "test_fn_gradgrad_unsqueeze_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31784+ "test_fn_gradgrad_unsqueeze_copy_npu_float64 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31785+ "test_fn_gradgrad_view_copy_npu_complex128 (__main__.TestBwdGradientsPRIVATEUSE1)": ["", [""]],
31786+ "test_conv (__main__.TestMetalRewritePass)": ["", [""]],
31787+ "test_as_sparse_gradcheck_SparseBSC_masked_fast_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31788+ "test_as_sparse_gradcheck_SparseBSC_masked_slow_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31789+ "test_as_sparse_gradcheck_SparseBSC_nonmasked_fast_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31790+ "test_as_sparse_gradcheck_SparseBSC_nonmasked_slow_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31791+ "test_as_sparse_gradcheck_SparseBSR_masked_fast_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31792+ "test_as_sparse_gradcheck_SparseBSR_masked_slow_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31793+ "test_as_sparse_gradcheck_SparseBSR_nonmasked_fast_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31794+ "test_as_sparse_gradcheck_SparseBSR_nonmasked_slow_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31795+ "test_as_sparse_gradcheck_SparseCOO_masked_fast_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31796+ "test_as_sparse_gradcheck_SparseCOO_masked_slow_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31797+ "test_as_sparse_gradcheck_SparseCOO_nonmasked_fast_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31798+ "test_as_sparse_gradcheck_SparseCOO_nonmasked_slow_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31799+ "test_as_sparse_gradcheck_SparseCSC_masked_fast_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31800+ "test_as_sparse_gradcheck_SparseCSC_masked_slow_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31801+ "test_as_sparse_gradcheck_SparseCSC_nonmasked_fast_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31802+ "test_as_sparse_gradcheck_SparseCSC_nonmasked_slow_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31803+ "test_as_sparse_gradcheck_SparseCSR_masked_fast_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31804+ "test_as_sparse_gradcheck_SparseCSR_masked_slow_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31805+ "test_as_sparse_gradcheck_SparseCSR_nonmasked_fast_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31806+ "test_as_sparse_gradcheck_SparseCSR_nonmasked_slow_npu (__main__.TestSparseAnyPRIVATEUSE1)": ["", [""]],
31807+ "test_dim_constraints_solve_full (__main__.TestDimConstraints)": ["", [""]],
31808+ "test_size_expressions (__main__.TestPySymInt)": ["", [""]],
31809+ "test_method_fn_bitwise_and_first_type_int_second_type_float (__main__.TestSymNumberMagicMethods)": ["", [""]],
31810+ "test_method_fn_bitwise_or_first_type_int_second_type_float (__main__.TestSymNumberMagicMethods)": ["", [""]],
31811+ "test_method_fn_float_pow_first_type_int_second_type_float (__main__.TestSymNumberMagicMethods)": ["", [""]],
31812+ "test_method_fn_float_pow_first_type_int_second_type_int (__main__.TestSymNumberMagicMethods)": ["", [""]],
31813+ "test_method_fn_float_truediv_first_type_int_second_type_float (__main__.TestSymNumberMagicMethods)": ["", [""]],
31814+ "test_method_fn_float_truediv_first_type_int_second_type_int (__main__.TestSymNumberMagicMethods)": ["", [""]],
31815+ "test_method_fn_int_floordiv_first_type_int_second_type_float (__main__.TestSymNumberMagicMethods)": ["", [""]],
31816+ "test_method_fn_int_floordiv_first_type_int_second_type_int (__main__.TestSymNumberMagicMethods)": ["", [""]],
31817+ "test_method_fn_int_truediv_first_type_int_second_type_float (__main__.TestSymNumberMagicMethods)": ["", [""]],
31818+ "test_method_fn_int_truediv_first_type_int_second_type_int (__main__.TestSymNumberMagicMethods)": ["", [""]],
31819+ "test_method_fn_pow_by_natural_first_type_int_second_type_float (__main__.TestSymNumberMagicMethods)": ["", [""]],
31820+ "test_method_fn_pow_by_natural_first_type_int_second_type_int (__main__.TestSymNumberMagicMethods)": ["", [""]],
31821+ "test_custom (__main__.TestFlopCounter)": ["", [""]],
31822+ "test_out_dtype_non_functional (__main__.TestOutDtypeOp)": ["", [""]],
31823+ "test_out_dtype_op_functional (__main__.TestOutDtypeOp)": ["", [""]],
31824+ "test_variant_consistency_jit__chunk_cat_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31825+ "test_variant_consistency_jit__unsafe_masked_index_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31826+ "test_variant_consistency_jit__unsafe_masked_index_put_accumulate_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31827+ "test_variant_consistency_jit_abs_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31828+ "test_variant_consistency_jit_alias_copy_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31829+ "test_variant_consistency_jit_all_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31830+ "test_variant_consistency_jit_angle_npu_float32 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31831+ "test_variant_consistency_jit_as_strided_copy_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31832+ "test_variant_consistency_jit_baddbmm_npu_float32 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31833+ "test_variant_consistency_jit_cholesky_npu_float32 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31834+ "test_variant_consistency_jit_dist_npu_float32 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31835+ "test_variant_consistency_jit_expand_copy_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31836+ "test_variant_consistency_jit_linalg_cholesky_ex_npu_float32 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31837+ "test_variant_consistency_jit_linalg_eigh_npu_float32 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31838+ "test_variant_consistency_jit_linalg_eigvalsh_npu_float32 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31839+ "test_variant_consistency_jit_linalg_inv_ex_npu_float32 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31840+ "test_variant_consistency_jit_linalg_pinv_npu_float32 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31841+ "test_variant_consistency_jit_linalg_qr_npu_float32 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31842+ "test_variant_consistency_jit_logical_not_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31843+ "test_variant_consistency_jit_logsumexp_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31844+ "test_variant_consistency_jit_nn_functional_channel_shuffle_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31845+ "test_variant_consistency_jit_nn_functional_multilabel_soft_margin_loss_npu_float32 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31846+ "test_variant_consistency_jit_nn_functional_normalize_npu_float32 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31847+ "test_variant_consistency_jit_nn_functional_rms_norm_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31848+ "test_variant_consistency_jit_permute_copy_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31849+ "test_variant_consistency_jit_pinverse_npu_float32 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31850+ "test_variant_consistency_jit_split_with_sizes_copy_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31851+ "test_variant_consistency_jit_squeeze_copy_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31852+ "test_variant_consistency_jit_t_copy_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31853+ "test_variant_consistency_jit_transpose_copy_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31854+ "test_variant_consistency_jit_unbind_copy_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31855+ "test_variant_consistency_jit_unfold_copy_npu_float32 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31856+ "test_variant_consistency_jit_unfold_npu_float32 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31857+ "test_variant_consistency_jit_unsqueeze_copy_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31858+ "test_variant_consistency_jit_view_copy_npu_complex64 (__main__.TestJitPRIVATEUSE1)": ["", [""]],
31859+ "test_cdist_NPU_backward_npu (__main__.TestTorchDeviceTypePRIVATEUSE1)": ["", [""]],
31860+ "test_cdist_empty_npu (__main__.TestTorchDeviceTypePRIVATEUSE1)": ["", [""]],
31861+ "test_cdist_euclidean_large_npu (__main__.TestTorchDeviceTypePRIVATEUSE1)": ["", [""]],
31862+ "test_cdist_large_npu (__main__.TestTorchDeviceTypePRIVATEUSE1)": ["", [""]],
31863+ "test_cdist_non_contiguous_batch_npu (__main__.TestTorchDeviceTypePRIVATEUSE1)": ["", [""]],
31864+ "test_cdist_non_contiguous_npu (__main__.TestTorchDeviceTypePRIVATEUSE1)": ["", [""]],
31865+ "test_cdist_norm_batch_npu (__main__.TestTorchDeviceTypePRIVATEUSE1)": ["", [""]],
31866+ "test_cdist_norm_npu (__main__.TestTorchDeviceTypePRIVATEUSE1)": ["", [""]],
31867+ "test_cdist_same_inputs_npu (__main__.TestTorchDeviceTypePRIVATEUSE1)": ["", [""]],
31868+ "test_device_guard_npu (__main__.TestTorchDeviceTypePRIVATEUSE1)": ["", [""]],
31869+ "test_dist_npu (__main__.TestTorchDeviceTypePRIVATEUSE1)": ["", [""]],
31870+ "setUpClass (__main__.TestPluggableAllocator)": ["", [""]],
31871+ "test_pruning_serialization_model (__main__.TestPruningNN)": ["", [""]],
31872+ "test_aten_clone_does_not_raise_warning_of_lack_of_memory_format (__main__.TestFxToOnnx)": ["", [""]],
31873+ "test_aten_div_no_opmath_type_promotion (__main__.TestFxToOnnx)": ["", [""]],
31874+ "test_exported_program_as_input_with_model_signature (__main__.TestFxToOnnx)": ["", [""]],
31875+ "test_exported_program_input_with_custom_fx_tracer (__main__.TestFxToOnnx)": ["", [""]],
31876+ "test_fake_tensor_mode_simple (__main__.TestFxToOnnx)": ["", [""]],
31877+ "test_float8_support_torch_float8_e4m3fn (__main__.TestFxToOnnx)": ["", [""]],
31878+ "test_float8_support_torch_float8_e4m3fnuz (__main__.TestFxToOnnx)": ["", [""]],
31879+ "test_float8_support_torch_float8_e5m2 (__main__.TestFxToOnnx)": ["", [""]],
31880+ "test_float8_support_torch_float8_e5m2fnuz (__main__.TestFxToOnnx)": ["", [""]],
31881+ "test_mnist_exported_with_no_warnings_get_attr_node_in_op_level_debug (__main__.TestFxToOnnx)": ["", [""]],
31882+ "test_mnist_exported_with_no_warnings_optional_inputs (__main__.TestFxToOnnx)": ["", [""]],
31883+ "test_no_warnings_on_complex_dtype_in_op_level_debug (__main__.TestFxToOnnx)": ["", [""]],
31884+ "test_perfect_match_on_sequence_and_bool_attributes (__main__.TestFxToOnnx)": ["", [""]],
31885+ "test_unsupported_function_schema_raises_diagnostic_warning_when_found_nearest_match (__main__.TestFxToOnnx)": ["", [""]],
31886+ "test_unsupported_indices_fake_tensor_generated_with_op_level_debug (__main__.TestFxToOnnx)": ["", [""]],
31887+ "test_onnx_dynamo_export_raises_when_model_contains_unsupported_fx_nodes (__main__.TestFxPasses)": ["", [""]],
31888+ "test_register_op (__main__.TestCustomAutogradFunction)": ["", [""]],
31889+ "test_broadcast_matmul (__main__.TestONNXShapeInference)": ["", [""]],
31890+ "test_cast (__main__.TestONNXShapeInference)": ["", [""]],
31891+ "test_constant_of_shape (__main__.TestONNXShapeInference)": ["", [""]],
31892+ "test_constant_of_shape_dynamic (__main__.TestONNXShapeInference)": ["", [""]],
31893+ "test_constant_of_shape_static (__main__.TestONNXShapeInference)": ["", [""]],
31894+ "test_expand (__main__.TestONNXShapeInference)": ["", [""]],
31895+ "test_gather_dynamic_index (__main__.TestONNXShapeInference)": ["", [""]],
31896+ "test_gather_scalar_index (__main__.TestONNXShapeInference)": ["", [""]],
31897+ "test_pad (__main__.TestONNXShapeInference)": ["", [""]],
31898+ "test_pad_with_dynamic_input_shape (__main__.TestONNXShapeInference)": ["", [""]],
31899+ "test_pad_with_dynamic_pad_size (__main__.TestONNXShapeInference)": ["", [""]],
31900+ "test_proceeding_nodes_use_prim_pack_padded_output_dtype_correctly (__main__.TestONNXShapeInference)": ["", [""]],
31901+ "test_reduce_prod_with_axes (__main__.TestONNXShapeInference)": ["", [""]],
31902+ "test_reduce_prod_without_axes (__main__.TestONNXShapeInference)": ["", [""]],
31903+ "test_reshape (__main__.TestONNXShapeInference)": ["", [""]],
31904+ "test_reshape_allowzero (__main__.TestONNXShapeInference)": ["", [""]],
31905+ "test_reshape_symbolic (__main__.TestONNXShapeInference)": ["", [""]],
31906+ "test_resize (__main__.TestONNXShapeInference)": ["", [""]],
31907+ "test_resize_after_concat (__main__.TestONNXShapeInference)": ["", [""]],
31908+ "test_slice (__main__.TestONNXShapeInference)": ["", [""]],
31909+ "test_slice_with_dynamic_start_index (__main__.TestONNXShapeInference)": ["", [""]],
31910+ "test_squeeze_after_dynamic_if (__main__.TestONNXShapeInference)": ["", [""]],
31911+ "test_torchscript (__main__.TestVerifyCorrectness)": ["", [""]],
31912+ "test_wrapper_subclass_guards_on_inner_tensor (__main__.SubclassTests)": ["", [""]],
31913+ "test_inputs_to_compiled_fn_are_views (__main__.TestNestedTensor)": ["", [""]],
31914+ "test_unbind (__main__.TestNestedTensor)": ["", [""]],
31915+ "test_dynamo_timed_profiling_isolated (__main__.DynamoProfilerTests)": ["", [""]],
31916+ "test_aot_graphs_effects (__main__.LoggingTests)": ["", [""]],
31917+ "test_loop_ordering (__main__.LoggingTests)": ["", [""]],
31918+ "test_skipfiles_inlinelist (__main__.TraceRuleTests)": ["", [""]],
31919+ "test_complex_state_mutation_in_intermediary_hooks_same_on_inductor (__main__.HooksTests)": ["", [""]],
31920+ "test_complex_state_mutation_in_intermediary_hooks_same_on_inductor_with_graph_break (__main__.HooksTests)": ["", [""]],
31921+ "test_functools_arg_vary (__main__.HooksTests)": ["", [""]],
31922+ "test_input_hooks_same (__main__.HooksTests)": ["", [""]],
31923+ "test_intermediary_hooks_same_on_aot_eager (__main__.HooksTests)": ["", [""]],
31924+ "test_intermediary_hooks_same_on_inductor (__main__.HooksTests)": ["", [""]],
31925+ "test_no_recompile_on_hook_identity_change (__main__.HooksTests)": ["", [""]],
31926+ "test_post_acc_grad_hook (__main__.HooksTests)": ["", [""]],
31927+ "test_no_recompile_on_nn_guarded_modules (__main__.OptimizedModuleTest)": ["", [""]],
31928+ "test_aot_ts (__main__.TestOptimizations)": ["", [""]],
31929+ "test_torchscript (__main__.TestOptimizations)": ["", [""]],
31930+ "test_adafactor (__main__.OptimizerTests)": ["", [""]],
31931+ "test_nadam (__main__.OptimizerTests)": ["", [""]],
31932+ "test_sgd (__main__.OptimizerTests)": ["", [""]],
31933+ "test_stride_in_bwd (__main__.AutogradFunctionTests)": ["", [""]],
31934+ "test_tensor_subclass_intermediary_input (__main__.AutogradFunctionTests)": ["", [""]],
31935+ "test_cache_size_limit (__main__.ConfigTests)": ["", [""]],
31936+ "test_no_saved_config (__main__.ConfigTests)": ["", [""]],
31937+ "test_no_saved_config_nested (__main__.ConfigTests)": ["", [""]],
31938+ "test_guards_correctly_property_assigned_on_tensor_type_change_inductor (__main__.TestInputAttrTracking)": ["", [""]],
31939+ "test_matcher_with_name_node_map_module (__main__.TestMatcher)": ["", [""]],
31940+ "test_zero_model_parallel_parameters_as_bucket_view_False (__main__.TestZeroRedundancyOptimizerDistributed)": ["", [""]],
31941+ "test_zero_model_parallel_parameters_as_bucket_view_True (__main__.TestZeroRedundancyOptimizerDistributed)": ["", [""]],
31942+ "test_tracing_all_reduce_nd (__main__.TraceDeviceMesh2DTest)": ["", [""]],
31943+ "test_tracing_all_reduce_nd (__main__.TraceDeviceMesh3DTest)": ["", [""]],
31944+ "test_fully_shard_backward_prefetch (__main__.TestFullyShardPrefetch)": ["", [""]],
31945+ "test_dtensor_dropout_backward (__main__.TestDTensorCustomOps)": ["", [""]],
31946+ "test_dtensor_npu_bmmV2 (__main__.TestDTensorCustomOps)": ["", [""]],
31947+ "test_2d_fsdp_tp_compile (__main__.TestDTensorCompileE2E)": ["", [""]],
31948+ "test_tp_compile_fullgraph_is_seq_parallel_False (__main__.TestDTensorCompileE2E)": ["", [""]],
31949+ "test_zeros_submesh (__main__.DTensorConstructorTest)": ["", [""]],
31950+ "test_sharded_embedding_colwise_errors (__main__.TestEmbeddingOp)": ["", [""]],
31951+ "test_sharded_embedding_rowwise (__main__.TestEmbeddingOp)": ["", [""]],
31952+ "test_forward_only_ScheduleClass0 (__main__.ScheduleTest)": ["", [""]],
31953+ "test_grad_with_manual_ScheduleClass0_shape_inference_False (__main__.ScheduleTest)": ["", [""]],
31954+ "test_grad_with_manual_ScheduleClass0_shape_inference_True (__main__.ScheduleTest)": ["", [""]],
31955+ "test_grad_with_manual_ScheduleClass1_shape_inference_False (__main__.ScheduleTest)": ["", [""]],
31956+ "test_grad_with_manual_ScheduleClass1_shape_inference_True (__main__.ScheduleTest)": ["", [""]],
31957+ "test_grad_with_tracer_ScheduleClass0_ModelClass0 (__main__.ScheduleTest)": ["", [""]],
31958+ "test_grad_with_tracer_ScheduleClass1_ModelClass0 (__main__.ScheduleTest)": ["", [""]],
31959+ "test_kwargs_with_tracer_ScheduleClass0 (__main__.ScheduleTest)": ["", [""]],
31960+ "test_kwargs_with_tracer_ScheduleClass1 (__main__.ScheduleTest)": ["", [""]],
31961+ "test_multi_iter_ScheduleClass0 (__main__.ScheduleTest)": ["", [""]],
31962+ "test_multi_iter_ScheduleClass1 (__main__.ScheduleTest)": ["", [""]],
31963+ "test_non_symmetric_stage_ids_schedule_class0_use_new_runtime_False (__main__.ScheduleTest)": ["", [""]],
31964+ "test_non_symmetric_stage_ids_schedule_class0_use_new_runtime_True (__main__.ScheduleTest)": ["", [""]],
31965+ "test_non_symmetric_stage_ids_schedule_class1_use_new_runtime_False (__main__.ScheduleTest)": ["", [""]],
31966+ "test_non_symmetric_stage_ids_schedule_class1_use_new_runtime_True (__main__.ScheduleTest)": ["", [""]],
31967+ "test_non_symmetric_stage_ids_schedule_class2_use_new_runtime_False (__main__.ScheduleTest)": ["", [""]],
31968+ "test_non_symmetric_stage_ids_schedule_class2_use_new_runtime_True (__main__.ScheduleTest)": ["", [""]],
31969+ "test_pipeline_schedule_runtime_custom_sched_ScheduleClass0 (__main__.ScheduleTest)": ["", [""]],
31970+ "test_schedule_with_native_zero_bubble_ScheduleClass0 (__main__.ScheduleTest)": ["", [""]],
31971+ "test_schedule_with_native_zero_bubble_ScheduleClass1 (__main__.ScheduleTest)": ["", [""]],
31972+ "test_custom_dw_with_fb_schedule (__main__.StageTest)": ["", [""]],
31973+ "test_manual (__main__.StageTest)": ["", [""]],
31974+ "test_tracer_ModelClass0 (__main__.StageTest)": ["", [""]],
31975+ "test_tracer_ModelClass1 (__main__.StageTest)": ["", [""]],
31976+ "test_tracer_kwargs_ModelClass0 (__main__.StageTest)": ["", [""]],
31977+ "test_dim_default_keepdim_argmax_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31978+ "test_dim_default_keepdim_argmin_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31979+ "test_dim_none_keepdim_argmax_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31980+ "test_dim_none_keepdim_argmin_npu (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31981+ "test_ref_small_input_masked_prod_npu_float16 (__main__.TestReductionsPRIVATEUSE1)": ["", [""]],
31982+ "test_distinfo_license (__main__.TestLicense)": ["", [""]],
31983+ "test_expect_true_with_s0 (__main__.TestPySymInt)": ["", [""]],
31984+ "test_sym_ceil (__main__.TestPySymInt)": ["", [""]],
31985+ "test_sym_floor (__main__.TestPySymInt)": ["", [""]],
31986+ "test_sym_int (__main__.TestPySymInt)": ["", [""]],
31987+ "test_sym_sqrt (__main__.TestPySymInt)": ["", [""]],
31988+ "test_unbacked_substitution (__main__.TestPySymInt)": ["", [""]],
31989+ "test_symnode_hashing (__main__.TestSymNumberMagicMethods)": ["", [""]],
31990+ "test_hook_registration (__main__.TestFlopCounter)": ["", [""]],
31991+ "test_batch_norm (__main__.TestCrossRefFunctionalization)": ["", [""]],
31992+ "test_cat (__main__.TestCrossRefFunctionalization)": ["", [""]],
31993+ "test_index_mutation_on_non_input (__main__.TestCrossRefFunctionalization)": ["", [""]],
31994+ "test_inplace_on_non_view (__main__.TestCrossRefFunctionalization)": ["", [""]],
31995+ "test_instance_norm (__main__.TestCrossRefFunctionalization)": ["", [""]],
31996+ "test_multi_out (__main__.TestCrossRefFunctionalization)": ["", [""]],
31997+ "test_mutable_op_not_inplace_or_other (__main__.TestCrossRefFunctionalization)": ["", [""]],
31998+ "test_optional_tensor_list (__main__.TestCrossRefFunctionalization)": ["", [""]],
31999+ "test_python_functionalization (__main__.TestCrossRefFunctionalization)": ["", [""]],
32000+ "test_reapply_views_simple (__main__.TestCrossRefFunctionalization)": ["", [""]],
32001+ "test_resize_larger_valid (__main__.TestCrossRefFunctionalization)": ["", [""]],
32002+ "test_resize_smaller (__main__.TestCrossRefFunctionalization)": ["", [""]],
32003+ "test_scalars (__main__.TestCrossRefFunctionalization)": ["", [""]],
32004+ "test_simple (__main__.TestCrossRefFunctionalization)": ["", [""]],
32005+ "test_simple_out (__main__.TestCrossRefFunctionalization)": ["", [""]],
32006+ "test_tensor_ctr (__main__.TestCrossRefFunctionalization)": ["", [""]],
32007+ "test_multidevice_serialization_npu (__main__.TestDevicePrecisionPRIVATEUSE1)": ["", [""]],
32008+ "test_cdist_grad_p_lt_1_no_nan_npu (__main__.TestTorchDeviceTypePRIVATEUSE1)": ["", [""]],
32009+ "test_serialization_npu (__main__.TestTorchDeviceTypePRIVATEUSE1)": ["", [""]],
32010+ "test_deprecation (__main__.TestNNInit)": ["", [""]],
32011+ "test_fake_tensor_mode_simple_invalid_input (__main__.TestFxToOnnx)": ["", [""]],
32012+ "test_missing_complex_onnx_variant_raises_errors_in_dispatcher (__main__.TestFxToOnnx)": ["", [""]],
32013+ "test_no_graph_break_on_item (__main__.SubGraphTests)": ["", [""]],
32014+ "test_compile_higher_order_with_functionalization (__main__.SubclassTests)": ["", [""]],
32015+ "test_compile_with_functionalization (__main__.SubclassTests)": ["", [""]],
32016+ "test_recompile_with_symbool_inputs (__main__.SubclassTests)": ["", [""]],
32017+ "test_user_overidden_attr_unsupported (__main__.SubclassTests)": ["", [""]],
32018+ "test_user_overidden_method_unsupported (__main__.SubclassTests)": ["", [""]],
32019+ "test_user_overidden_property_unsupported (__main__.SubclassTests)": ["", [""]],
32020+ "test_binary_does_not_recompile (__main__.TestNestedTensor)": ["", [""]],
32021+ "test_profiler_dynamo_compiled_region (__main__.DynamoProfilerTests)": ["", [""]],
32022+ "test_all (__main__.LoggingTests)": ["", [""]],
32023+ "test_cudagraph_static_inputs (__main__.LoggingTests)": ["", [""]],
32024+ "test_custom_format (__main__.LoggingTests)": ["", [""]],
32025+ "test_dynamo_error (__main__.LoggingTests)": ["", [""]],
32026+ "test_graph_sizes_dynamic (__main__.LoggingTests)": ["", [""]],
32027+ "test_guards_recompiles (__main__.LoggingTests)": ["", [""]],
32028+ "test_multiline_format (__main__.LoggingTests)": ["", [""]],
32029+ "test_pre_grad_graphs (__main__.LoggingTests)": ["", [""]],
32030+ "test_trace_bytecode (__main__.LoggingTests)": ["", [""]],
32031+ "test_trace_shape_events (__main__.LoggingTests)": ["", [""]],
32032+ "test_trace_source_if_stmt (__main__.LoggingTests)": ["", [""]],
32033+ "test_module_attribute_mutation_violation_positive_2 (__main__.MutationExportTests)": ["", [""]],
32034+ "test_module_attribute_mutation_violation_positive_4 (__main__.MutationExportTests)": ["", [""]],
32035+ "test_tensor_register_hook_in_graph_break_handle_lambda (__main__.HooksTests)": ["", [""]],
32036+ "test_nn_moduledict_contains (__main__.NNModuleTests)": ["", [""]],
32037+ "test_self_mutating1 (__main__.NNModuleTests)": ["", [""]],
32038+ "test_cache_size_limit_on_guarded_nn_modules (__main__.OptimizedModuleTest)": ["", [""]],
32039+ "test_hooks_inner (__main__.OptimizedModuleTest)": ["", [""]],
32040+ "test_hooks_skip_guards (__main__.OptimizedModuleTest)": ["", [""]],
32041+ "test_no_guard_on_torch_nn_modules (__main__.OptimizedModuleTest)": ["", [""]],
32042+ "test_no_op_assignment (__main__.OptimizedModuleTest)": ["", [""]],
32043+ "test_if_graph_minified (__main__.MinifierTests)": ["", [""]],
32044+ "test_get_local (__main__.ComptimeTests)": ["", [""]],
32045+ "test_print_graph (__main__.ComptimeTests)": ["", [""]],
32046+ "test_print_guards (__main__.ComptimeTests)": ["", [""]],
32047+ "test_print_locals (__main__.ComptimeTests)": ["", [""]],
32048+ "test_print_value_stack (__main__.ComptimeTests)": ["", [""]],
32049+ "test_print_in_bwd (__main__.AutogradFunctionTests)": ["", [""]],
32050+ "test_smoke_from_test_autograd (__main__.AutogradFunctionTests)": ["", [""]],
32051+ "test_mark_static_address_unguarded (__main__.DecoratorTests)": ["", [""]],
32052+ "test_config_changed_from_guarded_config_1 (__main__.ConfigTests)": ["", [""]],
32053+ "test_config_compile_ignored (__main__.ConfigTests)": ["", [""]],
32054+ "test_multiple_compile_recompiles (__main__.ConfigTests)": ["", [""]],
32055+ "test_disable_saved_tensors_hooks (__main__.CtxManagerTests)": ["", [""]],
32056+ "test_disable_saved_tensors_hooks_graph_break (__main__.CtxManagerTests)": ["", [""]],
32057+ "test_disable_saved_tensors_hooks_prev_disabled (__main__.CtxManagerTests)": ["", [""]],
32058+ "test_disable_saved_tensors_hooks_prev_disabled_nested (__main__.CtxManagerTests)": ["", [""]],
32059+ "test_set_data_on_input_tensor (__main__.TestInputAttrTracking)": ["", [""]],
32060+ "test_schedule_with_weight_update_mlp_e2e_ScheduleClass0 (__main__.ScheduleTest)": ["", [""]],
32061+ "test_constant_pad_nd_with_negative (__main__.TestConstantPadNd)": ["", [""]],
32062+ "test_floordiv_assumptions (__main__.TestFloorDiv)": ["", [""]],
32063+ "test_floordiv_bool (__main__.TestFloorDiv)": ["", [""]],
32064+ "test_floordiv_complex (__main__.TestFloorDiv)": ["", [""]],
32065+ "test_floordiv_div_by_one (__main__.TestFloorDiv)": ["", [""]],
32066+ "test_floordiv_float_int (__main__.TestFloorDiv)": ["", [""]],
32067+ "test_floordiv_simplify (__main__.TestFloorDiv)": ["", [""]],
32068+ "test_floordiv_simplify_rational (__main__.TestFloorDiv)": ["", [""]],
32069+ "test_floordiv_zero_base (__main__.TestFloorDiv)": ["", [""]],
32070+ "test_rmsprop (optim.test_optim.TestDifferentiableOptimizer)": ["", [""]],
32071+ "test_lr_scheduler_verbose_deprecation_warning_LRClass0 (optim.test_lrscheduler.TestLRScheduler)": ["", [""]],
32072+ "test_lr_scheduler_verbose_deprecation_warning_LRClass1 (optim.test_lrscheduler.TestLRScheduler)": ["", [""]],
32073+ "test_lr_scheduler_verbose_deprecation_warning_LRClass10 (optim.test_lrscheduler.TestLRScheduler)": ["", [""]],
32074+ "test_lr_scheduler_verbose_deprecation_warning_LRClass11 (optim.test_lrscheduler.TestLRScheduler)": ["", [""]],
32075+ "test_lr_scheduler_verbose_deprecation_warning_LRClass12 (optim.test_lrscheduler.TestLRScheduler)": ["", [""]],
32076+ "test_lr_scheduler_verbose_deprecation_warning_LRClass13 (optim.test_lrscheduler.TestLRScheduler)": ["", [""]],
32077+ "test_lr_scheduler_verbose_deprecation_warning_LRClass2 (optim.test_lrscheduler.TestLRScheduler)": ["", [""]],
32078+ "test_lr_scheduler_verbose_deprecation_warning_LRClass3 (optim.test_lrscheduler.TestLRScheduler)": ["", [""]],
32079+ "test_lr_scheduler_verbose_deprecation_warning_LRClass4 (optim.test_lrscheduler.TestLRScheduler)": ["", [""]],
32080+ "test_lr_scheduler_verbose_deprecation_warning_LRClass5 (optim.test_lrscheduler.TestLRScheduler)": ["", [""]],
32081+ "test_lr_scheduler_verbose_deprecation_warning_LRClass6 (optim.test_lrscheduler.TestLRScheduler)": ["", [""]],
32082+ "test_lr_scheduler_verbose_deprecation_warning_LRClass7 (optim.test_lrscheduler.TestLRScheduler)": ["", [""]],
32083+ "test_lr_scheduler_verbose_deprecation_warning_LRClass8 (optim.test_lrscheduler.TestLRScheduler)": ["", [""]],
32084+ "test_lr_scheduler_verbose_deprecation_warning_LRClass9 (optim.test_lrscheduler.TestLRScheduler)": ["", [""]],
32085+ "test_sparse_adam (optim.test_optim.TestOptim)": ["", [""]],
32086+ "test_matmul_backward (__main__.TestRegisterSharding)": ["", [""]],
32087+ "test_matmul_shardable (__main__.TestRegisterSharding)": ["", [""]],
32088+ "test_matmul_unshardable (__main__.TestRegisterSharding)": ["", [""]],
32089+ "test_npu_fussion_attention_forward (__main__.TestRegisterSharding)": ["", [""]],
32090+ "test_npu_fussion_attention_grad (__main__.TestRegisterSharding)": ["", [""]],
32091+ "test_cycle_lr_cycle_momentum_fail_with_momentumless_optimizer (optim.test_lrscheduler.TestLRScheduler)": ["", [""]],
32092+ "test_as_strided (__main__.TestFunctionalization)": ["", [""]],
32093+ "test_copy_ (__main__.TestFunctionalization)": ["", [""]],
32094+ "test_diagonal (__main__.TestFunctionalization)": ["", [""]],
32095+ "test_diagonal_mutated_input ( main .TestFunctionalization)": ["", [""]],
32096+ "test_everything (__main__.TestFunctionalization)": ["", [""]],
32097+ "test_fill_ (__main__.TestFunctionalization)": ["", [""]],
32098+ "test_split (__main__.TestFunctionalization)": ["", [""]],
32099+ "test_split_with_sizes (__main__.TestFunctionalization)": ["", [""]],
32100+ "test_view_clone_view_inplace (__main__.TestFunctionalization)": ["", [""]],
32101+ "test_view_inplace (__main__.TestFunctionalization)": ["", [""]],
31619 "test_module_backcompat (__main__.TestNN)": ["", [""]],32102 "test_module_backcompat (__main__.TestNN)": ["", [""]],
31620 "test_register_state_dict_pre_hook_backward_compat (__main__.TestNN)": ["", [""]],32103 "test_register_state_dict_pre_hook_backward_compat (__main__.TestNN)": ["", [""]],
31621 "test_softmax_forward_64bit_indexing_npu (__main__.TestNNDeviceTypePRIVATEUSE1)": ["", [""]],32104 "test_softmax_forward_64bit_indexing_npu (__main__.TestNNDeviceTypePRIVATEUSE1)": ["", [""]],
Mtest/utils/test_torch_npu_logs_filter.py+3-0
@@ -52,18 +52,21 @@ result_matmul = torch.matmul(matrix1, matrix2)
52result_add = torch.add(matrix1, matrix2)52result_add = torch.add(matrix1, matrix2)
53"""53"""
54 54 
55+ @unittest.skip("Temporarily skipping")
55 def test_no_filter(self):56 def test_no_filter(self):
56 # 不设置过滤器，所有日志输出57 # 不设置过滤器，所有日志输出
57 output = self._run_test('op_plugin', '')58 output = self._run_test('op_plugin', '')
58 self.assertIn('mm', output)59 self.assertIn('mm', output)
59 self.assertIn('add', output)60 self.assertIn('add', output)
60 61 
62+ @unittest.skip("Temporarily skipping")
61 def test_white_list(self):63 def test_white_list(self):
62 # 白名单过滤，只输出 matmul64 # 白名单过滤，只输出 matmul
63 output = self._run_test('op_plugin', '+mm')65 output = self._run_test('op_plugin', '+mm')
64 self.assertIn('mm', output)66 self.assertIn('mm', output)
65 self.assertNotIn('add', output)67 self.assertNotIn('add', output)
66 68 
69+ @unittest.skip("Temporarily skipping")
67 def test_black_list(self):70 def test_black_list(self):
68 # 黑名单过滤，排除 add71 # 黑名单过滤，排除 add
69 output = self._run_test('op_plugin', '-add')72 output = self._run_test('op_plugin', '-add')