[NPU]add nightly-test-npu (#14143)
This commit is contained in:
@@ -11,6 +11,7 @@ __all__ = [
|
||||
"register_cpu_ci",
|
||||
"register_cuda_ci",
|
||||
"register_amd_ci",
|
||||
"register_npu_ci",
|
||||
"ut_parse_one_file",
|
||||
]
|
||||
|
||||
@@ -22,6 +23,7 @@ class HWBackend(Enum):
|
||||
CPU = auto()
|
||||
CUDA = auto()
|
||||
AMD = auto()
|
||||
NPU = auto()
|
||||
|
||||
|
||||
@dataclass
|
||||
@@ -58,10 +60,21 @@ def register_amd_ci(
|
||||
return None
|
||||
|
||||
|
||||
def register_npu_ci(
|
||||
est_time: float,
|
||||
suite: str,
|
||||
nightly: bool = False,
|
||||
disabled: Optional[str] = None,
|
||||
):
|
||||
"""Marker for NPU CI registration (parsed via AST; runtime no-op)."""
|
||||
return None
|
||||
|
||||
|
||||
REGISTER_MAPPING = {
|
||||
"register_cpu_ci": HWBackend.CPU,
|
||||
"register_cuda_ci": HWBackend.CUDA,
|
||||
"register_amd_ci": HWBackend.AMD,
|
||||
"register_npu_ci": HWBackend.NPU,
|
||||
}
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user