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
104
105
106
107
108
109
110
111
112
113
|
from rpython.jit.backend.aarch64 import registers as r
def remap_frame_layout(assembler, src_locations, dst_locations, tmpreg):
pending_dests = len(dst_locations)
srccount = {} # maps dst_locations to how many times the same
# location appears in src_locations
for dst in dst_locations:
key = dst.as_key()
assert key not in srccount, "duplicate value in dst_locations!"
srccount[key] = 0
for i in range(len(dst_locations)):
src = src_locations[i]
if src.is_imm():
continue
key = src.as_key()
if key in srccount:
if key == dst_locations[i].as_key():
# ignore a move "x = x"
# setting any "large enough" negative value is ok, but
# be careful of overflows, don't use -sys.maxint
srccount[key] = -len(dst_locations) - 1
pending_dests -= 1
else:
srccount[key] += 1
while pending_dests > 0:
progress = False
for i in range(len(dst_locations)):
dst = dst_locations[i]
key = dst.as_key()
if srccount[key] == 0:
srccount[key] = -1 # means "it's done"
pending_dests -= 1
src = src_locations[i]
if not src.is_imm():
key = src.as_key()
if key in srccount:
srccount[key] -= 1
_move(assembler, src, dst, tmpreg)
progress = True
if not progress:
# we are left with only pure disjoint cycles
sources = {} # maps dst_locations to src_locations
for i in range(len(dst_locations)):
src = src_locations[i]
dst = dst_locations[i]
sources[dst.as_key()] = src
#
for i in range(len(dst_locations)):
dst = dst_locations[i]
originalkey = dst.as_key()
if srccount[originalkey] >= 0:
assembler.push_locations([dst])
while True:
key = dst.as_key()
assert srccount[key] == 1
# ^^^ because we are in a simple cycle
srccount[key] = -1
pending_dests -= 1
src = sources[key]
if src.as_key() == originalkey:
break
_move(assembler, src, dst, tmpreg)
dst = src
assembler.pop_locations([dst])
assert pending_dests == 0
def _move(assembler, src, dst, tmpreg):
if dst.is_stack() and src.is_stack():
assembler.regalloc_mov(src, tmpreg)
src = tmpreg
assembler.regalloc_mov(src, dst)
def remap_frame_layout_mixed(assembler,
src_locations1, dst_locations1, tmpreg1,
src_locations2, dst_locations2, tmpreg2):
# find and push the xmm stack locations from src_locations2 that
# are going to be overwritten by dst_locations1
extrapushes = []
extrapops = []
dst_keys = {}
for loc in dst_locations1:
dst_keys[loc.as_key()] = None
src_locations2red = []
dst_locations2red = []
for i in range(len(src_locations2)):
loc = src_locations2[i]
dstloc = dst_locations2[i]
if loc.is_stack():
key = loc.as_key()
if key in dst_keys:
extrapushes.append(loc)
extrapops.append(dstloc)
continue
src_locations2red.append(loc)
dst_locations2red.append(dstloc)
src_locations2 = src_locations2red
dst_locations2 = dst_locations2red
assembler.push_locations(extrapushes)
#
# remap the integer and pointer registers and stack locations
remap_frame_layout(assembler, src_locations1, dst_locations1, tmpreg1)
#
# remap the vfp registers and stack locations
remap_frame_layout(assembler, src_locations2, dst_locations2, tmpreg2)
#
# finally, pop the extra xmm stack locations
assembler.pop_locations(extrapops)
|
