mirror of https://github.com/python/cpython
80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
"""Generate the list of uop IDs.
|
|
Reads the instruction definitions from bytecodes.c.
|
|
Writes the IDs to pycore_uop_ids.h by default.
|
|
"""
|
|
|
|
import argparse
|
|
|
|
from analyzer import (
|
|
Analysis,
|
|
analyze_files,
|
|
)
|
|
from generators_common import (
|
|
DEFAULT_INPUT,
|
|
ROOT,
|
|
write_header,
|
|
)
|
|
from cwriter import CWriter
|
|
from typing import TextIO
|
|
|
|
|
|
DEFAULT_OUTPUT = ROOT / "Include/internal/pycore_uop_ids.h"
|
|
|
|
|
|
def generate_uop_ids(
|
|
filenames: list[str], analysis: Analysis, outfile: TextIO, distinct_namespace: bool
|
|
) -> None:
|
|
write_header(__file__, filenames, outfile)
|
|
out = CWriter(outfile, 0, False)
|
|
with out.header_guard("Py_CORE_UOP_IDS_H"):
|
|
next_id = 1 if distinct_namespace else 300
|
|
# These two are first by convention
|
|
out.emit(f"#define _EXIT_TRACE {next_id}\n")
|
|
next_id += 1
|
|
out.emit(f"#define _SET_IP {next_id}\n")
|
|
next_id += 1
|
|
PRE_DEFINED = {"_EXIT_TRACE", "_SET_IP"}
|
|
|
|
uops = [(uop.name, uop) for uop in analysis.uops.values()]
|
|
# Sort so that _BASE comes immediately before _BASE_0, etc.
|
|
for name, uop in sorted(uops):
|
|
if name in PRE_DEFINED:
|
|
continue
|
|
if uop.properties.tier == 1:
|
|
continue
|
|
if uop.implicitly_created and not distinct_namespace and not uop.replicated:
|
|
out.emit(f"#define {name} {name[1:]}\n")
|
|
else:
|
|
out.emit(f"#define {name} {next_id}\n")
|
|
next_id += 1
|
|
|
|
out.emit(f"#define MAX_UOP_ID {next_id-1}\n")
|
|
|
|
|
|
arg_parser = argparse.ArgumentParser(
|
|
description="Generate the header file with all uop IDs.",
|
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
|
|
)
|
|
|
|
arg_parser.add_argument(
|
|
"-o", "--output", type=str, help="Generated code", default=DEFAULT_OUTPUT
|
|
)
|
|
arg_parser.add_argument(
|
|
"-n",
|
|
"--namespace",
|
|
help="Give uops a distinct namespace",
|
|
action="store_true",
|
|
)
|
|
|
|
arg_parser.add_argument(
|
|
"input", nargs=argparse.REMAINDER, help="Instruction definition file(s)"
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
args = arg_parser.parse_args()
|
|
if len(args.input) == 0:
|
|
args.input.append(DEFAULT_INPUT)
|
|
data = analyze_files(args.input)
|
|
with open(args.output, "w") as outfile:
|
|
generate_uop_ids(args.input, data, outfile, args.namespace)
|