ّRMc @sddkZddkZddklZddklZddklZddklZddk l Z dZ dZ d Z d Zd Zd d ZdZdefdYZdZedddZdefdYZdZde_ei dei dei dei ddZde_dZde_de_defdYZd Zd!e_d"Zd#e_d$Zd%efd&YZ d'e fd(YZ!defd)YZdefd*YZedZ#d+efd,YZ$dZ%dZ&d-Z'hd.d/<d0d1<d2d3<e(d4<e'd5<d.d6YZ/d?efd@YZ0dS(BiN(tExtRegistryEntry(tCDefinedIntSymbolic(tkeepalive_until_here(tunrolling_iterable(t NonConstantcCs t|_|S(s Decorate a function as pure. Pure means precisely that: (1) the result of the call should not change if the arguments are the same (same numbers or same pointers) (2) it's fine to remove the call completely if we can guess the result according to rule 1 Most importantly it doesn't mean that pure function has no observable side effect, but those side effects can be ommited (ie caching). For now, such a function should never raise an exception. (tTruet_pure_function_(tfunc((s!/64/home/arigo/u/pypy/rlib/jit.pyt purefunction s cKs|S(s; Hint for the JIT possible arguments are: XXX ((txtkwds((s!/64/home/arigo/u/pypy/rlib/jit.pythintscCs t|_|S(s` Make sure the JIT does not trace inside decorated function (it becomes a call instead) (tFalset_jit_look_inside_(R((s!/64/home/arigo/u/pypy/rlib/jit.pytdont_look_inside s cCs t|_|S(s_ JIT can safely unroll loops in this function and this will not lead to code explosion (Rt_jit_unroll_safe_(R((s!/64/home/arigo/u/pypy/rlib/jit.pyt unroll_safe's cCst|t|_|S(s Describes a function with no argument that returns an object that is always the same in a loop. Use it only if you know what you're doing. (RRt_jit_loop_invariant_(R((s!/64/home/arigo/u/pypy/rlib/jit.pytloop_invariant.s  tallcsfd}|S(sV A decorator that promotes all arguments and then calls the supplied function c sddk}e||i|\}}}}g}ee|D]}|d|fqH~}|djo |djpt| ptdi|}d|fg} djo8g} idD]}| de |q~ }nx%|D]} | i d| | fqW| i d |fh|d <e d <} e i id i| i| U| d } |id| _| S(Nisv%ss, s def f(%s): Rt,sv%ds %s = hint(%s, promote=True) s return func(%s) RR s tft_promote(tinspectRt getargspectrangetlentNonetAssertionErrortjointsplittinttappendR tpytcodetSourcetcompilet func_name(RRtargstvarargstvarkwtdefaultst_[1]tit argstringR"t_[2]targtdtresult(t promote_args(s!/64/home/arigo/u/pypy/rlib/jit.pyt decorator<s&  4! 8# ((R1R2((R1s!/64/home/arigo/u/pypy/rlib/jit.pytpurefunction_promote8scsfd}|S(Ncs |_|S(N(toopspec(R(tspec(s!/64/home/arigo/u/pypy/rlib/jit.pyR2Rs ((R5R2((R5s!/64/home/arigo/u/pypy/rlib/jit.pyR4QstEntrycBs eZeZdZdZRS(c Ksddkl}|i|}d|j}d|j}|p|o|p tdt||ioddkl}|ii }|i d|di }|dj oM|i i} t| d<|ot| d R7t SomeIntegerR(RJRM((s!/64/home/arigo/u/pypy/rlib/jit.pyROscCs-ddkl}|i|i|itS(Ni(RP(RTRPR_R]tSignedt_we_are_jitted(RJRbRP((s!/64/home/arigo/u/pypy/rlib/jit.pyRjs (RkRlRnRmRORj(((s!/64/home/arigo/u/pypy/rlib/jit.pyR6s cCsttodSndS(seDuring JIT tracing, returns the current trace length (as a constant). If not tracing, returns -1.iIi(RR (((s!/64/home/arigo/u/pypy/rlib/jit.pytcurrent_trace_lengths sjit.current_trace_length()icCst|dS(sqWhen JITted, cause an extra operation JIT_DEBUG to appear in the graphs. Should not be left after debugging.N(R(tstringtarg1targ2targ3targ4((s!/64/home/arigo/u/pypy/rlib/jit.pyt jit_debugss)jit.debug(string, arg1, arg2, arg3, arg4)cCst|dS(sUVery strong assert: checks that 'value' is a green (a JIT compile-time constant).N(R(RF((s!/64/home/arigo/u/pypy/rlib/jit.pyt assert_greenssspecialize:argtype(0)sjit.assert_green(value)tAssertGreenFailedcBseZRS((RkRl(((s!/64/home/arigo/u/pypy/rlib/jit.pyR|scCs t|S(sCreates a 'vref' object that contains a reference to 'x'. Calls to virtual_ref/virtual_ref_finish must be properly nested. The idea is that the object 'x' is supposed to be JITted as a virtual between the calls to virtual_ref and virtual_ref_finish, but the 'vref' object can escape at any point in time. If at runtime it is dereferenced (by the call syntax 'vref()'), it returns 'x', which is then forced.(t DirectJitVRef(R ((s!/64/home/arigo/u/pypy/rlib/jit.pyt virtual_refs svirtual_ref(x)cCst|dS(ssSee docstring in virtual_ref(x). Note that virtual_ref_finish takes as argument the real object, not the vref.N(R(R ((s!/64/home/arigo/u/pypy/rlib/jit.pytvirtual_ref_finishssvirtual_ref_finish(x)cCs t|S(szCreates a 'vref' that just returns x when called; nothing more special. Used for None or for frames outside JIT scope.(t DirectVRef(R ((s!/64/home/arigo/u/pypy/rlib/jit.pytnon_virtual_refsRcBseZdZdZRS(cCs ||_dS(N(t_x(RJR ((s!/64/home/arigo/u/pypy/rlib/jit.pyt__init__scCs|iS(N(R(RJ((s!/64/home/arigo/u/pypy/rlib/jit.pyt__call__s(RkRlRR(((s!/64/home/arigo/u/pypy/rlib/jit.pyRs R}cBseZdZRS(cCs+|dj p tdti||dS(Ns virtual_ref(None) is not allowed(RRRR(RJR ((s!/64/home/arigo/u/pypy/rlib/jit.pyRs(RkRlR(((s!/64/home/arigo/u/pypy/rlib/jit.pyR}scBs&eZeefZdZdZRS(cCsddkl}|i|S(Ni(t _jit_vref(t pypy.rlibRtSomeVRef(RJts_objR((s!/64/home/arigo/u/pypy/rlib/jit.pyROscCs|ii|S(N(tr_resultRj(RJRb((s!/64/home/arigo/u/pypy/rlib/jit.pyRjs(RkRlRR}RmRORj(((s!/64/home/arigo/u/pypy/rlib/jit.pyR6s  cBseZeZdZRS(cCsOddkl}t|itpt|ii|i}|i|S(Ni(R( RRR@tinstanceRRt bookkeepertimmutablevalueR(RJRR((s!/64/home/arigo/u/pypy/rlib/jit.pytcompute_annotations(RkRlRt_type_R(((s!/64/home/arigo/u/pypy/rlib/jit.pyR6st JitHintErrorcBseZdZRS(sInconsistency in the JIT hints.(RkRlt__doc__(((s!/64/home/arigo/u/pypy/rlib/jit.pyRsiit thresholdittrace_eagernessi't trace_limittinliningt optimizertloop_longevityt JitDriverc BseZdZeZgZd d d d d d d d dZdZdZ dZ dZ dZ dZ de _d Zde_d ZRS( sBase class to declare fine-grained user control on the JIT. So far, there must be a singleton instance of JitDriver. This style will allow us (later) to support a single RPython program with several independent JITting interpreters in it. c Cs.|dj o ||_n|dj o ||_nt|d pt|d otdn|dj o ||_nx(|iD]} | |ijptqWtig} |i|iD]} d| jo | | qq~ |_ |i ||_ ||_ ||_ ||_||_dS(Ntgreenstredssno 'greens' or 'reds' suppliedt.(RRRthasattrtAttributeErrortvirtualizablesRtdicttfromkeyst _alllivevarst_make_extregistryentriestget_jitcell_attset_jitcell_attget_printable_locationtconfirm_enter_jittcan_never_inline( RJRRRRRRRRRhR*tname((s!/64/home/arigo/u/pypy/rlib/jit.pyRs&    "   B     cCstS(N(R(RJ((s!/64/home/arigo/u/pypy/rlib/jit.pyt_freeze_3scKs$ti||ijptdS(N(RRRR(t_selftlivevars((s!/64/home/arigo/u/pypy/rlib/jit.pytjit_merge_point6scKs$ti||ijptdS(N(RRRR(RR((s!/64/home/arigo/u/pypy/rlib/jit.pyt can_enter_jit:scCsdS(N((RJ((s!/64/home/arigo/u/pypy/rlib/jit.pyt loop_header>scCs|tjptdS(N(t PARAMETERSR(RJRRF((s!/64/home/arigo/u/pypy/rlib/jit.pyt _set_paramBscCsFx3tD]+}||jo|i||dSqqWtddS(s%Set one of the tunable JIT parameter.Nsno such parameter(tunroll_parametersRt ValueError(RJRRFtname1((s!/64/home/arigo/u/pypy/rlib/jit.pyt set_paramGs   sspecialize:arg(0)cCsx|idD]}|id}|id}t|djo tnyt|d}Wntj o nX|d}|i||qWdS(sSet the tunable JIT parameters from a user-supplied string following the format 'param=value,param=value'. For programmatic setting of parameters, use directly JitDriver.set_param(). Rt t=iiiN(RtstripRRRR(RJttexttstpartsRFR((s!/64/home/arigo/u/pypy/rlib/jit.pytset_user_paramPs  csi_i_i_i_dtffdY}dtffdY}dtffdY}dS(NR6cseZiifZRS((RkRlRRRm((RJ(s!/64/home/arigo/u/pypy/rlib/jit.pyR6lscseZiZRS((RkRlRRm((RJ(s!/64/home/arigo/u/pypy/rlib/jit.pyR6oscseZiZRS((RkRlRRm((RJ(s!/64/home/arigo/u/pypy/rlib/jit.pyR6rs(RRRRtExtEnterLeaveMarkert ExtLoopHeadert ExtSetParam(RJR6((RJs!/64/home/arigo/u/pypy/rlib/jit.pyRbs    N(RkRlRRtactiveRRRRRRRRRt_annspecialcase_RR(((s!/64/home/arigo/u/pypy/rlib/jit.pyRs"           t BaseJitCellcBseZdZRS(((RkRlt __slots__(((s!/64/home/arigo/u/pypy/rlib/jit.pyRysRcBs/eZdZdZgdZdZRS(c Ksddkl}|iidjo|i|n|ii}|i}|ig}|i|i D]"}d|jo|d|qkqk~}|i||jot d|i|fny|i i |}WnVt j o"h}h||<|i _ n*tj oh}||i i |R7RRktannotate_hookstim_selftkeystsortRRRRt_jit_annotation_cacheRtKeyErrorRUtgetts_ImpossibleValuetunionoft isdegeneratedts_None( RJRLRMtdriverRR*RtexpectedtcacheReRgt s_previous((s!/64/home/arigo/u/pypy/rlib/jit.pyROs<   %   cKss|ii}|iit}|i}||i|i|||i|i|g|||i |i|dS(N( RRRt valueoftypeRt annotate_hookRRRR(RJRLRt s_jitcellth((s!/64/home/arigo/u/pypy/rlib/jit.pyRs   cKs|djodSn|i}|i|}d|i}|}x|D]}d|jo|d|} no|id\} } |d| } | ii| } |ii}t| i |<| i } | dj pt |i | qHW|i |||dS(Ns jitdriver.%sRR(RRRR%RRCtfind_attributet position_keyRtread_locationstgetvalueRR temulate_pbc_call(RJRt variablesRVRLtbkts_funct uniquekeyRts_argtobjnamet fieldnamet s_instancetattrdeftposition((s!/64/home/arigo/u/pypy/rlib/jit.pyRs&       c Ks&ddkl}ddkl}|ii}g}g}x-|iD]"}d|jo4|d|} |i| } |i| d| } n|i i i djot i idn|id\} } | |ijpt|d| } |i| }|i| }xotogy|i| \}}PWntj onX|idj ptd ||iif|i}q W|ii}|i|ptd |t|d p h|_n||f|i|<|i|d| }|i|i|}|id ||gd |} |i i!| }|dj pt|i i"i#| ||i$| qBWxN|iD]C}|d|} |i| }|i|d| }|i$|qrW|i%|i|i|ii&|i|i|g}|i'||i'||id|d |iS(Ni(RQ(RPRRRR.t ootypesystems lltype onlysfield %r not found in %rs&field %r must be declared as immutabletll_greenfieldstgetfieldRSt jit_marker((RXRQRTRPRRRR\R[trtypert type_systemRR!ttesttskipRRRRVRt _get_fieldRtrbaseRt lowleveltypetTOt_immutable_fieldRRR]R^R`RCtabout_attributet annotatort setbindingR R_Rktextend(RJRbRcRQRPRtgreens_vtreds_vRR+tr_greentv_greenRRts_redtr_redt mangled_nametr_fieldtGTYPEtv_redtc_llnamets_greentvlist((s!/64/home/arigo/u/pypy/rlib/jit.pyRjsp               (RkRlRORRRj(((s!/64/home/arigo/u/pypy/rlib/jit.pyR}s $  RcBseZdZdZRS(cKsddkl}|iS(Ni(R7(R>R7R(RJRLRM((s!/64/home/arigo/u/pypy/rlib/jit.pyROscCsiddkl}|ii}|i|i|id|i|i|g}|id|d|iS(Ni(RPRRRS(RTRPRRR_R]R^R`(RJRbRPRR((s!/64/home/arigo/u/pypy/rlib/jit.pyRjs  (RkRlRORj(((s!/64/home/arigo/u/pypy/rlib/jit.pyRs RcBseZdZdZRS(cCsHddkl}|ipt|ii|pt|iS(Ni(R7(R>R7RWRRqtcontainsR(RJts_nameRgRM((s!/64/home/arigo/u/pypy/rlib/jit.pyROscCsddkl}|i|ii}|idi}|i|idd}|i |i d|i |i ||i |i ||g}|i d|d|i S( Ni(RPiR.iRRRS( RTRPR_RRRVRZR[RrR]R^R`(RJRbRPRRtv_valueR((s!/64/home/arigo/u/pypy/rlib/jit.pyRjs   (RkRlRORj(((s!/64/home/arigo/u/pypy/rlib/jit.pyRs ((1R!tsystpypy.rpython.extregistryRtpypy.rlib.objectmodelRRtpypy.rlib.unrollRtpypy.rlib.nonconstRRR RRRR3R4R6RnRsRttmaxintRzR{Rt ExceptionR|R~RRtobjectRR}Rt vref_NoneRtOPTIMIZER_SIMPLEtOPTIMIZER_NO_UNROLLtOPTIMIZER_FULLR RRRRRRRR(((s!/64/home/arigo/u/pypy/rlib/jit.pyssj        )                     g