mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-09-26 18:29:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 10

gh-119689: generate stack effect metadata for pseudo instructions (#119691)

Browse source
This commit is contained in:
Irit Katriel 2024-05-29 10:47:56 +01:00 • committed by GitHub
parent 7ca74a760a
commit c1e9647107
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 112 additions and 73 deletions
Download patch file Download diff file Expand all files Collapse all files

22
Tools/cases_generator/stack.py
View file

@ -1,7 +1,8 @@
import re
from analyzer import StackItem, Instruction, Uop
from analyzer import StackItem, StackEffect, Instruction, Uop, PseudoInstruction
from dataclasses import dataclass
from cwriter import CWriter
from typing import Iterator
UNUSED = {"unused"}
@ -208,13 +209,20 @@ class Stack:
return f"/* Variables: {[v.name for v in self.variables]}. Base offset: {self.base_offset.to_c()}. Top offset: {self.top_offset.to_c()} */"
def get_stack_effect(inst: Instruction) -> Stack:
def get_stack_effect(inst: Instruction | PseudoInstruction) -> Stack:
stack = Stack()
for uop in inst.parts:
if not isinstance(uop, Uop):
continue
for var in reversed(uop.stack.inputs):
def stacks(inst : Instruction | PseudoInstruction) -> Iterator[StackEffect]:
if isinstance(inst, Instruction):
for uop in inst.parts:
if isinstance(uop, Uop):
yield uop.stack
else:
assert isinstance(inst, PseudoInstruction)
yield inst.stack
for s in stacks(inst):
for var in reversed(s.inputs):
stack.pop(var)
for i, var in enumerate(uop.stack.outputs):
for var in s.outputs:
stack.push(var)
return stack