def test_constrained_move(self):
""" Test a constrained move.
Do this by creating two pre-colored registers. Move one
and add the other with the copy of the first.
The move can then not be coalesced, and will be frozen.
"""
f = Frame('tst')
t1 = R0
t2 = R0
t3 = ExampleRegister('t3')
t4 = ExampleRegister('t4')
f.instructions.append(Def(t1))
move = Mov(t3, t1, ismove=True)
f.instructions.append(move)
f.instructions.append(Def(t2))
f.instructions.append(Add(t4, t2, t3))
f.instructions.append(Use(t4))
self.register_allocator.alloc_frame(f)
# Check t1 and t2 are pre-colored:
self.assertEqual({self.register_allocator.node(R0)},
self.register_allocator.precolored)
self.assertEqual(set(), self.register_allocator.coalescedMoves)
self.assertEqual({move}, self.register_allocator.constrainedMoves)
self.conflict(t2, t3)
self.assertEqual(set(), self.register_allocator.frozenMoves)
self.assertIn(move, f.instructions)
def test_constrained_move_by_alias(self):
""" Test if aliased registers work and cannot be coalesced. """
f = Frame('tst')
t1 = R10
t2 = R10l
t3 = ExampleRegister('t3')
move = Mov(t3, t1, ismove=True)
f.instructions.append(Def(t1))
f.instructions.append(move)
f.instructions.append(DefHalf(t2))
f.instructions.append(UseHalf(t2))
f.instructions.append(Use(t3))
self.register_allocator.alloc_frame(f)
# Check t1 and t2 are pre-colored:
self.assertEqual(
{
self.register_allocator.node(R10),
self.register_allocator.node(R10l)
}, self.register_allocator.precolored)
self.assertEqual(set(), self.register_allocator.coalescedMoves)
self.assertEqual({move}, self.register_allocator.constrainedMoves)
self.conflict(t2, t3)
self.assertEqual(set(), self.register_allocator.frozenMoves)
self.assertIn(move, f.instructions)
def test_register_coalescing(self):
""" Register coalescing happens when a move can be eliminated """
f = Frame('tst')
t1 = ExampleRegister('t1')
t2 = ExampleRegister('t2')
t3 = ExampleRegister('t3')
t4 = ExampleRegister('t4')
t5 = ExampleRegister('t5')
t6 = ExampleRegister('t6')
f.instructions.append(Def(t1))
f.instructions.append(Def(t2))
f.instructions.append(Def(t3))
f.instructions.append(Add(t4, t2, t1))
f.instructions.append(Mov(t5, t3))
f.instructions.append(Add(t5, t4, t5))
f.instructions.append(Mov(t6, t5))
f.instructions.append(Use(t6))
self.register_allocator.alloc_frame(f)
self.conflict(t1, t2)
self.conflict(t2, t3)
self.conflict(t1, t3)
def test_freeze(self):
""" Create a situation where no select and no coalesc is possible
"""
f = Frame('tst')
t4 = ExampleRegister('t4')
t5 = ExampleRegister('t5')
f.instructions.append(Def(R0))
move = Mov(R1, R0, ismove=True)
f.instructions.append(move)
f.instructions.append(Def(t4))
f.instructions.append(Add(t5, t4, R1))
f.instructions.append(Use(t5))
self.register_allocator.alloc_frame(f)
self.assertEqual(set(), self.register_allocator.coalescedMoves)
# self.assertEqual({move}, self.register_allocator.frozenMoves)
self.conflict(R1, R0)
def test_binary_and_logstream(self):
arch = ExampleArch()
object1 = ObjectFile(arch)
stream = binary_and_logging_stream(object1)
stream.select_section('code')
stream.emit(Mov(R1, R0))
def test_text_stream(self, mock_stdout):
""" Test output to stdout """
arch = ExampleArch()
stream = TextOutputStream()
stream.select_section('code')
stream.emit(Mov(R1, R0))
def test_dummy_stream(self):
arch = ExampleArch()
stream = DummyOutputStream()
stream.select_section('code')
stream.emit(Mov(R1, R0))