\ K]c@`s dZddlmZmZmZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZddlmZmZmZddlmZddlZddlZddlmZmZmZmZmZmZmZmZmZddlm Z m!Z!m"Z"m#Z#m$Z$dd l%m&Z&dd l'm(Z(m)Z)dd lm*Z*dd l+m,Z,dd l-m.Z.ddl/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYddlZm[Z[m\Z\m]Z]m^Z^ddl_m`Z`ddlambZbmcZcddldmeZeddlfmgZgmhZhmiZimjZjmkZkmlZlddl-mmZmddl Z ddlnZnddlnZodZpe jqdepderZsdZterZudZvemevdZwdexfdYZydZzdZ{dZ|d Z}d!Z~d"Zd#Zd$Zd%Zd&Zd'Zd(Zd)Zd*Zd+Zid,d-d.f6d/d0d.f6ezd1d.f6e{d2d.f6ezd3d.f6e{d4d.f6ed5d.f6ed6d.f6ed7d.f6ed8d.f6ed9d.f6ed:d.f6ed;d.f6ed<d.f6Ze.d=Ze.d>Ze.d?Ze.d@ZedAdBdCgZiedDed-d.f6edEed0d.f6edFed1d.f6edGed2d.f6edFed3d.f6edGed4d.f6ZdHexfdIYZdJejejfdKYZdLZee jeTsc S`stjj|||}tj|j}tj|j}ttt ||d}t j ||}x,tjj |D]} || ||| ttargmintnumpycC`stS(N(R(RR((s+lib/python2.7/site-packages/numba/parfor.pyRRtargmaxRqR}tamintamaxtsumtprodRRtstdtdotRRcC`s|dkrtdndS(NisDzero-size array to reduction operation maximum which has no identity(R(tarr_size((s+lib/python2.7/site-packages/numba/parfor.pyR{s cC`s|dkrtdndS(NisDzero-size array to reduction operation minimum which has no identity(R(R((s+lib/python2.7/site-packages/numba/parfor.pyRks cC`s|dkrtdndS(Nis*attempt to get argmin of an empty sequence(R(R((s+lib/python2.7/site-packages/numba/parfor.pyRs cC`s|dkrtdndS(Nis*attempt to get argmax of an empty sequence(R(R((s+lib/python2.7/site-packages/numba/parfor.pyRs t checker_impltnametfuncRRRkR{tLoopNestcB`s)eZdZdZdZdZRS(sThe LoopNest class holds information of a single loop including the index variable (of a non-negative integer value), and the range variable, e.g. range(r) is 0 to r-1 with step size 1. cC`s(||_||_||_||_dS(N(tindex_variableRRR(tselfRRRR((s+lib/python2.7/site-packages/numba/parfor.pyt__init__s   cC`s"dj|j|j|j|jS(Ns3LoopNest(index_variable = {}, range = ({}, {}, {}))(RRRRR(R((s+lib/python2.7/site-packages/numba/parfor.pyt__repr__scC`sg}|j|jt|jtjr>|j|jnt|jtjrf|j|jnt|jtjr|j|jn|S(N(tappendRRRRtVarRR(Rtall_uses((s+lib/python2.7/site-packages/numba/parfor.pyt list_varss(RgRht__doc__RRR(((s+lib/python2.7/site-packages/numba/parfor.pyRs  tParforcB`sJeZdZeedZdZdZddZ ddZ RS(ic C`stt|jddd|t|j|_t|jd7_||_||_||_||_ d|_ ||_ t |dkst|g|_||_| |_| |_tjrd} t| j|j||ndS(NtopRjtlocis9Parallel for-loop #{} is produced from pattern '{}' at {}(tsuperRRttypet id_countertidt loop_nestst init_blockt loop_bodyt index_vartNonetparamst equiv_setRltAssertionErrortpatternstflagstno_sequential_loweringtracesR tDEBUG_ARRAY_OPT_STATSRZR( RRRRRRRtpatternRRRtfmt((s+lib/python2.7/site-packages/numba/parfor.pyRs(             cC`s8dt|jt|jt|jt|jS(Nsid=(tstrRtreprRRR(R((s+lib/python2.7/site-packages/numba/parfor.pyR scC`sg}xA|jjD]0\}}x!|jD]}||j7}q,WqWx!|jD]}||j7}qTWx$|jjD]}||j7}q{W|S(sslist variables used (read/written) in this parfor by traversing the body and combining block uses. (RtitemstbodyRRR(RRR\Rtstmttloop((s+lib/python2.7/site-packages/numba/parfor.pyRscC`sX|dk r'|jj}||j_n|jj|}|dk rT||j_n|S(svget the shape classes for a given variable. If a typemap is specified then use it for type resolution N(RRttypemaptget_shape_classes(RRRt save_typemaptres((s+lib/python2.7/site-packages/numba/parfor.pyRs   cC`s|p tj}tdj|jjddd|td|jd|x!|jD]}t|d|qWWtdd||jj |xGt |j j D]0\}}td|fd||j |qWtdj|jjddd|dS( Nsbegin parfor {}it-tfiles index_var = s init block:s label %s:s end parfor {}( tsyststdoutRZRRtcenterRRRtdumptsortedRR(RR tloopnesttoffsettblock((s+lib/python2.7/site-packages/numba/parfor.pyR 1s("N( RgRhRtFalsetsetRRRRRR (((s+lib/python2.7/site-packages/numba/parfor.pyRs    cC`sw|j}t|}|jjdd}|j||t|||jd|_|rm||jds   tParforDiagnosticscB`seZdZdZdZedZejdZedZ dZ dZ dZ d Z d Zd Zd Zd ZddZdZdZRS(sHolds parfor diagnostic info, this is accumulated throughout the PreParforPass and ParforPass, also in the closure inlining! cC`s[d|_t|_d|_tt|_tt|_g|_ i|_ t |_ dS(Nt__numba_parfor_gufunc( RRtdictt replaced_fnst internal_nameRtlistt fusion_infotnested_fusion_infotfusion_reportst hoist_infoRt has_setup(R((s+lib/python2.7/site-packages/numba/parfor.pyRTs     cC`s||_|jjj|_|jj|_||_|j|jkrQd|_nd|j|jf|_|j |_ t |_ dS(NsInternal parallel functionsFunction %s, %s( Rtfunc_idt func_qualnameRRtlinetfusion_enabledR!tpurposet get_parforstinitial_parforstTrueR'(RRR+((s+lib/python2.7/site-packages/numba/parfor.pytsetupas   cC`s|jS(N(t _has_setup(R((s+lib/python2.7/site-packages/numba/parfor.pyR'pscC`s ||_dS(N(R1(Rtstate((s+lib/python2.7/site-packages/numba/parfor.pyR'tscC`st|jS(N(RlR-(Rtblocks((s+lib/python2.7/site-packages/numba/parfor.pyt count_parforsyscC`s*t|}|j||t|dS(N(Rt _get_parforsR(RRjt parfors_listR3((s+lib/python2.7/site-packages/numba/parfor.pyt_get_nested_parfors|s cC`sdx]|jD]O\}}x@|jD]5}t|tr#|j||j||q#q#Wq WdS(N(RRRRRR7(RR3R6tlabeltblkR((s+lib/python2.7/site-packages/numba/parfor.pyR5s  cC`s g}|j|jj||S(N(R5RR3(RR6((s+lib/python2.7/site-packages/numba/parfor.pyR-scC`sg}x|jjD]\}}|jdg}x{|D]s}t|jtjr;|jjdkrtt |j |j}|dk r|dkr|j |qqq;q;WqW|S(NthoistedtcallRR(RR( R&RRRtvalueRtExprRR8R6RRR(Rtallocstpf_idtdataRtinstR;((s+lib/python2.7/site-packages/numba/parfor.pythoisted_allocationss c C`setj|}|ikr(gtfSt}x2|jD]$}x|D]}|j|qKWq>Wt|j}||}|dkrt}nt}xttt|jj |dD]T}|j |} | dk r| ||||D]2}||gkr"|j|j||q"q"W|S(sf returns a list of nodes reachable in an adjacency list from a specified root (textendtreachable_nodes(RtadjRRtfusersRU((s+lib/python2.7/site-packages/numba/parfor.pyR[s  cC`s<||d}|jd}td|jjd}|jjj}|j|j\}}|j|j\} } || g} t |t r8|ddkr8|dd} | j|krtd| jdSg} xb| D]Z}|rx+||D]}| j |j ||qW| r6tdt | dSqqWxT|jjD]C}x:|jD]/}|jj|krZtd|jjdSqZWqJWx|jjjD]}yp|jj|}xWt|dddD]?}|j|}t |tstd|jjd}PqqWWqtk r-qXqWq8n|S(sX pd_id - the parfors id parfors_simple - the simple parfors map iitinternalii(RR}RR*RtfilenameROR$R#RttupleRtsort_pf_by_lineRqRRERR3RRcRR(RR?tparfors_simpletpfRR*R_RQtnrootsRPtfrootstgraphst reported_locttmpR\RUR9RtidxRt((s+lib/python2.7/site-packages/numba/parfor.pyRasD      icM `s@ jstdnjjj}jj}j|krQd}d}nd||f}d}t}t}t}t} t} t} t} t} t}t}|dPkrt}t} n t d |dQkrt} n|dkrt} t} t} n|d kr!t}t}t}n|dkr8| r8dSt d t t d t d |j t d t t d t d|rt dj t dnt }x9tjddD]}|jd}|jd}|j}t|tr|ddkr|ddkr_djtt|dd}|dd}d|df}q|ddkrxd}q|ddkrd}qdstqnd}|rt ||j||fn|||f||jx3|7D]+}j(||}d8|f|>|j<}@|@j=|?fd:}Afd;}B| r t1d<j t d|A|@t1t dn| r t1d=j t d|B|@t1t dnt dt t dt d | s< |rU t d>j d?dn| r t}Ct1d@x[j>j'D]J\}D}E|Ej.dAg}Fx)|FD]!}Gt|Gj?t@jAr y|Gj?jB}H|HdBkr dC}/|Gj}t |/||Dfytjj|j}Wn&t k r: tjj|j}nXtCjD|}|r |j,r t dD||j,dkrw dn |j,dj/nt dEt}CnWq tEtFfk r q Xq q Wq{ W|Cs t dFq n|r< t1dt1dGt}Ij>r xj>j'D]\}D}E|Ej.dAd}J|Ej.dHd}K|J rj |K rj t1dI|Dq nt1dJ|D|Jr t1dKg|JD]}Lt1d,|L^q t}In|Kr t1dLg|KD]"\}}Lt1dM|L|f^q t}Iq q Wn|Is t dNnt d?dq< nt dOj0||dS(RNsself.setup has not been calledsInternal parallel functions R^s Function %s, %s tuseriiiis1Report level unknown, should be one of 1, 2, 3, 4s t=s% Parallel Accelerator Optimizing: %s RsLooking for parallel loopsRtkeycS`s |jjS(N(RR*(R[((s+lib/python2.7/site-packages/numba/parfor.pyRURiRt.s%s %ss(internal parallel version)suser defined prangeRsinternal pndindexs5Parallel for-loop #%s: is produced from %s: %s s Found %s parallel loops.s sParallel loop listing for %ss |loop #IDs {0:{1}}| {2}t#s, t sNo source availables+--s Fusing loops sPAttempting fusion of parallel loops (combines loops with similar properties)... s+Performing sequential lowering of loops... ic`sMtfd}j|\}||dS(Nc`s^fdx?|D]7}td|f|dtdqWdS(Nc`scx\||D]P}t|dd|f||gkr |||dq q WdS(NRos%s%s %si(R^(R\RRtdepthRU(tfactnode_msgtprint_gtsword(s+lib/python2.7/site-packages/numba/parfor.pyRss#s%s%s %siR(R^(R\RMR(RqR\Rrtroot_msgRt(Rss+lib/python2.7/site-packages/numba/parfor.pyt print_graphs  (RlRO(RRuRrRvRM(RRt(RqR\RrRus+lib/python2.7/site-packages/numba/parfor.pytdump_graph_indenteds  s# Trying to fuse loops #%s and #%s:s %ss Fused loop summary: s&has the following loops fused into it:s(fused)s4Following the attempted fusion of parallel for-loopssWith fusion disabledsL {} there are {} parallel for-loop(s) (originating from loops labelled: {}).s#%ss Optimising loop nests sNAttempting loop nest rewrites (optimising for the largest parallel loops)... sis a parallel loops--> rewritten as a serial looptfusetnestc `stfd}fd}d}g}xt|jD]\}}|\}}} |dkr||kr|d|| d|}qqU|dkr||||}qUdsUtqUWdS(Nc`sgfd||gkrctd||||dtd|d}n|S(Nc`s"t|dd|dfx||D]}||gkrg}|j|ddd|df||gkr|krj||}x;|D]0}|j|ddd|dfqWnj|tdj|q.||||dq.WdS(NRos%s%s %ss (parallel)is (R^RR[tjoin(tfadj_tnadj_tnrootRpRUtmsgtfusedRt(RqRstreportedRRt(s+lib/python2.7/site-packages/numba/parfor.pyRs!s#* 1 sParallel region %s:is i(R^RZ(R{R|ttherootRt region_id(RqRRt(RsRs+lib/python2.7/site-packages/numba/parfor.pyt print_nest s  c`sg}td||j|dd|df||gkrtj||}x7|D],}|j|dd|dfqiWn|d}tdj|td|S(NsParallel region %s:Ros%s%s %ss (parallel)is (R^RRR[RzRZ(ttyR?R\RpRR~RRU(RqRRt(s+lib/python2.7/site-packages/numba/parfor.pyt print_fuse7s& -  iRxtfRy(RlRRR( tlinesRRRRR*tinfotopt_tyR?R\(RPRQRRt(Rqs+lib/python2.7/site-packages/numba/parfor.pytprint_unoptimiseds    c`sttfd}fd}d}g}xt|jD]\}}|\}}} |dkr||kr|d|| d|}qqd|dkr||||}qddsdtqdWrxtjD]\} } d} | d} | d }| d }|dkrk| d 7} t| | | ||| fq| d 7} t| | | |fqWn td dS(Nc`sfd||gkrtdtd|dfi|d6dd6dd6<|||d td d nS( Nc`s2x+||D]}|dd|df}||gkrg}||gkr|krtj||}|d7}|djg|D]}t|^q7}n|d7}j|t|dct|7)sRiRRttopological_orderingiRb(R<RRR6treplace_functions_mapRRR9R`ReRCR(t __globals__RiRRtreplace_functions_checkers_mapRRRRR0RRRgRRR/( tfunc_deftcallnamet repl_functtypstnew_functgtcheckt new_blockst_t call_tableR;RURK(RtexprRtRtlhs_typRRt work_list(s+lib/python2.7/site-packages/numba/parfor.pyt replace_func$s8         # tgetattrRps $np_g_varRtbooltbool_s $np_typ_vars$dtype_attr_vars $dtype_varN((.Rtnumba.inline_closurecallRR"RtpopRRRRtAssignttargetRRR<R=RR8RR RRRptscopeRRRtmisctModuleRtGlobalRt StringLiteraltConstt find_templateRiRMt get_call_typeRt functionsRtDTypeR;RLRtinsert(RR3R8tinstrtlhsRttypRpRRtg_np_vartg_npt g_np_assignt dtype_strttyp_varttyp_var_assigntdtype_attr_varttempttfunctdtype_attr_getattrtdtype_attr_assignt dtype_vart dtype_getattrt dtype_assign((RRRtRRRRRs+lib/python2.7/site-packages/numba/parfor.pyRsj    !$" !      #' (RgRhRRRR(((s+lib/python2.7/site-packages/numba/parfor.pyRs  cC`s4x-tjjjjD]}|j|kr|SqWdS(N(RiR t templatestbuiltin_registryRRl(Rtft((s+lib/python2.7/site-packages/numba/parfor.pyRst ParforPasscB`s eZdZedZdZdZdZdZdZ dZ dZ d Z d Z d Zd Zd ZdZdZdZdZddZdZdZdZdZdZdZdZdZdZRS(sParforPass class is responsible for converting Numpy calls in Numba intermediate representation to Parfors, which will lower into either sequential or parallel loops during lowering stage. c C`s||_||_||_||_||_||_||_|j|_|j |_ |j |_ t j |||||_ t |jjt_||_dS(N(RRRRRxRt diagnosticsR t swapped_fnsR#R$Rt ArrayAnalysisR}R3RGR t _max_labelR( RRRRRxRRRR((s+lib/python2.7/site-packages/numba/parfor.pyRs           cC`st|jj|jj|jj|jjrlt|j|j|j |j|j |j }|jn|jj r|j |jjn|jjr|j|jjn|jjr|j|jjn|jjr|j|jjn|jj|j|jjt|jdt|jjt|j|j|j t|j|j|j |jjrNt|jj|j_t|j_|jj|jjt|j|jj|jdtt|jd|j |j|jjt|j|jj|jt|jd|j |j|jjt|jdnt|j|j|j t!|jjiit|j|j|j t|jdt"j#dkrt$dt%|jj&t$d |j nt'j(|j}|j|jj)r+t*|jj+|j,|jnt-rSt.|j |j|j|j nt/|jj|jj|j0}t"j1r|jj2j3}t4|}|d kr|jjrd nd }t$d j5||||qt$dj5|ndS(sgrun parfor conversion pass: replace Numpy calls with Parfors when possible and optimize the IR.safter parfor passt up_directionsafter maximize fusion downsafter maximize fusion ups after fusionsafter optimizationisvariable types: s call types: is After fusionsWith fusion disableds0{}, function {} has {} parallel for-loop(s) #{}.sFunction {} has no Parfor.N(6R'RR3RRRtstencilRRRRRtsetitemt_convert_setitemRt_convert_numpyt reductiont_convert_reduceRt _convert_loopRR0tfusionR,tsimplify_parfor_body_CFGR1R=t _definitionsRRtmaximize_fusionRt fuse_parforstpush_call_varsR tDEBUG_ARRAY_OPTRZRRRt PostProcessort is_generatortfix_generator_typestgenerator_infoRxRtlower_parfor_sequentialtget_parfor_paramsR$RR(R)RlR(Rt stencil_passt post_procRRRR((s+lib/python2.7/site-packages/numba/parfor.pyRsr          !    c C`sPt|}g}x7|D]/}||}g}|jj|}x|jD]}t|tjr.|j} |j} |j | j rt |j | r|j || | }t|tr|\} }|j| qqt| tjr| jdkr|j|| | |}qn|j| j n|j|qKW||_qWdS(si Convert supported Numpy functions, as well as arrayexpr nodes, to parfor nodes. t arrayexprN(R-Rt get_equiv_setRRRRR<Rt _is_C_orderRR8t_is_supported_npycallt_numpy_to_parforR`RZR=Rt_arrayexpr_to_parforR( RR3t topo_ordert avail_varsR8Rtnew_bodyRRRRt pre_stmts((s+lib/python2.7/site-packages/numba/parfor.pyRs,      !c C`st|}x|D]}||}g}|jj|}x|jD]}d}t|tjr|j} |j } |j } t t |j | } | dks| dkrt |j|| | j| }n|r|}qn|j|qEW||_qWdS(sB Find reduce() calls and convert them to parfors. RRnt _functoolsN(RRn(RR5(R-RR,RRRRRRRR<R8R6Rt_reduce_to_parforReR( RR3R1R8RR3RRRjRRRR((s+lib/python2.7/site-packages/numba/parfor.pyRs*          c C`sVt|}g}x=|D]5}||}g}|jj|}x|jD]}t|tjsut|tjr4|j} |j } t|tjr|j n|j } |j } |j | j} |j | j}|j | j}t| tjjr4t|tjjrt|jtjr| j|jkrt|tjre|j|| | | | }q.t|tjjr.tt|j| j}t|tjr|jdkr|j j| jkr|j|| | | |j }qq.q1|j|}|dkr1|j|| | | | d|}q1q4n|j|qKW||_qWdS(NtgetitemRv(R-RR,RRRt StaticSetItemtSetItemRRRRR<RRR RRRptBooleanRwtNumbert_setitem_to_parforR8R<RR=Rt get_shapeRR(RR3R1R2R8RR3RRRRRR<t target_typt index_typt value_typtval_defRv((s+lib/python2.7/site-packages/numba/parfor.pyR7sH   $  $       !c8 `st\}}t}t}t||j|j|j}g|jD]#}||t||j f^q^}g} x t |ddD]\} tj dkstj dkrqnt j d} x| j D]} t| tjrt| jtjr| jjdkr j| jjj|rgj D]'} | krd| jkrd| ^qd}| jj} j| jjj|\}}xajj D]O}t|tjrt|jtjr|jjdkr|jjPqqWtijd6 j\}}|d}tfd|Dj| j}| j } j!j"j}tj#||} j$| |||_ | j| _fd|Dt%}tj#|||st=dn|3j}3|3dkr*t=dq*ntj5|t6d|t|2t?rl|2dkrlt@jA})nt@jB})tC}!tD|2|4|3g}|) jj<fdD}5tE|5|!r jFntG||||r|+n|d||f jHd|}6| j d }7t j d|7_|7jj jId|6jJjx|D]} jJ| qyWqqWqWdS(NRlcS`s|dS(Ni((ttup((s+lib/python2.7/site-packages/numba/parfor.pyRkRiiR;t pair_firstc3`s'|]\}}|kr|VqdS(N((RttRK(t loop_index(s+lib/python2.7/site-packages/numba/parfor.pys sc`si|]}||qS(((RR\(R3(s+lib/python2.7/site-packages/numba/parfor.pys s ic`sctj}xnjD]`}xW|jtjD]C}t|jtjr5|jj|kr5|j |j jq5q5WqWg}g}xjD]}x|j D]}t d|j D}||@r|j|rqntt|tj|j}tt|tjo*|jdk|j|jj|j|qqWqW||fS(sfind expressions that involve getitem using the index variable. Return both the arrays and expressions. cs`s|]}|jVqdS(N(R(RR[((s+lib/python2.7/site-packages/numba/parfor.pys sR7tstatic_getitem(R7RF(RCREt find_instsRRRR<RRRFRRRRtissubsetR9R=RR(tindicesRRAtarrstexprstlvR(Rtloop_index_vars(s+lib/python2.7/site-packages/numba/parfor.pytfind_indexed_arrayss*  c `stj|}t|oHt|tjoH|jdkoH|jdk|j}t j gj D]}|^qe}\}}t|j gt |kt|j |ktj|j}j|}x|D]} |d| _qW|S(sFind the case where size_var is defined by A[M].shape, where M is a boolean array. RRvi(R<RR9RRR=RRR<RRHtexitsRR"t _find_mask( tsize_vartsize_deftarr_varR\t live_varst index_arrst index_exprstarr_deftresultR(RNtlive_mapRR(s+lib/python2.7/site-packages/numba/parfor.pytfind_mask_from_sizes ) Rc3`s%|]}|r|ndVqdS(iN((RR[(t index_vars(s+lib/python2.7/site-packages/numba/parfor.pys ss $mask_valiis,Only known step size is supported for pranges/Only constant step size is supported for pranges4Only constant step size of 1 is supported for pranget parfor_indexc`si|]}|qS(((RRK(R(s+lib/python2.7/site-packages/numba/parfor.pys 1s RR(KR0RIRFRGtusemaptdefmaptloopsRGRlRRtentriesROR"RRRR<R=Rt_is_parallel_loopRRtheaderRet_get_loop_kindRR*RRRFRRRR,tBlockt_get_prange_init_blockRRHt intersectiontJumpRR/t has_shapeRR8R=t_mk_parfor_loopsR`Rqt_make_index_varR5R}RRR7RZtBranchRR<RtNotImplementedErrorRRR tuintptintpRRR#t_replace_loop_access_indicesRRRR(8RR3RRtcfgtusedefsR_RUt sized_loopst moved_blocksRtentryRAR\t body_labelsRet loop_kindtloop_replacingRtcpsRRRRt end_labeltbodydefstblt exit_livesRt last_insttmask_vart mask_indicesRZtunsigned_indexRXRrtmask_typt size_varst orig_indextfirst_body_blockt body_blockt index_var_typRtorig_index_vart body_labeltlabelst true_labelt false_labeltmasktmask_valRRRQt index_var_mapRjtjump(( R3RNRR[RYRRRERMRs+lib/python2.7/site-packages/numba/parfor.pyRbs   6"*'                        "               c C`s|j|jt|t|}WdQX|d}t}x|D]}||}x|jD]y} t| tjrt| j tj r||kr| j j|kr| j j|kr|j| j j|j| j jq| j j|krt dj | j jqnt| rt| } | dkrFqcntt|j| dt} | j|ks| dk r| j|krt| |nt|j| ||nt| trc|j| j||qcqcWqIW||8}dS(s Replace array access indices in a loop body with a new index. index_set has all the variables that are equivalent to loop index. Nis&Overwrite of parallel loop index at {}tlhs_only(RFRtdummy_return_in_loop_bodyR-RRRRRR<RRRRRRBRCRR8R<RR/RDt_replace_multi_dim_indRRoR( RRt index_sett new_indexRt first_labelt added_indicesR\RRRtind_def((s+lib/python2.7/site-packages/numba/parfor.pyRoFsD           ! cC`st|dk tt|j|jtjtjft|j |}tt|t j ol|j dkg|j D]!}|j|kr|n|^qz|_ dS(sq replace individual indices in multi-dimensional access variable, which is a build_tuple t build_tupleN(R9RRRRR tTupletUniTupleR<RRR=RR(Rtind_varRRt ind_def_nodeRK((s+lib/python2.7/site-packages/numba/parfor.pyRscC`s6tt|tjo!|jdk|j}|j}|j|j}|j|j}|j }tt|t j j t|t j j rt|j t jr||j kr|||j dfSt|t jr,t|j|\}}t|dko t||kd} g} d} x|D]} |j| j}t|t j j rt|j t jr| } |j } | jdq*t|t j j rt|j t jr| } |j } | jdq*t|t jr*| d7} | j| q*q*Wt| o| |dk|| | | fStdS(scheck if an array is of B[...M...], where M is a boolean array, and other indices (if available) are ints. If found, return B, M, M's type, and a tuple representing mask indices. Otherwise, raise GuardException. R7RiiN(R9RRR=RR<RRRRwR RRRpR:Rt BaseTupleR7RRlRtIntegerR:(RRWR<RR@R?RwtseqRt count_constsRR~tindR((s+lib/python2.7/site-packages/numba/parfor.pyRPsF%   "    c C`sd}d}g}xt|jD]\}}t|tjrt|jtjr|jjdkr|j|jj j |r|}nt|tjr"t|jtjr"|jjdkr"|j |jj j |r"|}q"q"W|dkr|dkrd|D} g} xut t |d|D]Z}|j|}d|jD} | | @r| | O} | j|q4|j|q4W|j| j|j| | |j|d|_n|S(s If there is init_prange, find the code between init_prange and prange calls. Remove the code from entry_block and return it. iR;cS`sh|]}|jqS((R(RRK((s+lib/python2.7/site-packages/numba/parfor.pys s icS`sh|]}|jqS((R(RRK((s+lib/python2.7/site-packages/numba/parfor.pys s (RRRRRR<R=Rt_is_prange_initRRRaRRcRRtreverse( Rt entry_blockRt prange_argst init_call_indtprange_call_indt init_bodyRtRAtarg_related_varst saved_nodest inst_vars((s+lib/python2.7/site-packages/numba/parfor.pyRes4' '       &cC`sJ||krtS||}t|dkoI|ddkpI|dtkS(NiR_(RRlR_(Rtfunc_varRR;((s+lib/python2.7/site-packages/numba/parfor.pyRs  cC`s||krtS||}t|dko|ddkp|dtkp|ddkp|dtkp|ddkp|dtkS(NiRRbR(RRlRRbR(RRRR;((s+lib/python2.7/site-packages/numba/parfor.pyRas   2 cC`s||kst||}t|dks4td}|ddksZ|dtkry,d|j|d|j|dff}Wqtk rdd f}qXn)|ddks|dtkrd }n|S( s(see if prange is user prange or internaliRjRRbR^iR(RjR(RR(RR(RRlRbRRR(RRRR;tkind((s+lib/python2.7/site-packages/numba/parfor.pyRcs  ,   cC`s>|j|}t|tjjo=|jdko=|jdkS(NtCi(RRR RRtlayoutRw(Rtarr_nameR((s+lib/python2.7/site-packages/numba/parfor.pyR-s c C`st|}|j}|dkrtj|td|}tjjtj||j |j R?(RRRRR[Rrtacc_varRt reduce_funcRtin_typRRtindex_var_typettmp_varRt reduce_f_irRRyRtfirst_reduce_labeltfirst_reduce_block((s+lib/python2.7/site-packages/numba/parfor.pyt_mk_reduction_body s>        c `sS|j}|d}|d}t|j|j}d} d} t|j|} | rk| \}} } } n|d} |j| dkr|n| }|j|||\}}| rt fd| Dn|}t j ||}|j j t j| ||t}|j|||||\}}| rS|d}n| dkrt|j}t|j}t j ||}|||Q ss $mask_vals {} functionRN(RR<RRRR8RPR=RiR`RRdRRRRRRqRGR}RRRR=R7RZRkRRR(RRRReRRRRrRWR~RRXRRRRt mask_indexRRRRRRRRRRRj((R[s+lib/python2.7/site-packages/numba/parfor.pyR6< sL    !      c C`sx|jD]\}}|j|}t}x||rt}g}d}x@|t|jdkr|j|} |j|d} t| trOt| trO|j|}|| _|| _t || | \} } |j j j | | dk rOt}|j j| jj| jg|j | |j| ||d7}qLqOn|j | t| tr~|j| |n|d7}qLW|j |jd||_q1Wq WdS(Niiii(RR,R/RRlRRRRttry_fuseRR%RRR#RRZtfuse_recursive_parfor( RRR3R8RRtfusion_happenedR3RtRt next_stmtt fused_nodet fuse_report((s+lib/python2.7/site-packages/numba/parfor.pyR r s<          cC`stt|}t|j||jtj|j|j|j|j}|j|||j ||t |dS(N( RRRRRRRRRR R(RRjRR3t arr_analysis((s+lib/python2.7/site-packages/numba/parfor.pyR s N( RgRhRRRRRRRRRoRRPReRRaRcR-RjRiR0RR<RR.RR/RRR6R R(((s+lib/python2.7/site-packages/numba/parfor.pyRs8 Z   + 9  , '     5 ]    ; ( 6 #cC`sht|j}|jddt|j|_|ttkrPg|_n|tkrt |jdkr|jjqn|t krt |jdkr|jjqn|dkr0t |jdkr|jjnt |jdkr0|jj}|jj|jj |q0n|dkrdt |jdkrd|jjqdndS( s%remove size argument from args or kwsRRiitrandintit triangularN( RRRR`RRORNReRQRlRPR(RRRtdt_arg((s+lib/python2.7/site-packages/numba/parfor.pyR s,      cC`sog}x%|jD]}|j||jqWi}x(|jD]\}}||j||Wt||fS(N(ReRRRR`(RRRRyRR((s+lib/python2.7/site-packages/numba/parfor.pyR sc !`s|j} |j} |j} g}t|tr|\}}g}xq|D]i}tj| td| }| |j<|t||||||||| | 7}|j |qMW|t j kr|dj}t |dkrs|dj}|j ||| fi}tjj||d|d| }|tjkrt|d|d|\}}qn4|j ||fi}tjj||d| }|||<|j} |j tj||| nxtjD]}t||rt|| }tj| t|| }||j<|j|}t|tjr|jdkr|jdkrtj| td| }tjjt|j ss1Don't know how to translate array expression '%r'(N(/RRRRR`RRRRRRKtsupported_array_operatorsRlRR=tbinopRttruedivt_gen_np_dividetunaryRxRRt MAP_TYPESt_find_func_varR<RRR RRRRRR;RRRpt_gen_arrayexpr_getitemRR Rtresolve_value_typeR<RlRR(!RRRRRRRRtparfor_index_tuple_vartall_parfor_indicesR2RRRtout_irRt arr_expr_argstarg_varsRyt arg_out_vartel_typ1tel_typ2tfunc_typtir_exprtTt func_var_nameRt func_var_deft g_math_vartg_matht g_math_assigntcall_typtvar_typ((Rs+lib/python2.7/site-packages/numba/parfor.pyR s        #&  0 %  &    cC`sA|j}|j}tj|td|}tjjt||j x7t|D])\}}t|tr |j r |Sq WdS(Ni(RRRR(RRtRA((s+lib/python2.7/site-packages/numba/parfor.pyR% scC`st}t}t|\}}t|}x|D]}||} xt| jD]\} } tj| j| j} | j| | _ti| d6j d} || O}t | |||| j B| _ |j | jqWW|||O}q7W|S(sfind variables used in body of parfors from outside and save them. computed as live variables at entry of first block. i(RRFR-t _find_parforsRRRdRRR^tget_parfor_params_innerRRRFR(R3toptions_fusionR#Rtpre_defsRtall_defsR1R8RRtRjt dummy_blockt before_defs((s+lib/python2.7/site-packages/numba/parfor.pyR( s      cC`st|}t|}t|}t|||j|j}t|||}t|} | dkrtj r|r~dnd} t dj | |j | |nt |||j x7t|D])\}}t|tr ||fVq q WdS(N(RRR(RRtRA((s+lib/python2.7/site-packages/numba/parfor.pyR2I scC`st|jj}g}xn|jjD]]}xT|jD]I}t|tjr;|jj |j j kr|j |j j qq;q;Wq+Wt t|t|@}t|S(shget arrays that are written to inside the parfor and need to be passed as parameters to gufunc. (R}RRGRERRRR9RRRRRR"RR(Rjt parfor_paramst last_labeltoutputsR9R((s+lib/python2.7/site-packages/numba/parfor.pytget_parfor_outputsO s!c C`s|dkri}n|dkr*g}n|dkrEtt}n|dkr`tt}n|dkrui}nt|}t|} | d} t|xt| D]} xt|j| jD]t} t | t j r| j j |ks| j j |kr| j j } | j} | |kr.| n|| }g}t | t jr_| j g}n:t | t jrg| jjD]}|j ^q}n||j|x|D]}||| sis#initisInvalid reduction formatN(RR/RRR<RRRRR=RRt get_expr_argsRlRRRRR$( RRQRLRtRRRGtin_varsR[ReR]t non_red_argst replace_dict((RTRUs+lib/python2.7/site-packages/numba/parfor.pyR@ s.   !" 1.4 '   cC`sa|jdkr|j|jgS|jdkrHg|jD] }|^q8Stdj|dS(s- Get arguments of an expression node RRMR;sget arguments for expression {}N(RRM(RRRGReRlR(RRK((s+lib/python2.7/site-packages/numba/parfor.pyRV s cC`sx|jD]}|ddkr |dd}xMtt|D]9}t||tjrAt||||||) s cS`s1h|]'}t|jtjr|jjqS((RRRRR(RR\((s+lib/python2.7/site-packages/numba/parfor.pys , s cS`s1h|]'}t|jtjr|jjqS((RRRRR(RR\((s+lib/python2.7/site-packages/numba/parfor.pys / s R]R^N(RRRRFRIR}RGRR1t dominatorsR_RERRRtget_parfor_pattern_varsRt_use_defs_result( Rjtuse_settdef_setR3tusesRTRpR<R1tdefinitely_executedRR8t loop_vars((s+lib/python2.7/site-packages/numba/parfor.pyt parfor_defs s4           c C`st|}t|}t|}t|||j|j}t||||j}d|jD}|d|jDO}|d|jDO}|d|jDO}t}x|j D]t} |j | c|M<|j | c|8<||j | O}|j | c|M<|j | c|8<||j | O}qWdt fdY} t j| |} | j|j |j t|||BS(s~insert dels in parfor. input: dead variable set right after parfor. returns the variables for which del was inserted. cS`s1h|]'}t|jtjr|jjqS((RRRRR(RR\((s+lib/python2.7/site-packages/numba/parfor.pys F s cS`s1h|]'}t|jtjr|jjqS((RRRRR(RR\((s+lib/python2.7/site-packages/numba/parfor.pys I s cS`s1h|]'}t|jtjr|jjqS((RRRRR(RR\((s+lib/python2.7/site-packages/numba/parfor.pys L s cS`sh|]}|jjqS((RR(RR\((s+lib/python2.7/site-packages/numba/parfor.pys N s t DummyFuncIRcB`seZdZRS(cS`s ||_dS(N(R3(RR3((s+lib/python2.7/site-packages/numba/parfor.pyR_ s(RgRhR(((s+lib/python2.7/site-packages/numba/parfor.pyRi] s(RRIRFRGR]R^RHRRRGR^tescapingtobjectRR#t_patch_var_delsR( Rjt curr_dead_setR3RpRqRYtdead_mapRgtdead_setR8RiR*((s+lib/python2.7/site-packages/numba/parfor.pytparfor_insert_dels: s0     c C`svt|\}}t||j||\}}x?|jD]1}t} x"| rmt|||||} qLWq=WdS(sm Reorder statements to maximize parfor fusion. Push all parfors up or down so they are adjacent. N(R0REt arg_namesRER/tmaximize_fusion_inner( RR3RRRRt alias_mapt arg_aliasesRt order_changed((s+lib/python2.7/site-packages/numba/parfor.pyRk s  c C`st}d}x|t|jdkr|j|}|j|d}|rdt|||||nt|||||} | r||j|<||j|d s cS`sh|]}|jqS((R(RRK((s+lib/python2.7/site-packages/numba/parfor.pys s i(RRRtPrintRR3R<RR8tis_assert_equivRzRtget_parfor_writesR.RlR/R( RRRRRst stmt_accessest stmt_writest next_accessest next_writes((s+lib/python2.7/site-packages/numba/parfor.pyRv s  cC`st||\}}|dkS(NRS(R6(RRRR((s+lib/python2.7/site-packages/numba/parfor.pyR| scC`st|tstt}|jj}|j|d}|j t |t|trW|j t |qWqWWqGW|S(Ni( RRRRRRCRRERRR.R}(RjtwritesR3RR((s+lib/python2.7/site-packages/numba/parfor.pyR} s  t FusionReporttfirsttsecondtmessagec`std|d|d}t|jt|jkrtdd}d}||jt|jf}||jt|jf}t|j|j|||f}d|fSt|j}fd} xt|D]} |j| } |j| } | | j| jo8| | j| jo8| | j | j std| d}|d | j| j| j f7}|d | j| j| j f7}t|j|j|| }d|fSqWt |st |rtd d }t|j|j||j|jf}d|fSt |j } t }x!| jjD]}||O}q:Wt |j }t i|jd 6jd }x!|jjD]}||O}qW|j|stdd}t|j|j||j|jf}d|fSt||S(sItry to fuse parfors and return a fused parfor, otherwise return None stry_fuse: trying to fuse s s/try_fuse: parfors number of dimensions mismatchs4- fusion failed: number of loops mismatched, %s, %s.s(parallel loop #%s has a nest of %s loopsc`s||kpj||S(N(tis_equiv(R[R](R(s+lib/python2.7/site-packages/numba/parfor.pyR ss/try_fuse: parfor dimension correlation mismatchs7- fusion failed: loop dimension mismatched in axis %s. sslice(%s, %s, %s) != sslice(%s, %s, %s)s1try_fuse: parfor cross iteration dependency foundsK- fusion failed: cross iteration dependency found between loops #%s and #%sis)try_fuse: parfor2 depends on parfor1 bodysT- fusion failed: parallel loop %s has a dependency on the body of parallel loop %s. N(tdprintRRlRRRRcRRRthas_cross_iter_depRFRRR^RERR]t isdisjointtfuse_parfors_inner(Rtparfor1tparfor2RR~RRRRRRttnest1tnest2tp1_body_usedefst p1_body_defsRTt p2_usedefstp2_usesRe((Rs+lib/python2.7/site-packages/numba/parfor.pyR sZ "           c C`s|jjj|jjt|jj}|j|j}t|jj}t|jj}|j|jj||jj|j|jj|t |j }i|j |j j 6}x5t |D]'}|j |j||j |jj $ ss1Parallel for-loop #{} is fused into for-loop #{}.sE- fusion succeeded: parallel for-loop #{} is fused into for-loop #{}.(RRRZRqRRGR}RRRlRRRRcRR#RR/RRREtremove_duplicate_definitionsRR RRZRRR( RRtparfor2_first_labeltparfor2_first_blocktparfor1_first_labeltparfor1_last_labelRt index_dictRtR3tnamesetR~R((s+lib/python2.7/site-packages/numba/parfor.pyR s6    c C`sx|jD]\}}|j}g}t}xj|D]b}t|tjr|jj}||kr||krzq8n|j|qn|j |q8W||_q WdS(ssRemove duplicated definition for variables in the given nameset, which is often a result of parfor fusion. N( RRRRRRRRRFR( R3RR8RRR3tdefinedRAR((s+lib/python2.7/site-packages/numba/parfor.pyR2 s       cC`sd|jD}x|jjD]}x|jD]}t|tjtjfrZq3nt|tjrt|j tj r|j j }|dkrq3qn|j |j s3td||tSq3Wq#WtS(NcS`sh|]}|jqS((R(RR\((s+lib/python2.7/site-packages/numba/parfor.pys K s RR7RFshas_cross_iter_dep found(RR7RF(RRRERRRR9R8RR<R=RRRRR/R(RjRIRRR((s+lib/python2.7/site-packages/numba/parfor.pyRF s'    cG`s tjdkrt|ndS(Ni(R R"RZ(R((s+lib/python2.7/site-packages/numba/parfor.pyR^ scC`st}xz|jD]o}|ddkr|dd}|dd}x:||D]+}t|tjrP|j|jqPqPWqqW|S(s: get the variables used in parfor pattern information iRi(RRRRRRFR(RjtoutRR\R]RK((s+lib/python2.7/site-packages/numba/parfor.pyRab s c"C`st|jt|j}WdQX|d}i}t|j|j|j|||t|j} x|D]} | |krqqnx|j| jD]} t| t j rt| j t j r| j j dkr| j jj|jjkrqntd| jD} | | @} x| D]}|j|dq'WqWqqWxU|D]M} | |krhqPn|j| }|j}t|j|j|||qPW|jj}t|j}t|||\}}t j|t jdd}||jj|t|}t|}t|||j|j}t|j}x8|jD]*\}}g}d|jjD}x+|j |D]\}}|||O}qWxt!|jD]} ||@}x|D]} ||| O}qWt| t j"rJ| jj|jjkrJ| j#j|krJ| j#j|krJqn|d| jDO}|j| qW|j$||_q^W|j|j||jjt%||||||t&|j'jdk}!x0|jj(D]}|!t&|jdkM}!qW|!rdS|S( s7 remove dead code inside parfor including get/sets NiR7cs`s|]}|jVqdS(N(R(RRK((s+lib/python2.7/site-packages/numba/parfor.pys st parfors_dummyicS`sh|]}|jqS((R(RRK((s+lib/python2.7/site-packages/numba/parfor.pys s cS`sh|]}|jqS((R(RRK((s+lib/python2.7/site-packages/numba/parfor.pys s ()RRR-t_update_parfor_get_setitemsRRRRGRRRR<R=RRRRRRRCR}t_add_liveness_return_blockRgtLocRRIRFRGR]R^Rt terminatort successorsRR9RRtremove_dead_parfor_recursiveRlRRE("RjtlivesRtRsRRRRtfirst_block_saved_valuest saved_arrsR\Rtvarnamestrm_arrsRRt saved_valuesR3R<t return_labelRRRpRqRYt alias_setR8R3tin_livestout_blkt_datat alias_livesRKtis_empty((s+lib/python2.7/site-packages/numba/parfor.pytremove_dead_parforp s     '            c C`sx}|D]u}t|tjr|jj|jkr|jj|krx0|j|jjgD]}|j|dq_W|j ||jj}s iscopy propagate parfor gens:tkill_set( RR)R*RRRGR}R R"RZ( RjRR3tin_copies_parfortout_copies_parfort in_gen_copiest in_extra_killRR8R<tgens((s+lib/python2.7/site-packages/numba/parfor.pytget_copies_parforps   cC`ssxPt|jD]?\}}|ddkrdt|d|f|j|g}x"|jD]}|j||qWt||_dS(spostproc updates generator_info with live variables after transformations but generator variables have types in return_type that are updated here. N(t state_varsRR`t state_types(R&RxRtnew_state_typesRK((s+lib/python2.7/site-packages/numba/parfor.pyR%s cC`sb|dkri}n|dkr*i}nt|}t|||\}}t|||fS(N(RRR0R(RjRtreverse_call_tableR3((s+lib/python2.7/site-packages/numba/parfor.pytget_parfor_call_tables       cC`sA|dkri}nt|}tj||}t||S(N(RRR tget_tuple_tableR(Rjt tuple_tableR3((s+lib/python2.7/site-packages/numba/parfor.pytget_parfor_tuple_tables     cC`sD|dkrt}nt|}tj||}t||S(N(RRRR tget_array_accessesR(RjtaccessesR3((s+lib/python2.7/site-packages/numba/parfor.pytget_parfor_array_accesses*s     cC`sGt|}t||}|||d<|j|t||dS(Ni(RR5RR(RjRR3((s+lib/python2.7/site-packages/numba/parfor.pytparfor_add_offset_to_labels7s    cC`s)t|}tj|}t||S(N(RR R!R(RjR3t max_label((s+lib/python2.7/site-packages/numba/parfor.pytparfor_find_max_labelCs  c C`s|j}t|}g|jD]}|j^q}t|j}||j}g|D]0}tjtj dd|dt ||^qZ} ||j } | ||j ||_ ||_|j ||_| ||_ t |dS(NiRtuse_literal_type(R3RRRRqRGRRRRRRtbuild_constraintR( Rjt typeinferert save_blocksR3R\R[t first_blockRRKt index_assignstsave_first_block_body((s+lib/python2.7/site-packages/numba/parfor.pytparfor_typeinferLs   =     cC`sF|dkrtt}nt|j|ti|jd6||S(s)get variable definition table for parforsiN(RRR"R=RR(Rjt definitions((s+lib/python2.7/site-packages/numba/parfor.pytbuild_parfor_definitionsas  cc`st|j}||j}tj|tdtjdd}||jjtj |tjdddV||jj dS(sLadds dummy return to last block of parfor loop body for CFG computation s$constRiN( R}RGRRRRRRRRR(RR<RR((s+lib/python2.7/site-packages/numba/parfor.pyRns ' t ReduceInfercB`seZdZRS(cC`sU| s tt|dks%tt|dtjsAtt|dj|S(Nii(RRlRR RRLRp(RReR((s+lib/python2.7/site-packages/numba/parfor.pytgeneric~s (RgRhR(((s+lib/python2.7/site-packages/numba/parfor.pyR|scC`sVtj}tj }tj}|o%|}|}|s:|rRd}tj|ndS(sKCheck if the platform supports parallel=True and raise if it does not. s{The 'parallel' target is not currently supported on Windows operating systems when using Python 2.7, or on 32 bit hardware.N(R tIS_WIN32RtIS_PY3t IS_32BITSRtUnsupportedParforsError(tis_win32tis_py2tis_32bittuns1tuns2R~((s+lib/python2.7/site-packages/numba/parfor.pytensure_parallel_supports     (Rt __future__RRRR tpytypesR RRttextwrapRCRRt functoolsRt collectionsRRRt contextlibRRRiRR R R R RRRRRRRRtnumba.numpy_supportRtnumba.typing.templatesRRRtnumba.stencilparforRtnumba.extendingRtnumba.ir_utilsRRRRR R!R"R#R$R%R&R'R(R)R*R+R,R-R.R/R0R1R2R3R4R5R6R7R8R9R:R;R<R=R>R?R@RARBRCRDREtnumba.analysisRFRGRHRItnumba.controlflowRJt numba.typingRKRLtnumba.types.functionsRMtnumba.array_analysisRNRORPRQRRRSRTRRRt TextWrapperRRXR^RR_RaRkRbRzRRRRRRRRRRRRRRRR{RkRRRRRR=tStmtRRtarray_analysis_extensionsRRRRRRRRRRR'R"R%R(R3R2R>RR?RAR@RVR^R_tvisit_vars_extensionsRhtir_extension_usedefsRptir_extension_insert_delsR/RRrRzRvR|R}RRRRRRRaRRtremove_dead_extensionsRRRtalias_analysis_extensionsRRRRtcopy_propagate_extensionsRtapply_copy_propagate_extensionsR!RRR%Rtcall_table_extensionsRttuple_table_extensionsRtarray_accesses_extensionsRtadd_offset_to_labels_extensionsRtfind_max_label_extensionsRttypeinfer_extensionsRtbuild_defs_extensionsRRR(((s+lib/python2.7/site-packages/numba/parfor.pytsR        @(,".               -            b    ( n  J  F       <  %   4  .      G +     `             *