ó \ K]c@svddlmZddlmZmZddlZddlZddlZd„Z d„Z de fd„ƒYZ dS(iÿÿÿÿ(ttrace(t deprecatedtNumbaDeprecationWarningNcCsÁ|dks$|dks$|dkr(dS|dkrMt|ƒ}|jƒng}|j}x?tt|ƒddƒD]%}|j|ƒ|||d9}quW|dkr·|jƒnt|ƒS(NtFiiÿÿÿÿitC(RN(tNonetlisttreversetitemsizetrangetlentappendttuple(tdtypetshapetordertstridesRti((s/lib/python2.7/site-packages/numba/smartarray.pyt_o2s s$       cCs|dks|d|jkr(d}n(|d|jkrDd}n tdƒ‚t|||ƒ}||kr}tdƒ‚n|S(NiÿÿÿÿRiRs3strides do not correspond to contiguous data layout(RRt ValueErrorR(R RRRts2((s/lib/python2.7/site-packages/numba/smartarray.pyt_s2os    t SmartArraycBsŽeZdZd Zd!ed!d!d!dd„Zed„ƒZed„ƒZ ed„ƒZ ed„ƒZ ed„ƒZ dd „Z ed ƒd „ƒZed ƒd „ƒZdd„Zedƒd„ƒZedƒd„ƒZd„Zd„Zed„ƒZed!d!d!d!ed„ƒZed„ƒZed„ƒZed„ƒZed„ƒZd„Zd„Zd„Z d„Z!d„Z"RS("s1An array type that supports host and GPU storage.thosttgpuc Cs$d}d|}tjt|ƒddƒ||jkrKtd|ƒ‚n|dk si|rc|sit‚d|_|_t |_ |_ |j ||||t |||ƒ|ƒ|dkrÒt|_ |j} nt|_ |j} | j|_| j|_| j|_| j|_| j|_dS(s]Construct a SmartArray in the memory space defined by 'where'. Valid invocations: * SmartArray(obj=, copy=): to create a SmartArray from an existing array-like object. The 'copy' argument specifies whether to adopt or to copy it. * SmartArray(shape=, dtype=, order=) to create a new SmartArray from scratch, given the typical NumPy array attributes. (The optional 'where' argument specifies where to allocate the array initially. (Default: 'host') s]http://numba.pydata.org/numba-doc/latest/reference/deprecation.html#deprecation-of-smartarrays9 SmartArray is deprecated. For more information visit %st stacklevelis"%s" is not a valid targetRN(twarningstwarnRt_targetsRRtAssertionErrort_hostt_gputFalset _host_validt _gpu_validt _allocateRtTrueRt_shapeRt_stridesR t_dtypetndimt_ndimtsizet_size( tselftobjtcopyRR Rtwhereturltmsgtt((s/lib/python2.7/site-packages/numba/smartarray.pyt__init__/s((         cCs|jS(N(R%(R,((s/lib/python2.7/site-packages/numba/smartarray.pyRZscCs|jS(N(R&(R,((s/lib/python2.7/site-packages/numba/smartarray.pyR]scCs|jS(N(R'(R,((s/lib/python2.7/site-packages/numba/smartarray.pyR `scCs|jS(N(R)(R,((s/lib/python2.7/site-packages/numba/smartarray.pyR(cscCs|jS(N(R+(R,((s/lib/python2.7/site-packages/numba/smartarray.pyR*fscCsi||jkr"td|ƒ‚n|j|ƒ|dkrB|jS|dkrU|jStd|ƒ‚dS(s>Return the representation of 'self' in the given memory space.s"%s" is not a valid targetRRsunknown memory space "%s"N(RRt_syncRR(R,R/((s/lib/python2.7/site-packages/numba/smartarray.pytgetis   s get('host')cCs |jdƒS(NR(R5(R,((s/lib/python2.7/site-packages/numba/smartarray.pyRsss get('gpu')cCs |jdƒS(NR(R5(R,((s/lib/python2.7/site-packages/numba/smartarray.pyRuscCsÌ||jkr"td|ƒ‚n|dkru|jdƒ|jdk rÈtj|jƒdkrÈ|jdƒqÈnS|dkrÈ|jdƒ|jdk rÈtj|jƒdkrÈ|jdƒqÈndS(s@Mark the given location as changed, broadcast updates if needed.s"%s" is not a valid targetRRiN( RRt _invalidateRRtsyst getrefcountR4R(R,R/((s/lib/python2.7/site-packages/numba/smartarray.pyt mark_changedxs  '  'smark_changed('host')cCs |jdƒS(NR(R9(R,((s/lib/python2.7/site-packages/numba/smartarray.pyt host_changedˆssmark_changed('gpu')cCs |jdƒS(NR(R9(R,((s/lib/python2.7/site-packages/numba/smartarray.pyt gpu_changedŠscGs |jdƒtj|j|ŒS(NR(R4tnptarrayR(R,targs((s/lib/python2.7/site-packages/numba/smartarray.pyt __array__s cCs§|dkrY|jdkr@|jdd|j|j|jƒn|js£|jƒq£nJ|jdkr|jdd|j|j|jƒn|j s£|j ƒndS(s1Sync the data in one memory space with the other.RRN( RRR#R RRR"t _copy_to_gpuRR!t _copy_to_host(R,R/((s/lib/python2.7/site-packages/numba/smartarray.pyR4’s % % cCs%|dkrt|_n t|_dS(s,Mark the host / device array as out-of-date.RN(R R"R!(R,R/((s/lib/python2.7/site-packages/numba/smartarray.pyR6 s  cCs |rtj|ƒ}n|dkru|dk rNtj||d|ƒ|_q tj||t|||ƒƒ|_n”ddlm}|dk rÐt |tj ƒs»tj|dt ƒ}n|j |ƒ|_ n9|dkrñt||dƒ}n|j|||ƒ|_ dS(NRR.iÿÿÿÿ(t devicearrayR(R<R RR=RtemptyRtnumba.cuda.cudadrvRBt isinstancetndarrayR tfrom_array_likeRRt DeviceNDArray(R,R/R-R RRR.tda((s/lib/python2.7/site-packages/numba/smartarray.pyR#©s  '  cCs |jj|jƒt|_dS(N(Rtcopy_to_deviceRR$R"(R,((s/lib/python2.7/site-packages/numba/smartarray.pyR@ÂscCs |jj|jƒt|_dS(N(Rt copy_to_hostRR$R!(R,((s/lib/python2.7/site-packages/numba/smartarray.pyRAÇscCs*t|tjƒr"t|dtƒS|SdS(s[If `value` is an ndarray, wrap it in a SmartArray, otherwise return `value` itself.R.N(RER<RFRR (tvalue((s/lib/python2.7/site-packages/numba/smartarray.pyt _maybe_wrapÌscCsM|jdkr4|jdd|j|j|jƒn|jt|j|ƒƒS(s9Transparently forward attribute access to the host array.RN(RRR#R RRRMtgetattr(R,tname((s/lib/python2.7/site-packages/numba/smartarray.pyt __getattr__Ös%cCs |jdS(Ni(R(R,((s/lib/python2.7/site-packages/numba/smartarray.pyt__len__àtcCsAt|ƒt|ƒk rtS|j|jdƒ|jdƒkƒS(NR(ttypeR RMR5(R,tother((s/lib/python2.7/site-packages/numba/smartarray.pyt__eq__áscGs|j|jdƒj|ŒƒS(NR(RMR5t __getitem__(R,R>((s/lib/python2.7/site-packages/numba/smartarray.pyRVåscGs|j|jdƒj|ŒƒS(NR(RMR5t __setitem__(R,R>((s/lib/python2.7/site-packages/numba/smartarray.pyRWçscGs|j|jdƒj|ŒƒS(NR(RMR5tastype(R,R>((s/lib/python2.7/site-packages/numba/smartarray.pyRXés(RRN(#t__name__t __module__t__doc__RRR$R3tpropertyRRR R(R*R5RRRR9R:R;R?R4RR6R#R@RAt staticmethodRMRPRQRURVRWRX(((s/lib/python2.7/site-packages/numba/smartarray.pyR*s<*           ( t numba.tracingRt numba.errorsRRRR7tnumpyR<RRtobjectR(((s/lib/python2.7/site-packages/numba/smartarray.pyts