greg/cpython
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

gh-134584: Specialize POP_TOP by reference and type in JIT (GH-135761)

Browse Source
This commit is contained in:
Ken Jin
2025-06-24 00:57:14 +08:00
committed by GitHub
parent 99712c45cc
commit 569fc6870f
12 changed files with 239 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
Tools/cases_generator/analyzer.py
View File

@@ -596,6 +596,7 @@ NON_ESCAPING_FUNCTIONS = (
"PyStackRef_IsNull",
"PyStackRef_MakeHeapSafe",
"PyStackRef_None",
"PyStackRef_RefcountOnObject",
"PyStackRef_TYPE",
"PyStackRef_True",
"PyTuple_GET_ITEM",

12
Tools/cases_generator/opcode_metadata_generator.py
View File

@@ -242,14 +242,10 @@ def generate_expansion_table(analysis: Analysis, out: CWriter) -> None:
assert name2 in analysis.instructions, f"{name2} doesn't match any instr"
instr1 = analysis.instructions[name1]
instr2 = analysis.instructions[name2]
assert (
len(instr1.parts) == 1
), f"{name1} is not a good superinstruction part"
assert (
len(instr2.parts) == 1
), f"{name2} is not a good superinstruction part"
expansions.append((instr1.parts[0].name, "OPARG_TOP", 0))
expansions.append((instr2.parts[0].name, "OPARG_BOTTOM", 0))
for part in instr1.parts:
expansions.append((part.name, "OPARG_TOP", 0))
for part in instr2.parts:
expansions.append((part.name, "OPARG_BOTTOM", 0))
elif not is_viable_expansion(inst):
continue
else: