1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
from pypy.interpreter.astcompiler.assemble import Instruction, _remove_redundant_nops, Block
from pypy.tool import stdlib_opcode as ops
# unit tests for remove_redundant_nops
def create(*args):
assert len(args) % 2 == 0
res = []
for i in range(0, len(args), 2):
res.append(Instruction(args[i], position_info=(args[i+1], -1, -1, -1)))
block = Block()
block.instructions = res
block.next_block = Block()
return block
def check(block, *args):
_remove_redundant_nops(block)
assert len(args) % 2 == 0
got = []
for op in block.instructions:
got.append(op.opcode)
got.append(op.position_info[0])
assert got == list(args)
def test_remove_redundant_nops():
# chains of nops removed to one
block = create(
ops.NOP, 1,
ops.NOP, 1,
ops.NOP, 1,
ops.NOP, 1,
ops.NOP, 1,
ops.NOP, 2,
ops.NOP, 2,
ops.NOP, 2,
)
check(block, ops.NOP, 1, ops.NOP, 2)
# nops without position get removed
block = create(
ops.NOP, 1,
ops.NOP, 1,
ops.NOP, -1,
ops.NOP, 1,
ops.NOP, 1,
ops.NOP, -1,
ops.NOP, 2,
ops.NOP, -1,
ops.NOP, 2,
ops.NOP, -1,
ops.NOP, 2,
ops.NOP, -1,
)
check(block, ops.NOP, 1, ops.NOP, 2)
# nops with the same line as a non-nop op get removed
block = create(
ops.POP_TOP, 1,
ops.NOP, 1,
)
check(block, ops.POP_TOP, 1)
block = create(
ops.NOP, 1,
ops.POP_TOP, 1,
)
check(block, ops.POP_TOP, 1)
# tests that do precise checks on the shape of generated assembly
class TestInstructionDetails(object):
def compile_ast_to_blocks(self, src, mode="exec"):
from pypy.interpreter.pyparser import pyparse
from pypy.interpreter.astcompiler import ast, assemble, symtable, optimize, codegen
space = self.space
p = pyparse.PegParser(space)
info = pyparse.CompileInfo("<test>", mode)
ast = p.parse_source(src, info)
module = optimize.optimize_ast(space, ast, info)
symbols = symtable.SymtableBuilder(space, module, info)
generator = codegen.TopLevelCodeGenerator(space, module, symbols, info)
pycode = generator.assemble()
return pycode, generator._final_blocks
def extract_opnames(self, blocks):
res = []
for block in blocks:
for instr in block.instructions:
res.append(ops.opname[instr.opcode])
return res
def test_match_uses_accept_jump_if(self):
code, blocks = self.compile_ast_to_blocks("""match x:
case [a] if not a:
f()
case _:
g()
""")
opnames = self.extract_opnames(blocks)
assert "UNARY_NOT" not in opnames
|
