cpython/Tools/jit/build.py

"""Build an experimental just-in-time compiler for CPython."""

import argparse
import pathlib
import shlex
import sys

import _targets

if __name__ == "__main__":
    comment = f"$ {shlex.join([pathlib.Path(sys.executable).name] + sys.argv)}"
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument(
        "target",
        nargs="+",
        type=_targets.get_target,
        help="a PEP 11 target triple to compile for",
    )
    parser.add_argument(
        "-d", "--debug", action="store_true", help="compile for a debug build of Python"
    )
    parser.add_argument(
        "-f", "--force", action="store_true", help="force the entire JIT to be rebuilt"
    )
    parser.add_argument(
        "-v", "--verbose", action="store_true", help="echo commands as they are run"
    )
    args = parser.parse_args()
    for target in args.target:
        target.debug = args.debug
        target.force = args.force
        target.verbose = args.verbose
        target.build(
            pathlib.Path.cwd(),
            comment=comment,
            stencils_h=f"jit_stencils-{target.triple}.h",
            force=args.force,
        )

    with open("jit_stencils.h", "w") as fp:
        for idx, target in enumerate(args.target):
            fp.write(f"#{'if' if idx == 0 else 'elif'} {target.condition}\n")
            fp.write(f'#include "jit_stencils-{target.triple}.h"\n')

        fp.write("#else\n")
        fp.write('#error "unexpected target"\n')
        fp.write("#endif\n")