[Apple Silicon] [CI] Move the MLX lane to the check-changes + pr-gate composite (#30121)

This commit is contained in:
Jae B.
2026-07-10 22:00:15 -07:00
committed by GitHub
parent e1d51be91f
commit a91c2e6596
15 changed files with 288 additions and 145 deletions
+16
View File
@@ -16,6 +16,7 @@ __all__ = [
"register_npu_ci",
"register_xpu_ci",
"register_musa_ci",
"register_mlx_ci",
"ut_parse_one_file",
]
@@ -35,6 +36,7 @@ class HWBackend(Enum):
NPU = auto()
XPU = auto()
MUSA = auto()
MLX = auto()
@dataclass
@@ -148,6 +150,19 @@ def register_musa_ci(
return None
def register_mlx_ci(
est_time: float,
suite: Optional[str] = None,
nightly: bool = False,
disabled: Optional[str] = None,
*,
stage: Optional[str] = None,
runner_config: Optional[str] = None,
):
"""Marker for MLX CI registration (parsed via AST; runtime no-op)."""
return None
REGISTER_MAPPING = {
"register_cpu_ci": HWBackend.CPU,
"register_cuda_ci": HWBackend.CUDA,
@@ -156,6 +171,7 @@ REGISTER_MAPPING = {
"register_npu_ci": HWBackend.NPU,
"register_xpu_ci": HWBackend.XPU,
"register_musa_ci": HWBackend.MUSA,
"register_mlx_ci": HWBackend.MLX,
}