rpython/jit/metainterp/optimizeopt/earlyforce.py


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

from rpython.jit.codewriter.effectinfo import EffectInfo
from rpython.jit.metainterp.optimizeopt.optimizer import Optimization
from rpython.jit.metainterp.resoperation import rop, OpHelpers


def is_raw_free(op, opnum):
    if not OpHelpers.is_real_call(opnum):
        return False
    einfo = op.getdescr().get_extra_info()
    return einfo.oopspecindex == EffectInfo.OS_RAW_FREE


class OptEarlyForce(Optimization):
    def propagate_forward(self, op):
        opnum = op.getopnum()

        if (opnum != rop.SETFIELD_GC and
            opnum != rop.SETARRAYITEM_GC and
            opnum != rop.SETARRAYITEM_RAW and
            opnum != rop.QUASIIMMUT_FIELD and
            opnum != rop.SAME_AS_I and
            opnum != rop.SAME_AS_R and
            opnum != rop.SAME_AS_F and
            not is_raw_free(op, opnum)):

            for arg in op.getarglist():
                self.optimizer.force_box(arg, self)
        return self.emit(op)

    def setup(self):
        self.optimizer.optearlyforce = self