Function main

install_executorch.py:209–264

(args)

def main(args):
    if not python_is_compatible():
        sys.exit(1)

    args = _parse_args()

    if args.clean:
        clean()
        return

    check_and_update_submodules()
    # This option is used in CI to make sure that PyTorch build from the pinned commit
    # is used instead of nightly. CI jobs wouldn't be able to catch regression from the
    # latest PT commit otherwise
    use_pytorch_nightly = not args.use_pt_pinned_commit

    # Step 1: Install core dependencies first
    install_requirements(use_pytorch_nightly)

    # Step 2: Install build dependencies for optional dependencies
    # They need to be installed before optional dependencies due to --no-build-isolation
    optional_build_dependencies: list[str] = []
    for optional_dep in args.optional_dependency:
        match optional_dep:
            case "cortex_m":
                optional_build_dependencies.extend(
                    ["pybind11>=2.10", "scikit-build-core>=0.7"]
                )
    if len(optional_build_dependencies) > 0:
        cmd = [sys.executable, "-m", "pip", "install", *optional_build_dependencies]
        subprocess.run(cmd, check=True)

    # Step 3: Install core package
    package_spec = "."
    if args.optional_dependency:
        extras = ",".join(dict.fromkeys(args.optional_dependency))
        package_spec = f".[{extras}]"
    cmd = (
        [
            sys.executable,
            "-m",
            "pip",
            "install",
        ]
        + (["--editable"] if args.editable else [])
        + [
            package_spec,
            "--no-build-isolation",
            "-v",
        ]
    )
    subprocess.run(cmd, check=True)

    # Step 4: Extra (optional) packages that is only useful for running examples.
    if not args.minimal:
        install_optional_example_requirements(use_pytorch_nightly)