\ K]c@`sjdZddlmZmZmZddlZddlmZddlZ ddl Z ddl m Z m Z ddlmZmZmZmZddlmZddlmZmZdd lmZdd l m Z dd l mZd d lmZmZm Z ddl!m"Z"ddl#m$Z$ej%dZ&e&j'Z(e(Z)ej%dZ*e*j'Z+e j,Z,e,j'Z-e j.Z/e j.Z0ide j16de j26de j36de j46Z5ddZ6dZ7dZ8dZ9ddldYZ:ddmdYZ;ej<d Z=d!Z>d"Z?d#Z@d$ZAd%ZBd&ZCd'ZDd(ZEd)ZFeGd*ZHee jIe jJe jJd+ZKe$ree jLe jJe jJeKnee jMe jJe jJd,ZMd-ZNd.ZOd/ZPd0ZQee jIe jJe jJe jJd1ZRe jSZTe jUd2eTZVed3ZWeXd4ZYd5ZZed6Z[ed7Z\ee j]j^d8Z_ed9Z`eXd:Zaednkruee j]jbd;Zcee j]jdd<Zeee j]jfd=Zgee j]jhd>Ziee j]jjd?Zkee j]jld d@Zmnee j]jndAZodBZpeepdCZqdDZreerdEZsdFZteetdGZudHZveevdIZwdJZxeexdKZydLZzeezdMZ{ee j]j|dNdOZ}dPZ~ee~dQZee j]jdRZee j]jdSdTZdUZee j]jdVZee j]jdWZdXZeedYZdZZeed[Zd\Zee j]jdd]Zee j]jdd^Zed_Zee j]jdd`Zee j]jdaZee jddbZeXdcZddZdeZedokr$ee jddgZnee jdhZdiZdjZee jdkZdS(ps. Implementation of linear algebra operations. i(tprint_functiontabsolute_importtdivisionN(tir(ttypestcgutils(t lower_builtintimpl_ret_borrowedtimpl_ret_new_reftimpl_ret_untracked(t signature(toverloadtregister_jitable(tversion(R(t numpy_supporti(t make_arrayt_empty_nd_implt array_copyi(t TypingError(tHAS_MATMUL_OPERATORii tstdtctzscC`s5tj|}|dkr1td|fn|S(Nsunsupported dtype for %s()(t _blas_kindstgettNonet TypeError(tdtypet func_nametkind((s3lib/python2.7/site-packages/numba/targets/linalg.pyt get_blas_kind3s cC`s4yddl}Wntk r/tdnXdS(Nis*scipy 0.16+ is required for linear algebra(tscipy.linalg.cython_blast ImportError(tscipy((s3lib/python2.7/site-packages/numba/targets/linalg.pyt ensure_blas:s cC`s4yddl}Wntk r/tdnXdS(Nis*scipy 0.16+ is required for linear algebra(tscipy.linalg.cython_lapackR!(R"((s3lib/python2.7/site-packages/numba/targets/linalg.pyt ensure_lapackAs cC`s%|j|||}tj||S(N(tget_constant_genericRtalloca_once_value(tcontexttbuilderttytvaltconst((s3lib/python2.7/site-packages/numba/targets/linalg.pytmake_constant_slotHst_BLAScB`s5eZdZdZedZedZRS(sM Functions to return type signatures for wrapped BLAS functions. cC`s tdS(N(R#(tself((s3lib/python2.7/site-packages/numba/targets/linalg.pyt__init__SscC`sXt|d|}tjtjtjtj|tjtj|}tjd|S(Ntunderlying_floatt numba_xxnrm2(tgetattrRtintctchartintptCPointertExternalFunction(tclsRtrtypetsig((s3lib/python2.7/site-packages/numba/targets/linalg.pyR2Vs  cC`stjtjtjtjtjtjtjtj|tj|tjtj|tjtj|tj|tj}tjd|S(Nt numba_xxgemm(RR4R5R6R7R8(R9RR;((s3lib/python2.7/site-packages/numba/targets/linalg.pyR<as       (t__name__t __module__t__doc__R0t classmethodR2R<(((s3lib/python2.7/site-packages/numba/targets/linalg.pyR.Ms  t_LAPACKcB`seZdZdZedZedZedZedZedZ edZ edZ ed Z ed Z ed Zed ZRS( sO Functions to return type signatures for wrapped LAPACK functions. cC`s tdS(N(R%(R/((s3lib/python2.7/site-packages/numba/targets/linalg.pyR0|scC`sLtjtjtjtjtj|tjtjt}tjd|S(Nt numba_xxgetrf(RR4R5R6R7t F_INT_nbtypeR8(R9RR;((s3lib/python2.7/site-packages/numba/targets/linalg.pyRBs  cC`sFtjtjtjtj|tjtjt}tjd|S(Ntnumba_ez_xxgetri(RR4R5R6R7RCR8(R9RR;((s3lib/python2.7/site-packages/numba/targets/linalg.pyRDs   c C`stjtjtjtjtjtj|tjtj|tj|tj|tjtj|tj }tjd|S(Ntnumba_ez_rgeev(RR4R5R6R7R8(R9RR;((s3lib/python2.7/site-packages/numba/targets/linalg.pyREs       c C`svtjtjtjtjtjtj|tjtj|tj|tjtj|tj }tjd|S(Ntnumba_ez_cgeev(RR4R5R6R7R8(R9RR;((s3lib/python2.7/site-packages/numba/targets/linalg.pyRFs      c C`sdt|d|}tjtjtjtjtjtj|tjtj|}tjd|S(NR1tnumba_ez_xxxevd(R3RR4R5R6R7R8(R9RtwtypeR;((s3lib/python2.7/site-packages/numba/targets/linalg.pyRGs  cC`s@tjtjtjtjtj|tj}tjd|S(Nt numba_xxpotrf(RR4R5R6R7R8(R9RR;((s3lib/python2.7/site-packages/numba/targets/linalg.pyRIs    c C`st|d|}tjtjtjtjtjtj|tjtj|tj|tjtj|tj }tjd|S(NR1tnumba_ez_gesdd(R3RR4R5R6R7R8(R9RtstypeR;((s3lib/python2.7/site-packages/numba/targets/linalg.pyRJs     cC`sLtjtjtjtjtj|tjtj|}tjd|S(Ntnumba_ez_geqrf(RR4R5R6R7R8(R9RR;((s3lib/python2.7/site-packages/numba/targets/linalg.pyRLs c C`sRtjtjtjtjtjtj|tjtj|}tjd|S(Ntnumba_ez_xxgqr(RR4R5R6R7R8(R9RR;((s3lib/python2.7/site-packages/numba/targets/linalg.pyRMs c C`st|d|}tjtjtjtjtjtj|tjtj|tjtj|tjtjtj }tjd|S(NR1tnumba_ez_gelsd(R3RR4R5R6R7tfloat64R8(R9RR:R;((s3lib/python2.7/site-packages/numba/targets/linalg.pyRNs   c C`s^tjtjtjtjtj|tjtjttj|tj}tjd|S(Nt numba_xgesv(RR4R5R6R7RCR8(R9RR;((s3lib/python2.7/site-packages/numba/targets/linalg.pyRP s    (R=R>R?R0R@RBRDRERFRGRIRJRLRMRNRP(((s3lib/python2.7/site-packages/numba/targets/linalg.pyRAvs       c c`sg}g}g}xt|j|D]\}}t|tj sS|jdkrc||} } nL|jdd} t| |} t||| |f} |j | | f|j | |j | q%Wt|j |t |fVx*|D]"\}}|j j |||qWdS(s Ensure that all array arguments are contiguous, if necessary by copying them. A new (sig, args) tuple is yielded. tCFtlayouttCN(tziptargst isinstanceRtArrayRRtcopyR Rtappendt return_typettupletnrttdecref( R(R)R;RUtnewtystnewargstcopiesR*R+tnewtytnewvaltcopysig((s3lib/python2.7/site-packages/numba/targets/linalg.pytmake_contiguouss" c`sAdfd}|j||ttjtj|fdS(s. Check whether *n* fits in a C `int`. iiic`s|krtdndS(Ns$array size too large to fit in C int(t OverflowError(tn(t_maxint(s3lib/python2.7/site-packages/numba/targets/linalg.pytimpl8s NIi(tcompile_internalR RtnoneR6(R(R)RfRh((Rgs3lib/python2.7/site-packages/numba/targets/linalg.pyt check_c_int2s cC`sR|jtj||dt+|j|}|j|jdWdQXdS(s[ Check the integer error return from one of the BLAS wrappers in _helperlib.c. tlikelys#BLAS wrapper returned with an errorN(tif_thenRt is_not_nulltFalsetget_python_apit gil_ensuret fatal_error(R(R)trestpyapi((s3lib/python2.7/site-packages/numba/targets/linalg.pytcheck_blas_return@s" cC`sR|jtj||dt+|j|}|j|jdWdQXdS(s] Check the integer error return from one of the LAPACK wrappers in _helperlib.c. Rls%LAPACK wrapper returned with an errorN(RmRRnRoRpRqRr(R(R)RsRt((s3lib/python2.7/site-packages/numba/targets/linalg.pytcheck_lapack_returnLs" c C`stjtjdttttttg}|jj|dd} t|} tj tt | } tj tt |}|j | | |||j |t|j |t|j |tf} t||| dS(sQ Call the BLAS vector * vector product function for the given arguments. i tnamet numba_xxdotN(Rt FunctionTypetIntTypetll_chartintp_tt ll_void_ptmoduletget_or_insert_functionRtConstanttordtinttcalltbitcastRu( R(R)t conjugateRRfta_datatb_datatout_datatfntytfnRtkind_valRs((s3lib/python2.7/site-packages/numba/targets/linalg.pyt call_xxdotXs  cC`stjtjdttttttttttg }|jj|dd} |j} t ||| d} t ||| d} |j dkr|\} }|d}n|\}} |d}t | }tj tt |}tj t|rt d n t d }|j| ||| ||j| t|j|t||j|t|j| t|j|tf }t|||d S( sQ Call the BLAS matrix * vector product function for the given arguments. i Rwt numba_xxgemvg?gtFiittRfN(RRyRzR{R|R}R~RRR-RRRRRRRRu(R(R)tdo_transtm_typetm_shapestm_datatv_dataRRRRtalphatbetatmRftldaRRttransRs((s3lib/python2.7/site-packages/numba/targets/linalg.pyt call_xxgemvns0       *c !`stjtjdttttttttttttttg} jj| dd} |\} }|\}}|j}t ||d}t ||d}tj tt dtj tt dfd}||||\}}}||||\}}}|| | \}}}t |}tj tt |}j | ||||| |j|t||||j|t||f} t|| d S( sQ Call the BLAS matrix * matrix product function for the given arguments. i RwR<g?gRRfc`sN|jjkrn|jdkr4|dn|dj|tfS(NRSii(RRRR}(R*tshapestdata(R)tnotranstout_typeR(s3lib/python2.7/site-packages/numba/targets/linalg.pytget_array_params N(RRyRzR{R|R}R~RRR-RRRRRRu(!R(R)tx_typetx_shapestx_dataty_typety_shapesty_dataRt out_shapesRRRRtkt_kRfRRRRttransaRtdata_attransbtldbtdata_bt_tldctdata_cRRRs((R)RRRs3lib/python2.7/site-packages/numba/targets/linalg.pyt call_xxgemms4        cC`s7d}|j||||}t|||j|S(s np.dot(matrix, matrix) cS`sL|j\}}|j\}}tj||f|j}tj|||S(N(tshapetnptemptyRtdot(tatbRRRRftout((s3lib/python2.7/site-packages/numba/targets/linalg.pytdot_impls(RiRRZ(R(R)R;RURRs((s3lib/python2.7/site-packages/numba/targets/linalg.pytdot_2_mms cC`s7d}|j||||}t|||j|S(s np.dot(vector, matrix) cS`sF|j\}|j\}}tj|f|j}tj|||S(N(RRRRR(RRRt_mRfR((s3lib/python2.7/site-packages/numba/targets/linalg.pyRs (RiRRZ(R(R)R;RURRs((s3lib/python2.7/site-packages/numba/targets/linalg.pytdot_2_vms cC`s7d}|j||||}t|||j|S(s np.dot(matrix, vector) cS`sF|j\}}|j\}tj|f|j}tj|||S(N(RRRRR(RRRRft_nR((s3lib/python2.7/site-packages/numba/targets/linalg.pyRs (RiRRZ(R(R)R;RURRs((s3lib/python2.7/site-packages/numba/targets/linalg.pytdot_2_mvs c C`s|j\}}|j}t||||d}t||||d} tj||j\} d} |j|| ttj |j|t ||| tj ||j |} t ||||| |j| j| |j| S(s< np.dot(vector, vector) np.vdot(vector, vector) iicS`s7|j\}|j\}||kr3tdndS(Ns;incompatible array sizes for np.dot(a, b) (vector * vector)(Rt ValueError(RRRRf((s3lib/python2.7/site-packages/numba/targets/linalg.pyt check_argss   (RURZRRt unpack_tupleRRiR RRjRkt alloca_oncetget_value_typeRRtload( R(R)R;RURtatytbtyRRRRfRR((s3lib/python2.7/site-packages/numba/targets/linalg.pytdot_2_vvs   %c C`stt||||\}}g|jd D]}|j^q3}|ddgkrmt||||S|ddgkrt||||S|ddgkrt||||S|ddgkrt||||SdstWdQXdS(s np.dot(a, b) a @ b iiiN( R#RdRUtndimRRRRtAssertionError(R(R)R;RUtxtndims((s3lib/python2.7/site-packages/numba/targets/linalg.pytdot_2 s#c C`sHtt||||&\}}t||||dtSWdQXdS(s np.vdot(a, b) RN(R#RdRtTrue(R(R)R;RU((s3lib/python2.7/site-packages/numba/targets/linalg.pytvdot#scC`s[|j\}|j\}}||kr6tdn|j|fkrWtdndS(Ns;incompatible array sizes for np.dot(a, b) (vector * matrix)sFincompatible output array size for np.dot(a, b, out) (vector * matrix)(RR(RRRRRRf((s3lib/python2.7/site-packages/numba/targets/linalg.pytdot_3_vm_check_args.s   cC`s[|j\}}|j\}||kr6tdn|j|fkrWtdndS(Ns;incompatible array sizes for np.dot(a, b) (matrix * vector)sFincompatible output array size for np.dot(a, b, out) (matrix * vector)(RR(RRRRRRf((s3lib/python2.7/site-packages/numba/targets/linalg.pytdot_3_mv_check_args9s   c C`s|j\}}}||jks't|j}t||||d}t||||d} t||||d} tj||j} tj|| j} tj|| j} |j|jkr |}| }|j dk}| j |j }}t }n4|}| }|j dk}|j | j }}t }|j ||ttj|j|x|D]}t|||qlWt|||||||| j t|||j| jS(sE np.dot(vector, matrix, out) np.dot(matrix, vector, out) iiiRRS(RURZRRRRRRRRRRRRRiR RRjRkRRt _getvalue(R(R)R;RUtxtytytytouttyRRtyRRRRtmtyRRRRRR+((s3lib/python2.7/site-packages/numba/targets/linalg.pytdot_3_vmCs8     c !C`s|j\}}}||jks't|j}t||||d}t||||d} t||||d} tj||j} tj|| j} tj|| j} | \}}| \}}|jdkstd}|j ||t t j |j|t |||t |||t ||||j}| j}| j}tjtd}|jd||}|jd||}|j|@\}}||j|w\}}|$t||t|||||WdQX|6|j|jk}t||||| |||WdQXWdQXWdQX||j|\}}|6|j|jk}t||||| |||WdQX|-t|||| ||| ||| | WdQXWdQXWdQXWdQXt|||j| jS(s% np.dot(matrix, matrix, out) iiiRScS`sa|j\}}|j\}}||kr9tdn|j||fkr]tdndS(Ns;incompatible array sizes for np.dot(a, b) (matrix * matrix)sFincompatible output array size for np.dot(a, b, out) (matrix * matrix)(RR(RRRRRRRf((s3lib/python2.7/site-packages/numba/targets/linalg.pyRs  s==N(RURZRRRRRRRRRiR RRjRkRRRR|t icmp_signedtif_elseRRoRRRR( R(R)R;RURRRRRRRRRRRRRRfRRRRtonet is_left_vect is_right_vectr_vectr_mattv_vtm_vRtv_mtm_m((s3lib/python2.7/site-packages/numba/targets/linalg.pytdot_3_mmrs^        %    %c C`stt||||\}}td|jd D}|tdgkrjt||||S|tddgkrt||||SdstWdQXdS(s np.dot(a, b, out) cs`s|]}|jVqdS(N(R(t.0R((s3lib/python2.7/site-packages/numba/targets/linalg.pys siiiN(R#RdtsetRURRR(R(R)R;RUR((s3lib/python2.7/site-packages/numba/targets/linalg.pytdot_3stnumba_fatal_errorcC`sHxAtj|D]0}tj|jstjjdqqWdS(Ns$Array must not contain infs or NaNs.(Rtnditertisfinitetitemtlinalgt LinAlgError(Rtv((s3lib/python2.7/site-packages/numba/targets/linalg.pyt_check_finite_matrixs cC`s|r dnd}||f}t|tjr<|j}nt|tjsmd|}t|dtn|jdksd|}t|dtnt|jtj tj fsd|}t|dtndS(Ns np.linalgRs&%s.%s() only supported for array typest highlightingis%%s.%s() only supported on 2-D arrays.s3%s.%s() only supported on float and complex arrays.( RVRtOptionalttypeRWRRoRRtFloattComplex(RRt la_prefixtprefixtinterptmsg((s3lib/python2.7/site-packages/numba/targets/linalg.pyt_check_linalg_matrixs    cG`sT|dj}x@|dD]4}|j|krd|}t|dtqqWdS(NiisAnp.linalg.%s() only supports inputs that have homogeneous dtypes.R(RRRo(RRtt0RR((s3lib/python2.7/site-packages/numba/targets/linalg.pyt_check_homogeneous_typess   cC`sV|dkrR|dkr.tds.tn|dkrRtjjdqRndS(Nis(Matrix is singular to machine precision.(tfatal_error_funcRRRR(tr((s3lib/python2.7/site-packages/numba/targets/linalg.pyt_inv_err_handlers    cC`s|dS(sFpass a list of variables to be preserved through dead code eliminationi((R((s3lib/python2.7/site-packages/numba/targets/linalg.pyt_dummy_liveness_funcsc`stt|dtj|jtj|jtt|jd|jdkfd}|S(NtinvRc`s|jd}|jd|kr;d}tjj|nt|r]tj|}ntj|}|dkr||Stj|dt}|||j ||j }t |||j ||j }t |t |j |j g|S(Niis.Last 2 dimensions of the array must be square.iR( RRRRRRXtasfortranarrayRt F_INT_nptypetctypesRRtsize(RRfRtacpytipivR(tF_layoutRRBt numba_xxgetri(s3lib/python2.7/site-packages/numba/targets/linalg.pytinv_impls"   !  ( R%RRARBRRDRRRR(RR((RRRBRs3lib/python2.7/site-packages/numba/targets/linalg.pyR s cC`sP|dkrL|dkr.tds.tn|dkrLtdqLndS(Nis&Internal algorithm failed to converge.(RRR(R((s3lib/python2.7/site-packages/numba/targets/linalg.pyt%_handle_err_maybe_convergence_problem6s    cC`s|r dnd}||f}t|tjsCtd|n|jdksetd|nt|jtjtjfstd|ndS(Ns np.linalgRs'%s.%s() only supported for array types is+%s.%s() only supported on 1 and 2-D arrays s3%s.%s() only supported on float and complex arrays.(RVRRWRRRRR(RRRRR((s3lib/python2.7/site-packages/numba/targets/linalg.pyt_check_linalg_1_or_2d_matrix@s   c`srtt|dtj|jtt|jdtdtd}fd}|S(NtcholeskytUtLc`s|jd}|jd|kr;d}tjj|n|j}|dkrW|S||j|}|dkr|dkrtdstn|dkrtjjdqnx*t|D]}d|d||f,c`stt|d|jdkt|jd|j}tj|tj|jt t |jdt dt dfd}|S(NteighRR1RR c`s|jd}|jd|kr;d}tjj|nt|r]tj|}ntj|}tj|d}|dkr||fS||j||j}t |t |j |j g||fS(Niis.Last 2 dimensions of the array must be square.Ri( RRRRRRXRRRR RR(RRfRRR!R(RtJOBZtUPLORRGtw_dtype(s3lib/python2.7/site-packages/numba/targets/linalg.pyt eigh_impls*      ( R%RRRR3Rt np_supporttas_dtypeRARGRR(Rtw_typeR-((RR*R+RRGR,s3lib/python2.7/site-packages/numba/targets/linalg.pyR-ys   !c`stt|d|jdkt|jd|j}tj|tj|jt t |jdt dt dfd}|S(NteigvalshRR1RR c`s|jd}|jd|kr;d}tjj|nt|r]tj|}ntj|}tj|d}|dkr|S||j||j}t |t |j |j g|S(Niis.Last 2 dimensions of the array must be square.Ri( RRRRRRXRRRR RR(RRfRRR!R(RR*R+RRGR,(s3lib/python2.7/site-packages/numba/targets/linalg.pyt eigvalsh_impls*     ( R%RRRR3RR.R/RARGRR(RR0R2((RR*R+RRGR,s3lib/python2.7/site-packages/numba/targets/linalg.pyR2s   !c`stt|d|jdkt|jd|j}tj|tj|jt t |jdt dt ddfd}|S(NtsvdRR1tAtSic `s|jd}|jd}|dks2|dkrGtjjdnt|ritj|}ntj|}|}t||}|r}|}|} n}|}|} tj||fd|j } tj|d} tj|| fd|j } ||||j || j | j || j | } t | t |j | j | j | j g| j| | jfS(NiiisArrays cannot be emptyR(RRRRRRXRtminRRRR RRR(Rt full_matricesRfRRtldutminmnR*tucoltldvttuRtvtR(RtJOBZ_AtJOBZ_SRRJts_dtype(s3lib/python2.7/site-packages/numba/targets/linalg.pytsvd_implsF      "( R%RRRR3RR.R/RARJRR(RR7ts_typeRA((RR>R?RRJR@s3lib/python2.7/site-packages/numba/targets/linalg.pyRAs   !1c`stt|dtj|jtj|jtt|jd|jdkfd}|S(NtqrRc `s|jd}|jd}|dks2|dkrGtjjdnt|ritj|}ntj|}|}t||}tj|d|j }|||j ||j }|dkrt dst ntj ||fd|j j}xIt|D];} x2t| dD] } || | f|| | f Returns a function to compute singular values of `a` R3RR1iRRc `s|jd}|jd}|dks2|dkrGtjjdnt||}t||}d}d}rtj|}ntj|}tj|d}|||j ||j j |j | } t | t |j j j |j g|S(s+ Computes singular values. iiisArrays cannot be emptyiR( RRRRRR6RXRRRR RR( RRfRR8R9R:R;RRR(RtJOBZ_NRt np_ret_typeRJR<R=(s3lib/python2.7/site-packages/numba/targets/linalg.pyt sv_functions8     "(ii(ii( RARJRRRR3R.R/RRRR(Rt nb_ret_typetnp_dtypeR((RRRRRJR<R=s3lib/python2.7/site-packages/numba/targets/linalg.pyt_compute_singular_values_impls !2cC`s tdS(s. Compute the L2-norm of 1D-array *a*. N(RK(R((s3lib/python2.7/site-packages/numba/targets/linalg.pyt _oneD_norm_2Ksc`smt|jd|j}tj|tj|jtt|jdfd}|S(NR1tnormc`st|}tjdd}t|jd|j}||j||j}|dkr{tds{tnt |j |j g|dS(NiRi(i( tlenRRRtstridestitemsizeRRRRR(RRfRGtjmpR(RRtxxnrm2(s3lib/python2.7/site-packages/numba/targets/linalg.pyRh[s   (R3RR.R/R.R2RR(RRRh((RRRs3lib/python2.7/site-packages/numba/targets/linalg.pyt_oneD_norm_2_implRs c `snt|jd|j}tj|}tj|j}tj|j}tt|jd}|jdkr|dt j fkrdd}n dd}|S|jdkr^|dt j fkr3|j dkrt dn0|j d krt d nt d dfd }n'tj|jjdfd }|SdsjtdS(NR1RicS`s t|S(N(R(RR((s3lib/python2.7/site-packages/numba/targets/linalg.pyROscS`st|}|dkrdS|dkr2t|S|tjkrt|d}x<td|D]+}t||}||kra|}qaqaW|S|tj krt|d}x<td|D]+}t||}||kr|}qqW|S|dkrAd}x1t|D]#}||dkr|d7}qqW|S|dkrd}x(t|D]}|t||7}q`W|Sd}x,t|D]}|t|||7}qW|d|SdS(Nigiig?(RRRRR^R(RRRfRGRR+((s3lib/python2.7/site-packages/numba/targets/linalg.pyROsD          iRScS`s|S(N((R((s3lib/python2.7/site-packages/numba/targets/linalg.pyt array_preparesRcS`s|jS(N(R(R((s3lib/python2.7/site-packages/numba/targets/linalg.pyRscS`s |jS(N(RX(R((s3lib/python2.7/site-packages/numba/targets/linalg.pyRsc`s8|j}|dkrdS|}t|j|S(Nig(RRRx(RRRfta_c(R(s3lib/python2.7/site-packages/numba/targets/linalg.pyRPs    c `s\|jd}|jd}|jdkr-dS|tjkrd}x`t|D]R}d}x.t|D] }|t|||f7}qhW||krO|}qOqOW|S|tj kr&}x`t|D]R}d}x.t|D] }|t|||f7}qW||kr|}qqW|S|dkrd}x`t|D]R}d}x.t|D] }|t|||f7}q^W||krE|}qEqEW|S|dkr}x`t|D]R}d}x.t|D] }|t|||f7}qW||kr|}qqW|S|dkr2t|dS|dkrLt|dStddS(Niiigiis Invalid norm order for matrices.(RRRRRR^RR( RRRfRt global_maxtiittmptjjt global_min(tmax_val(s3lib/python2.7/site-packages/numba/targets/linalg.pyRPsZ              i(R3RR.R/R.R2RRRRRRjRRR RtfinfoRRgR( Rtord_flagRRRRRRORP((RRs3lib/python2.7/site-packages/numba/targets/linalg.pyt_get_norm_implvs, 9 EcC`s!tt|dt||S(NR(R%R R(RR((s3lib/python2.7/site-packages/numba/targets/linalg.pyt norm_impl= s cC`s$tt|ddd}|S(NtcondcS`s|dks$|dks$|dkrt|}|dksH|dkretj|d|d}qtj|d|d}n@tjj||}tjjtjj||}||}tj|rtjS|SdS(Niiii( RRRtdivideRRRtisnanR(RtpRRtnorm_at norm_inv_a((s3lib/python2.7/site-packages/numba/targets/linalg.pyRhL s$ ! (R%RR(RRRh((s3lib/python2.7/site-packages/numba/targets/linalg.pyt cond_implF s  $cC`sEd}x8tt|D]$}|||kr<|d}qPqW|S(sJ Gets rank from singular values with cut-off at a given tolerance ii(RR(tsvRRlR((s3lib/python2.7/site-packages/numba/targets/linalg.pyt_get_rank_from_singular_valuess s  c`s9tt|ddfd}|||S(sh Computes rank for matrices and vectors. The only issue that may arise is that because numpy uses double precision lapack calls whereas numba uses type specific lapack calls, some singular values may differ and therefore counting the number of them above a tolerance may lead to different counts, and therefore rank, in some cases. t matrix_rankc`sx|dtjfkrdt|jd|j}tj|}tj|j dfd}|Sdd}|SdS(NR1c`sTt|}|jd}|jd}t||}|d|}t||S(Nii(RRRgR(RttolRRRtlR(teps_val(s3lib/python2.7/site-packages/numba/targets/linalg.pyt_2d_tol_none_impl s    cS`st|}t||S(N(RR(RRR((s3lib/python2.7/site-packages/numba/targets/linalg.pyt_2d_tol_not_none_impl s ( RRRjR3RR.R/RRteps(RRtnb_typetnp_typeRR((Rs3lib/python2.7/site-packages/numba/targets/linalg.pyt_2d_matrix_rank_impl s c`sN|j}|dkr%dd}|S|dkr>||SdsJtdS(NicS`s5x.tt|D]}||dkrdSqWdS(Ngii(RR(RRR((s3lib/python2.7/site-packages/numba/targets/linalg.pyt_1d_matrix_rank_impl sii(RRR(RRRR(R(s3lib/python2.7/site-packages/numba/targets/linalg.pyt_get_matrix_rank_impl s    (R%R (RRR((Rs3lib/python2.7/site-packages/numba/targets/linalg.pytmatrix_rank_impl s   c`set|dtj|jt|d|}t|tjsRtdnfd}|S(sL Computes matrix power. Only integer powers are supported in numpy. t matrix_powerRsExponent must be an integer.c `s|dkrVtj|jd}x+t|jdD]}d|||ftd|ndd}|S(s) Computes the trace of an array. ttraceRs!integer argument expected, got %sicS`s|j\}}|}|dkr.||}n|dkrG||}ntt||d}d}|dkrx^t|D]}|||||f7}q~Wn/x,t|D]}|||||f7}qW|S(Ni(RRgR6R(RtoffsettrowstcolsRRfRGRH((s3lib/python2.7/site-packages/numba/targets/linalg.pytmatrix_trace_impl s     (RRoRVRRRR(RRR((s3lib/python2.7/site-packages/numba/targets/linalg.pyR s  cC`s_|r dnd}||f}t|tjr[|jdks[td|dtq[ndS(Ns np.linalgRis+%s.%s() only supported on 1 and 2-D arrays R(RVRRWRRRo(RRRRR((s3lib/python2.7/site-packages/numba/targets/linalg.pyt_check_scalar_or_lt_2d_mat0 s  cC`s<t|tjs%td}|Std}|SdS(NcS`stj|fS(N(RR(R((s3lib/python2.7/site-packages/numba/targets/linalg.pytasarray< scS`s|S(N((R((s3lib/python2.7/site-packages/numba/targets/linalg.pyRA s(RVRRWR (RR((s3lib/python2.7/site-packages/numba/targets/linalg.pyt _get_as_array: s c`sit|t||dtjfkrItfd}|Stfd}|SdS(Nc`sX|}|}tj|jj|jdf|jjd|jfS(Ni(RtmultiplyRdRxR(RRRtaatbb(ta_arrtb_arr(s3lib/python2.7/site-packages/numba/targets/linalg.pyt outer_implL s  !c`s_|}|}tj|jj|jdf|jjd|jf||S(Ni(RRRdRxR(RRRRR(RR(s3lib/python2.7/site-packages/numba/targets/linalg.pyRT s   !(RRRRjR (RRRR((RRs3lib/python2.7/site-packages/numba/targets/linalg.pyt_get_outer_implG s  i c`sNt|ddtt|ddtt|||dfd}|S(NtouterRc`s|||S(N((RRR(Rh(s3lib/python2.7/site-packages/numba/targets/linalg.pyRh s(RRoRR(RRRR((Rhs3lib/python2.7/site-packages/numba/targets/linalg.pyR` s c`sKt|ddtt|ddtt||dfd}|S(NRRc`s||dS(N(R(RR(Rh(s3lib/python2.7/site-packages/numba/targets/linalg.pyRu s(RRoRR(RRR((Rhs3lib/python2.7/site-packages/numba/targets/linalg.pyRm s cC`st|tjrt|jdkr<tdj|jq|jdkr^td}|Std}|Sntd}|SdS( NRSRs_np.linalg.kron only supports 'C' or 'F' layout input arrays. Receieved an input of layout '{}'.icS`s*|jd}|jd}|j||S(Nii(RRx(Rtxntxm((s3lib/python2.7/site-packages/numba/targets/linalg.pyt nrm_shape s  cS`s|jd}|jd|S(Nii(RRx(RR((s3lib/python2.7/site-packages/numba/targets/linalg.pyR s cS`s&tjdt|}||d<|S(Nii(ii(RRR(RR((s3lib/python2.7/site-packages/numba/targets/linalg.pyR s (RSR(RVRRWRRRtformatRR (RR((s3lib/python2.7/site-packages/numba/targets/linalg.pyt_kron_normaliser_impl{ s cC`st|tj}t|tj}|rw|rw|jdksN|jdkratd}|Std}|SnE|rtd}|S|rtd}|Std}|SdS(NicS`s|S(N((RRR((s3lib/python2.7/site-packages/numba/targets/linalg.pyRG scS`s|j|jS(N(RxR(RRR((s3lib/python2.7/site-packages/numba/targets/linalg.pyRG scS`s|j|jS(N(RxR(RRR((s3lib/python2.7/site-packages/numba/targets/linalg.pyRG scS`s|j|jS(N(RxR(RRR((s3lib/python2.7/site-packages/numba/targets/linalg.pyRG scS`s|dS(Ni((RRR((s3lib/python2.7/site-packages/numba/targets/linalg.pyRG s(RVRRWRR (RRta_is_arrtb_is_arrRG((s3lib/python2.7/site-packages/numba/targets/linalg.pyt _kron_return s  c`s{t|ddtt|ddtt|t|t||t|d|fd}|S(NtkronRRc `s,|}|}|jd}|jd}|jd}|jd}||}||} tj|| fd} xt|D]} | |} x}t|D]o} | | }|| ddf}xFt|D]8}||}|| |f|| ||||fs   "              )   % 2    $"$ / M    -  ,{66FV     %  d  C $@ L $ ,AN       "