test update ut-pytorch-AtomGit

已合并

test update ut #32257

huangyunlong创建于 3月24日从huangyunlong2022/pytorch： 2.10ut 合入到 Ascend/pytorch： v2.10.0

test update ut #32257

已合并

huangyunlong创建于 3月24日

从huangyunlong2022/pytorch 2.10ut合入到Ascend/pytorch v2.10.0

共 5 个文件变更+4-45

Mtest/onnx/test_pytorch_onnx_onnxruntime_npu.py+1-0

	@@ -118,6 +118,7 @@ class TestONNXRuntime_npu(onnx_test_common._TestONNXRuntime):
118		118
119	# ONNX supports bfloat16 for opsets >= 13	119	# ONNX supports bfloat16 for opsets >= 13
120	# Add, Sub and Mul ops don't support bfloat16 cpu in onnxruntime.	120	# Add, Sub and Mul ops don't support bfloat16 cpu in onnxruntime.
		121	+ @unittest.skip("Add, Sub and Mul ops don't support bfloat16 cpu in onnxruntime.")
121	@skipIfUnsupportedMinOpsetVersion(13)	122	@skipIfUnsupportedMinOpsetVersion(13)
122	@skipIfNoBFloat16NPU	123	@skipIfNoBFloat16NPU
123	def test_arithmetic_bfp16(self):	124	def test_arithmetic_bfp16(self):

Mtest/onnx/test_utility_funs.py+1-0

	@@ -642,6 +642,7 @@ class TestUtilityFuns(_BaseTestCase):
642	onnx_model = onnx.load(io.BytesIO(f.getvalue()))	642	onnx_model = onnx.load(io.BytesIO(f.getvalue()))
643	self.assertEqual(len(onnx_model.graph.initializer), 0)	643	self.assertEqual(len(onnx_model.graph.initializer), 0)
644		644
		645	+ # onnx need protobuf
645	def test_verbose(self):	646	def test_verbose(self):
646	class MyModule(torch.nn.Module):	647	class MyModule(torch.nn.Module):
647	def forward(self, input_):	648	def forward(self, input_):

Mtest/profiler/test_memory_profiler.py+1-1

	@@ -1401,7 +1401,7 @@ class TestMemoryProfilerE2E(TestCase):
