移动 AI 运行时裁剪后如何验收算子兼容性

from pathlib import Path
import hashlib
import json
import sys

if len(sys.argv) != 3:
    raise SystemExit(2)
requirements_path = Path(sys.argv[1])
capabilities_path = Path(sys.argv[2])
if not requirements_path.is_file() or not capabilities_path.is_file():
    raise SystemExit(2)
requirements_raw = requirements_path.read_bytes()
capabilities_raw = capabilities_path.read_bytes()
requirements = json.loads(requirements_raw.decode("utf-8"))
capabilities = json.loads(capabilities_raw.decode("utf-8"))
if not isinstance(requirements.get("operators"), list):
    raise SystemExit(2)
if not isinstance(capabilities.get("builtin"), list):
    raise SystemExit(2)
custom = set(capabilities.get("custom", []))
fallback = set(capabilities.get("fallback", []))
builtin = set()
for item in capabilities["builtin"]:
    fields = (item.get("domain"), item.get("operator"), item.get("opset"), tuple(item.get("types", [])))
    if not all(fields[:3]) or not fields[3]:
        raise SystemExit(2)
    builtin.add(fields)
results = []
missing = []
for node in requirements["operators"]:
    key = (node.get("domain"), node.get("operator"), node.get("opset"), tuple(node.get("types", [])))
    node_id = str(node.get("nodeId", ""))
    if not node_id or not all(key[:3]) or not key[3]:
        raise SystemExit(2)
    if node_id in custom:
        route = "custom"
    elif node_id in fallback and key in builtin:
        route = "fallback"
    elif key in builtin:
        route = "builtin"
    else:
        route = "missing"
        missing.append(node_id)
    results.append({"nodeId": node_id, "route": route})
report = {"requirementsSha256": hashlib.sha256(requirements_raw).hexdigest(), "capabilitiesSha256": hashlib.sha256(capabilities_raw).hexdigest(), "results": results, "missing": sorted(missing)}
print(json.dumps(report, ensure_ascii=False, indent=2))
if missing:
    raise SystemExit(3)