skip ci failed testcase-pytorch-AtomGit

已合并

skip ci failed testcase #36499

skip ci failed testcase #36499

已合并

XianglongZeng创建于 5月22日

共 4 个文件变更+23-0

	@@ -3,6 +3,7 @@ import sys
3	import shutil	3	import shutil
4	import subprocess	4	import subprocess
5	import ctypes	5	import ctypes
		6	+import unittest
6	import torch	7	import torch
7	import torch.utils.cpp_extension	8	import torch.utils.cpp_extension
8		9
	@@ -26,6 +27,10 @@ def build_stub(base_dir):
26	raise RuntimeError('Failed to build stub: {}'.format(build_stub_cmd))	27	raise RuntimeError('Failed to build stub: {}'.format(build_stub_cmd))
27		28
28		29
		30	+@unittest.skip(
		31	+ "Skip: pre-existing CI environment issue, "
		32	+ "acl/acl_base_rt.h header missing on ARM CI"
		33	+)
29	class TestPluggableAllocator(TestCase):	34	class TestPluggableAllocator(TestCase):
30	module = None	35	module = None
31	new_alloc = None	36	new_alloc = None

	@@ -15,6 +15,7 @@ import copy
15	import logging	15	import logging
16	import tempfile	16	import tempfile
17	from dataclasses import dataclass	17	from dataclasses import dataclass
		18	+import unittest
18		19
19	from model_registry import ModelWithKwargs, MultiMLP, MultiMLPKwargs, MultiMLPWithDw	20	from model_registry import ModelWithKwargs, MultiMLP, MultiMLPKwargs, MultiMLPWithDw
20	from schedule_registry import (	21	from schedule_registry import (
	@@ -285,6 +286,10 @@ def patch_stage_init_method(stages):
285	stage._get_init_p2p_neighbors_ops = original	286	stage._get_init_p2p_neighbors_ops = original
286		287
287		288
		289	+@unittest.skip(
		290	+ "Skip: pre-existing CI environment issue, "
		291	+ "HCCL communication error (hcclCommInitRootInfoConfig failed) on ARM CI"
		292	+)
288	class ScheduleTest(MultiProcContinuousTest):	293	class ScheduleTest(MultiProcContinuousTest):
289	world_size = 4	294	world_size = 4
290		295
	@@ -870,6 +875,10 @@ class ScheduleTest(MultiProcContinuousTest):
870	instantiate_parametrized_tests(ScheduleTest)	875	instantiate_parametrized_tests(ScheduleTest)
871		876
872		877
		878	+@unittest.skip(
		879	+ "Skip: pre-existing CI environment issue, "
		880	+ "HCCL communication error (hcclCommInitRootInfoConfig failed) on ARM CI"
		881	+)
873	class CustomSchedulesTest(MultiProcContinuousTest):	882	class CustomSchedulesTest(MultiProcContinuousTest):
874	"""	883	"""
875	These schedules are from the ScheduleRegistry and require world_size == 2	884	These schedules are from the ScheduleRegistry and require world_size == 2

	@@ -2,6 +2,7 @@ import os
2	import gc	2	import gc
3	import shutil	3	import shutil
4	import threading	4	import threading
		5	+import unittest
5	import subprocess	6	import subprocess
6		7
7	import torch	8	import torch
	@@ -28,6 +29,10 @@ def build_stub(base_dir):
28	raise RuntimeError('Failed to build stub: {}'.format(build_stub_cmd))	29	raise RuntimeError('Failed to build stub: {}'.format(build_stub_cmd))
29		30
30		31
		32	+@unittest.skip(
		33	+ "Skip: pre-existing CI environment issue, "
		34	+ "acl/acl_base_rt.h header missing on ARM CI"
		35	+)
31	class TestPluggableAllocator(TestCase):	36	class TestPluggableAllocator(TestCase):
32	torch.npu.memory._set_allocator_settings("expandable_segments:True")	37	torch.npu.memory._set_allocator_settings("expandable_segments:True")
33	module = None	38	module = None

	@@ -60,6 +60,10 @@ def reset_sanitizer():
60	sanitizer.npu_sanitizer.enabled = False	60	sanitizer.npu_sanitizer.enabled = False
61		61
62		62
		63	+@unittest.skip(
		64	+ "Skip: pre-existing CI environment issue, "
		65	+ "acl/acl_base_rt.h header missing on ARM CI"
		66	+)
63	@unittest.skipUnless(IS_ARM64, "Only working on ARM")	67	@unittest.skipUnless(IS_ARM64, "Only working on ARM")
64	class TestSanitizerPluggableAllocator(TestCase):	68	class TestSanitizerPluggableAllocator(TestCase):
65	module = None	69	module = None