1401	for _, action, (key, version), size in prof._memory_profile().timeline	1401	for _, action, (key, version), size in prof._memory_profile().timeline
1402	# We generally don't care about tiny allocations during memory	1402	# We generally don't care about tiny allocations during memory
1403	# profiling and they add a lot of noise to the unit test.	1403	# profiling and they add a lot of noise to the unit test.
1404	- if size > 1024	1404	+ if size > 1024 and isinstance(key, _memory_profiler.TensorKey)
1405	]	1405	]
1406		1406
1407	self.assertExpectedInline(	1407	self.assertExpectedInline(

Mtest/requirements.txt+1-0

	@@ -19,3 +19,4 @@ parameterized==0.9.0
19	torch-scatter==2.1.2	19	torch-scatter==2.1.2
20	torchvision==0.25.0	20	torchvision==0.25.0
21	ml-dtypes==0.2.0	21	ml-dtypes==0.2.0
		22	+protobuf==5.29.5

Mtest/unsupported_test_cases/.pytorch-disabled-tests.json+0-44

	@@ -27677,52 +27677,8 @@
27677	"test_resnet18 (__main__.TestQuantizedModelsONNXRuntime_is_script_True)": ["", [""]],	27677	"test_resnet18 (__main__.TestQuantizedModelsONNXRuntime_is_script_True)": ["", [""]],
27678	"test_resnet50 (__main__.TestQuantizedModelsONNXRuntime_is_script_True)": ["", [""]],	27678	"test_resnet50 (__main__.TestQuantizedModelsONNXRuntime_is_script_True)": ["", [""]],
27679	"test_resnext101_32x8d (__main__.TestQuantizedModelsONNXRuntime_is_script_True)": ["", [""]],	27679	"test_resnext101_32x8d (__main__.TestQuantizedModelsONNXRuntime_is_script_True)": ["", [""]],
27680	- "test_verbose (__main__.TestUtilityFuns_opset_9)": ["", [""]],
27681	- "test_verbose (__main__.TestUtilityFuns_opset_10)": ["", [""]],
27682	- "test_verbose (__main__.TestUtilityFuns_opset_11)": ["", [""]],
27683	- "test_verbose (__main__.TestUtilityFuns_opset_12)": ["", [""]],
27684	- "test_verbose (__main__.TestUtilityFuns_opset_13)": ["", [""]],
27685	- "test_verbose (__main__.TestUtilityFuns_opset_14)": ["", [""]],
27686	- "test_verbose (__main__.TestUtilityFuns_opset_15)": ["", [""]],
27687	- "test_verbose (__main__.TestUtilityFuns_opset_16)": ["", [""]],
27688	- "test_verbose (__main__.TestUtilityFuns_opset_17)": ["", [""]],
27689	- "test_verbose (__main__.TestUtilityFuns_opset_18)": ["", [""]],
27690	- "test_verbose (__main__.TestUtilityFuns_opset_19)": ["", [""]],
27691	- "test_verbose (__main__.TestUtilityFuns_opset_20)": ["", [""]],
27692	"test_api_compatibility (__main__.TestPublicApiCompatibility)": ["", [""]],	27680	"test_api_compatibility (__main__.TestPublicApiCompatibility)": ["", [""]],
27693	"test_wrapper_npu_fused_attention_layernorm_qkv_fwd (__main__.TestOnnxOps)": ["", [""]],	27681	"test_wrapper_npu_fused_attention_layernorm_qkv_fwd (__main__.TestOnnxOps)": ["", [""]],
27694	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_13_is_script_False_keep_initializers_as_inputs_False)": ["", [""]],
27695	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_13_is_script_False_keep_initializers_as_inputs_True)": ["", [""]],
27696	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_13_is_script_True_keep_initializers_as_inputs_False)": ["", [""]],
27697	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_13_is_script_True_keep_initializers_as_inputs_True)": ["", [""]],
27698	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_14_is_script_False_keep_initializers_as_inputs_False)": ["", [""]],
27699	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_14_is_script_False_keep_initializers_as_inputs_True)": ["", [""]],
27700	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_14_is_script_True_keep_initializers_as_inputs_False)": ["", [""]],
27701	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_14_is_script_True_keep_initializers_as_inputs_True)": ["", [""]],
27702	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_15_is_script_False_keep_initializers_as_inputs_False)": ["", [""]],
27703	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_15_is_script_False_keep_initializers_as_inputs_True)": ["", [""]],
27704	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_15_is_script_True_keep_initializers_as_inputs_False)": ["", [""]],
27705	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_15_is_script_True_keep_initializers_as_inputs_True)": ["", [""]],
27706	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_16_is_script_False_keep_initializers_as_inputs_False)": ["", [""]],
27707	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_16_is_script_False_keep_initializers_as_inputs_True)": ["", [""]],
27708	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_16_is_script_True_keep_initializers_as_inputs_False)": ["", [""]],
27709	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_16_is_script_True_keep_initializers_as_inputs_True)": ["", [""]],
27710	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_17_is_script_False_keep_initializers_as_inputs_False)": ["", [""]],
27711	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_17_is_script_False_keep_initializers_as_inputs_True)": ["", [""]],
27712	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_17_is_script_True_keep_initializers_as_inputs_False)": ["", [""]],
27713	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_17_is_script_True_keep_initializers_as_inputs_True)": ["", [""]],
27714	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_18_is_script_False_keep_initializers_as_inputs_False)": ["", [""]],
27715	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_18_is_script_False_keep_initializers_as_inputs_True)": ["", [""]],
27716	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_18_is_script_True_keep_initializers_as_inputs_False)": ["", [""]],
27717	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_18_is_script_True_keep_initializers_as_inputs_True)": ["", [""]],
27718	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_19_is_script_False_keep_initializers_as_inputs_False)": ["", [""]],
27719	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_19_is_script_False_keep_initializers_as_inputs_True)": ["", [""]],
27720	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_19_is_script_True_keep_initializers_as_inputs_False)": ["", [""]],
27721	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_19_is_script_True_keep_initializers_as_inputs_True)": ["", [""]],
27722	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_20_is_script_False_keep_initializers_as_inputs_False)": ["", [""]],
27723	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_20_is_script_False_keep_initializers_as_inputs_True)": ["", [""]],
27724	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_20_is_script_True_keep_initializers_as_inputs_False)": ["", [""]],
27725	- "test_arithmetic_bfp16 (__main__.TestONNXRuntime_npu_opset_version_20_is_script_True_keep_initializers_as_inputs_True)": ["", [""]],
27726	"test_faster_rcnn (__main__.TestModelsONNXRuntime_is_script_False)": ["", [""]],	27682	"test_faster_rcnn (__main__.TestModelsONNXRuntime_is_script_False)": ["", [""]],
27727	"test_shufflenet (__main__.TestModels)": ["", [""]],	27683	"test_shufflenet (__main__.TestModels)": ["", [""]],
27728	"test_r2plus1d_18_video (__main__.TestModels)": ["", [""]],	27684	"test_r2plus1d_18_video (__main__.TestModels)": ["", [""]],