File: regalloc.py

package info (click to toggle)
pypy 7.0.0%2Bdfsg-3
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 107,216 kB
  • sloc: python: 1,201,787; ansic: 62,419; asm: 5,169; cpp: 3,017; sh: 2,534; makefile: 545; xml: 243; lisp: 45; awk: 4
file content (80 lines) | stat: -rw-r--r-- 2,334 bytes parent folder | download | duplicates (8)
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
from rpython.jit.metainterp.history import ConstInt, FLOAT
from rpython.jit.backend.ppc.locations import imm

def check_imm_box(arg, lower_bound=-2**15, upper_bound=2**15-1):
    if isinstance(arg, ConstInt):
        i = arg.getint()
        return lower_bound <= i <= upper_bound
    return False

def _check_imm_arg(i):
    return (-2**15) <= i <= (2**15-1)

def _prepare_cmp_op(signed):
    lower_bound = -2**15 if signed else 0
    upper_bound = 2**15-1 if signed else 2**16-1
    def f(self, op):
        l0 = self.ensure_reg(op.getarg(0))
        a1 = op.getarg(1)
        if check_imm_box(a1, lower_bound, upper_bound):
            l1 = imm(a1.getint())
        else:
            l1 = self.ensure_reg(a1)
        self.free_op_vars()
        res = self.force_allocate_reg_or_cc(op)
        return [l0, l1, res]
    return f
prepare_cmp_op          = _prepare_cmp_op(signed=True)
prepare_cmp_op_unsigned = _prepare_cmp_op(signed=False)

def prepare_unary_cmp(self, op):
    l0 = self.ensure_reg(op.getarg(0))
    l1 = imm(0)
    self.free_op_vars()
    res = self.force_allocate_reg_or_cc(op)
    return [l0, l1, res]

def prepare_float_cmp(self, op):
    l0 = self.ensure_reg(op.getarg(0))
    l1 = self.ensure_reg(op.getarg(1))
    self.free_op_vars()
    res = self.force_allocate_reg_or_cc(op)
    return [l0, l1, res]

def prepare_unary_op(self, op):
    l0 = self.ensure_reg(op.getarg(0))
    self.free_op_vars()
    res = self.force_allocate_reg(op)
    return [l0, res]

def prepare_binary_op(self, op):
    reg1 = self.ensure_reg(op.getarg(0))
    reg2 = self.ensure_reg(op.getarg(1))
    self.free_op_vars()
    res = self.force_allocate_reg(op)
    return [reg1, reg2, res]

def prepare_int_add_or_mul(self, op):
    a0 = op.getarg(0)
    a1 = op.getarg(1)
    if check_imm_box(a0):
        a0, a1 = a1, a0
    l0 = self.ensure_reg(a0)
    if check_imm_box(a1):
        l1 = imm(a1.getint())
    else:
        l1 = self.ensure_reg(a1)
    self.free_op_vars()
    res = self.force_allocate_reg(op)
    return [l0, l1, res]

def prepare_int_sub(self, op):
    l0 = self.ensure_reg(op.getarg(0))
    a1 = op.getarg(1)
    if check_imm_box(a1, -2**15+1, 2**15):
        l1 = imm(a1.getint())
    else:
        l1 = self.ensure_reg(a1)
    self.free_op_vars()
    res = self.force_allocate_reg(op)
    return [l0, l1, res]