def consider_jump(self, op):
assembler = self.assembler
assert self.jump_target_descr is None
descr = op.descr
assert isinstance(descr, LoopToken)
self.jump_target_descr = descr
nonfloatlocs, floatlocs = assembler.target_arglocs(
self.jump_target_descr)
# compute 'tmploc' to be all_regs[0] by spilling what is there
box = TempBox()
box1 = TempBox()
tmpreg = X86RegisterManager.all_regs[0]
tmploc = self.rm.force_allocate_reg(box, selected_reg=tmpreg)
xmmtmp = X86XMMRegisterManager.all_regs[0]
xmmtmploc = self.xrm.force_allocate_reg(box1, selected_reg=xmmtmp)
# Part about non-floats
src_locations = [self.loc(arg) for arg in op.args if arg.type != FLOAT]
assert tmploc not in nonfloatlocs
dst_locations = [loc for loc in nonfloatlocs if loc is not None]
remap_frame_layout(assembler, src_locations, dst_locations, tmploc)
# Part about floats
src_locations = [self.loc(arg) for arg in op.args if arg.type == FLOAT]
dst_locations = [loc for loc in floatlocs if loc is not None]
remap_frame_layout(assembler, src_locations, dst_locations, xmmtmp)
self.rm.possibly_free_var(box)
self.xrm.possibly_free_var(box1)
self.possibly_free_vars(op.args)
assembler.closing_jump(self.jump_target_descr)
def test_constants_and_cycle():
assembler = MockAssembler()
c3 = imm(3)
s12 = frame_pos(13, INT)
remap_frame_layout(assembler, [ebx, c3, s12], [s12, eax, ebx], edi)
assert assembler.ops == [('mov', c3, eax), ('push', s12),
('mov', ebx, s12), ('pop', ebx)]
def test_cycle_2():
assembler = MockAssembler()
s8 = frame_pos(8, INT)
s12 = frame_pos(12, INT)
s20 = frame_pos(19, INT)
s24 = frame_pos(1, INT)
s2 = frame_pos(2, INT)
s3 = frame_pos(3, INT)
remap_frame_layout(assembler,
[eax, s8, edi, s20, eax, s20, s24, esi, s2, s3],
[s8, s20, edi, eax, edx, s24, ebx, s12, s3, s2],
ecx)
assert assembler.got([('mov', eax, edx),
('mov', s24, ebx),
('mov', esi, s12),
('mov', s20, ecx),
('mov', ecx, s24),
('push', s8),
('mov', eax, s8),
('mov', s20, eax),
('pop', s20),
('push', s3),
('mov', s2, ecx),
('mov', ecx, s3),
('pop', s2)])
def consider_jump(self, op):
assembler = self.assembler
assert self.jump_target_descr is None
descr = op.descr
assert isinstance(descr, LoopToken)
self.jump_target_descr = descr
nonfloatlocs, floatlocs = assembler.target_arglocs(self.jump_target_descr)
# compute 'tmploc' to be all_regs[0] by spilling what is there
box = TempBox()
box1 = TempBox()
tmpreg = X86RegisterManager.all_regs[0]
tmploc = self.rm.force_allocate_reg(box, selected_reg=tmpreg)
xmmtmp = X86XMMRegisterManager.all_regs[0]
xmmtmploc = self.xrm.force_allocate_reg(box1, selected_reg=xmmtmp)
# Part about non-floats
src_locations = [self.loc(arg) for arg in op.args if arg.type != FLOAT]
assert tmploc not in nonfloatlocs
dst_locations = [loc for loc in nonfloatlocs if loc is not None]
remap_frame_layout(assembler, src_locations, dst_locations, tmploc)
# Part about floats
src_locations = [self.loc(arg) for arg in op.args if arg.type == FLOAT]
dst_locations = [loc for loc in floatlocs if loc is not None]
remap_frame_layout(assembler, src_locations, dst_locations, xmmtmp)
self.rm.possibly_free_var(box)
self.xrm.possibly_free_var(box1)
self.possibly_free_vars(op.args)
assembler.closing_jump(self.jump_target_descr)
def test_cycle():
assembler = MockAssembler()
s8 = frame_pos(8, INT)
s12 = frame_pos(12, INT)
s20 = frame_pos(19, INT)
s24 = frame_pos(1, INT)
remap_frame_layout(assembler, [eax, s8, s20, ebx], [s8, ebx, eax, s20], "?")
assert assembler.got([("push", s8), ("mov", eax, s8), ("mov", s20, eax), ("mov", ebx, s20), ("pop", ebx)])
def test_reordering():
assembler = MockAssembler()
s8 = frame_pos(8, INT)
s12 = frame_pos(12, INT)
s20 = frame_pos(19, INT)
s24 = frame_pos(1, INT)
remap_frame_layout(assembler, [eax, s8, s20, ebx], [s8, ebx, eax, edi], "?")
assert assembler.got([("mov", ebx, edi), ("mov", s8, ebx), ("mov", eax, s8), ("mov", s20, eax)])
def test_no_tmp_reg():
assembler = MockAssembler()
s8 = frame_pos(0, INT)
s12 = frame_pos(13, INT)
s20 = frame_pos(20, INT)
s24 = frame_pos(221, INT)
remap_frame_layout(assembler, [s8, eax, s12], [s20, s24, edi], None)
assert assembler.ops == [("push", s8), ("pop", s20), ("mov", eax, s24), ("mov", s12, edi)]
def test_simple_framelocs():
assembler = MockAssembler()
s8 = frame_pos(0, INT)
s12 = frame_pos(13, INT)
s20 = frame_pos(20, INT)
s24 = frame_pos(221, INT)
remap_frame_layout(assembler, [s8, eax, s12], [s20, s24, edi], edx)
assert assembler.ops == [("mov", s8, edx), ("mov", edx, s20), ("mov", eax, s24), ("mov", s12, edi)]
def test_simple_framelocs():
assembler = MockAssembler()
s8 = frame_pos(0, INT)
s12 = frame_pos(13, INT)
s20 = frame_pos(20, INT)
s24 = frame_pos(221, INT)
remap_frame_layout(assembler, [s8, eax, s12], [s20, s24, edi], edx)
assert assembler.ops == [('mov', s8, edx), ('mov', edx, s20),
('mov', eax, s24), ('mov', s12, edi)]
def test_constants():
assembler = MockAssembler()
c3 = imm(3)
remap_frame_layout(assembler, [c3], [eax], '?')
assert assembler.ops == [('mov', c3, eax)]
assembler = MockAssembler()
s12 = frame_pos(12, INT)
remap_frame_layout(assembler, [c3], [s12], '?')
assert assembler.ops == [('mov', c3, s12)]
def test_constants():
assembler = MockAssembler()
c3 = imm(3)
remap_frame_layout(assembler, [c3], [eax], '?')
assert assembler.ops == [('mov', c3, eax)]
assembler = MockAssembler()
s12 = frame_pos(12, INT)
remap_frame_layout(assembler, [c3], [s12], '?')
assert assembler.ops == [('mov', c3, s12)]
def test_no_tmp_reg():
assembler = MockAssembler()
s8 = frame_pos(0, INT)
s12 = frame_pos(13, INT)
s20 = frame_pos(20, INT)
s24 = frame_pos(221, INT)
remap_frame_layout(assembler, [s8, eax, s12], [s20, s24, edi], None)
assert assembler.ops == [('push', s8), ('pop', s20), ('mov', eax, s24),
('mov', s12, edi)]
def test_reordering():
assembler = MockAssembler()
s8 = frame_pos(8, INT)
s12 = frame_pos(12, INT)
s20 = frame_pos(19, INT)
s24 = frame_pos(1, INT)
remap_frame_layout(assembler, [eax, s8, s20, ebx], [s8, ebx, eax, edi],
'?')
assert assembler.got([('mov', ebx, edi), ('mov', s8, ebx),
('mov', eax, s8), ('mov', s20, eax)])
def test_cycle():
assembler = MockAssembler()
s8 = frame_pos(8, INT)
s12 = frame_pos(12, INT)
s20 = frame_pos(19, INT)
s24 = frame_pos(1, INT)
remap_frame_layout(assembler, [eax, s8, s20, ebx], [s8, ebx, eax, s20],
'?')
assert assembler.got([('push', s8), ('mov', eax, s8), ('mov', s20, eax),
('mov', ebx, s20), ('pop', ebx)])
def test_constants_and_cycle():
assembler = MockAssembler()
c3 = imm(3)
s12 = frame_pos(13, INT)
remap_frame_layout(assembler, [ebx, c3, s12],
[s12, eax, ebx], edi)
assert assembler.ops == [('mov', c3, eax),
('push', s12),
('mov', ebx, s12),
('pop', ebx)]
def test_reordering():
assembler = MockAssembler()
s8 = frame_pos(8, 1)
s12 = frame_pos(12, 1)
s20 = frame_pos(19, 1)
s24 = frame_pos(1, 1)
remap_frame_layout(assembler, [eax, s8, s20, ebx],
[s8, ebx, eax, edi], '?')
assert assembler.got([('mov', ebx, edi),
('mov', s8, ebx),
('mov', eax, s8),
('mov', s20, eax)])
def test_cycle():
assembler = MockAssembler()
s8 = frame_pos(8, INT)
s12 = frame_pos(12, INT)
s20 = frame_pos(19, INT)
s24 = frame_pos(1, INT)
remap_frame_layout(assembler, [eax, s8, s20, ebx],
[s8, ebx, eax, s20], '?')
assert assembler.got([('push', s8),
('mov', eax, s8),
('mov', s20, eax),
('mov', ebx, s20),
('pop', ebx)])
def test_cycle_2():
assembler = MockAssembler()
s8 = frame_pos(8, INT)
s12 = frame_pos(12, INT)
s20 = frame_pos(19, INT)
s24 = frame_pos(1, INT)
s2 = frame_pos(2, INT)
s3 = frame_pos(3, INT)
remap_frame_layout(assembler,
[eax, s8, edi, s20, eax, s20, s24, esi, s2, s3],
[s8, s20, edi, eax, edx, s24, ebx, s12, s3, s2], ecx)
assert assembler.got([('mov', eax, edx), ('mov', s24, ebx),
('mov', esi, s12), ('mov', s20, ecx),
('mov', ecx, s24), ('push', s8), ('mov', eax, s8),
('mov', s20, eax), ('pop', s20), ('push', s3),
('mov', s2, ecx), ('mov', ecx, s3), ('pop', s2)])
def test_trivial():
assembler = MockAssembler()
remap_frame_layout(assembler, [], [], "?")
assert assembler.ops == []
remap_frame_layout(assembler, [eax, ebx, ecx, edx, esi, edi], [eax, ebx, ecx, edx, esi, edi], "?")
assert assembler.ops == []
s8 = frame_pos(1, INT)
s12 = frame_pos(31, INT)
s20 = frame_pos(6, INT)
remap_frame_layout(
assembler, [eax, ebx, ecx, s20, s8, edx, s12, esi, edi], [eax, ebx, ecx, s20, s8, edx, s12, esi, edi], "?"
)
assert assembler.ops == []
def test_trivial():
assembler = MockAssembler()
remap_frame_layout(assembler, [], [], '?')
assert assembler.ops == []
remap_frame_layout(assembler, [eax, ebx, ecx, edx, esi, edi],
[eax, ebx, ecx, edx, esi, edi], '?')
assert assembler.ops == []
s8 = frame_pos(1, INT)
s12 = frame_pos(31, INT)
s20 = frame_pos(6, INT)
remap_frame_layout(assembler, [eax, ebx, ecx, s20, s8, edx, s12, esi, edi],
[eax, ebx, ecx, s20, s8, edx, s12, esi, edi], '?')
assert assembler.ops == []
def main():
srclocs = [StackLoc(9999, x, 'i') for x, y in CASE]
dstlocs = [StackLoc(9999, y, 'i') for x, y in CASE]
remap_frame_layout(FakeAssembler(), srclocs, dstlocs, eax)
def test_simple_registers():
assembler = MockAssembler()
remap_frame_layout(assembler, [eax, ebx, ecx], [edx, esi, edi], '?')
assert assembler.ops == [('mov', eax, edx),
('mov', ebx, esi),
('mov', ecx, edi)]
def main():
srclocs = [StackLoc(9999, x, "i") for x, y in CASE]
dstlocs = [StackLoc(9999, y, "i") for x, y in CASE]
remap_frame_layout(FakeAssembler(), srclocs, dstlocs, eax)
def test_simple_registers():
assembler = MockAssembler()
remap_frame_layout(assembler, [eax, ebx, ecx], [edx, esi, edi], '?')
assert assembler.ops == [('mov', eax, edx), ('mov', ebx, esi),
('mov', ecx, edi)]
