Author: hager <[email protected]>
Branch: ppc-jit-backend
Changeset: r47659:fd48c28dba3d
Date: 2011-09-28 17:25 +0200
http://bitbucket.org/pypy/pypy/changeset/fd48c28dba3d/
Log: Wrote tests for regalloc_mov and began with implementation.
diff --git a/pypy/jit/backend/ppc/ppcgen/ppc_assembler.py
b/pypy/jit/backend/ppc/ppcgen/ppc_assembler.py
--- a/pypy/jit/backend/ppc/ppcgen/ppc_assembler.py
+++ b/pypy/jit/backend/ppc/ppcgen/ppc_assembler.py
@@ -325,6 +325,21 @@
clt.asmmemmgr_blocks = []
return clt.asmmemmgr_blocks
+ def regalloc_mov(self, prev_loc, loc):
+ if prev_loc.is_imm():
+ value = prev_loc.getint()
+ # move immediate value to register
+ if loc.is_reg():
+ reg = loc.as_key()
+ self.mc.load_imm(reg, value)
+ # move immediate value to memory
+ else:
+ offset = loc.as_key() * WORD - WORD
+ self.mc.load_imm(r.r0.value, value)
+ self.mc.stw(r.r0, r.SPP, offset)
+ return
+ assert 0, "not supported location"
+
def make_operations():
def not_implemented(builder, trace_op, cpu, *rest_args):
raise NotImplementedError, trace_op
diff --git a/pypy/jit/backend/ppc/ppcgen/test/test_regalloc.py
b/pypy/jit/backend/ppc/ppcgen/test/test_regalloc.py
new file mode 100644
--- /dev/null
+++ b/pypy/jit/backend/ppc/ppcgen/test/test_regalloc.py
@@ -0,0 +1,108 @@
+from pypy.rlib.objectmodel import instantiate
+from pypy.jit.backend.ppc.ppcgen.locations import (imm, RegisterLocation,
+ ImmLocation, StackLocation)
+from pypy.jit.backend.ppc.ppcgen.register import *
+from pypy.jit.backend.ppc.ppcgen.codebuilder import hi, lo
+from pypy.jit.backend.ppc.ppcgen.ppc_assembler import AssemblerPPC
+from pypy.jit.backend.ppc.ppcgen.arch import WORD
+
+class MockBuilder(object):
+
+ def __init__(self):
+ self.reset()
+
+ def __getattr__(self, name):
+ instr = MockInstruction(name)
+ self.instrs.append(instr)
+ return instr
+
+ def str_instrs(self):
+ return [str(inst) for inst in self.instrs]
+
+ def reset(self):
+ self.instrs = []
+
+class MockInstruction(object):
+
+ def __init__(self, name, *args):
+ self.name = name
+ self.args = args
+
+ def __call__(self, *args):
+ self.args = args
+
+ def __eq__(self, other):
+ assert isinstance(other, MockInstruction)
+ #return self.name == other.name and self.args == other.args
+ return str(self) == str(other)
+
+ def __repr__(self):
+ return self.__str__()
+
+ def __str__(self):
+ return "%s %r" % (self.name, self.args)
+
+
+MI = MockInstruction
+
+class TestMocks(object):
+
+ def setup_method(self, method):
+ self.builder = MockBuilder()
+
+ def test_cmp_instruction(self):
+ assert MI("a", 1, 2) == MI("a", 1, 2)
+ assert not MI("a", 1, 2) == MI("b", 1, 2)
+ assert not MI("a", 1, 2) == MI("a", 2, 2)
+ assert not MI("a", 1) == MI("a", 1, 2)
+ assert not MI("a", 1, 2) == MI("a")
+ assert MI("a") == MI("a")
+
+ def test_basic(self):
+ exp_instrs = [MI("mr", 3, 5),
+ MI("foobar"),
+ MI("li", 3, 2),
+ MI("stw", 3, 5, 100)]
+
+ self.builder.mr(3, 5)
+ self.builder.foobar()
+ self.builder.li(3, 2)
+ self.builder.stw(3, 5, 100)
+
+ assert self.builder.instrs == exp_instrs
+
+ self.builder.blub()
+ assert self.builder.instr != exp_instrs
+
+class TestRegallocMov(object):
+
+ def setup_method(self, method):
+ self.builder = MockBuilder()
+ self.asm = instantiate(AssemblerPPC)
+ self.asm.mc = self.builder
+
+ def test_immediate_to_reg(self):
+ self.asm.regalloc_mov(imm(5), reg(10))
+ big = 2 << 28
+ self.asm.regalloc_mov(imm(big), reg(0))
+
+ exp_instr = [MI("load_imm", 10, 5),
+ MI("load_imm", 0, big)]
+ assert self.asm.mc.instrs == exp_instr
+
+ def test_immediate_to_mem(self):
+ self.asm.regalloc_mov(imm(5), stack(6))
+ big = 2 << 28
+ self.asm.regalloc_mov(imm(big), stack(7))
+
+ exp_instr = [MI("load_imm", 0, 5),
+ MI("stw", r0, SPP, -(6 * WORD + WORD)),
+ MI("load_imm", 0, big),
+ MI("stw", r0, SPP, -(7 * WORD + WORD))]
+ assert self.asm.mc.instrs == exp_instr
+
+def reg(i):
+ return RegisterLocation(i)
+
+def stack(i):
+ return StackLocation(i)
_______________________________________________
pypy-commit mailing list
[email protected]
http://mail.python.org/mailman/listinfo/pypy-commit
