ó \ K]c@s)dZddlmZddlmZddlZddlmZm Z m Z dZ dZ d„ZdZdefd „ƒYZd efd „ƒYZd „Zd „Zd„Zdedd„Zed„ƒZejddd„Zd„Zdd„Zded„Zd„Zd„Zd„Z dS(sz The Device Array API is not implemented in the simulator. This module provides stubs to allow tests to import correctly. iÿÿÿÿ(tcontextmanager(twarnN(tsixttypest numpy_supportcCst|dtƒS(Nt__cuda_ndarray__(tgetattrtFalse(tobj((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pytis_cuda_ndarrayss™Array contains non-contiguous buffer and cannot be transferred as a single memory region. Please ensure contiguous buffer with numpy .ascontiguousarray()t FakeShapecBseZdZd„ZRS(sº The FakeShape class is used to provide a shape which does not allow negative indexing, similar to the shape in CUDA Python. (Numpy shape arrays allow negative indexing) cCsCt|tjƒr-|dkr-tdƒ‚ntt|ƒj|ƒS(Nistuple index out of range(t isinstanceRt integer_typest IndexErrortsuperR t __getitem__(tselftk((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyR s(t__name__t __module__t__doc__R(((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyR st FakeCUDAArraycBsþeZdZeZdd„Zed„ƒZed„ƒZd„Z dd„Z ed„ƒZ dd„Z d „Zd „Zddd „Zdd „Zd „Zed„ƒZd„Zd„Zd„Zd„Zd„Zd„Zd„Zdd„ZRS(sj Implements the interface of a DeviceArray/DeviceRecord, but mostly just wraps a NumPy array. icCs4|jdkr|jdƒn||_||_dS(Nii(tndimtreshapet_arytstream(RtaryR((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyt__init__/s'cCs |jjS(N(Rtnbytes(R((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyt alloc_size3scCs |jjS(N(RR(R((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyR7scCsNyt|j|ƒ}|SWn-tk rI}tjtd|ƒ|ƒnXdS(Ns#Wrapped array has no attribute '%s'(RRtAttributeErrorRt raise_from(Rtattrnametattrte((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyt __getattr__<s  cCst|j|ƒS(N(RR(RR((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pytbindDscCs |jƒS(N(t transpose(R((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pytTGscCsttj|jd|ƒƒS(Ntaxes(RtnpR%R(RR'((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyR%KscCs;|jj|ƒ}t|tjƒr7t|d|jƒS|S(NR(RRR R(tndarrayRR(Rtidxtitem((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyRNscCs|jj||ƒS(N(Rt __setitem__(RR*tval((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyR,TscCsE|dkr!tj|jƒ}n t||ƒtj||jƒ|S(N(tNoneR(t empty_likeRtcheck_array_compatibilitytcopyto(RRR((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyt copy_to_hostWs   cCs¤t|ƒt|ƒt|ƒ}}t|tƒrLt|ƒt||ƒnAtj|d|jdrkdnddtdt ƒ}t||ƒtj |j |ƒdS(s Copy from the provided array into this array. This may be less forgiving than the CUDA Python implementation, which will copy data up to the length of the smallest of the two arrays, whereas this expects the size of the arrays to be equal. tordert C_CONTIGUOUStCtFtsuboktcopyN( tsentry_contiguoust array_coreR RR0R(tarraytflagstTrueRR1R(RRRt self_coretary_core((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pytcopy_to_device_s    cCstdƒt‚dS(Ns+to_host() is deprecated and will be removed(RtNotImplementedError(R((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pytto_hostus cCst|jjƒS(N(R Rtshape(R((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyRCyscOst|jj||ŽƒS(N(RRtravel(Rtargstkwargs((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyRD}scOst|jj||ŽƒS(N(RRR(RRERF((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyR€scCs |jjjS(N(RR<t c_contiguous(R((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pytis_c_contiguousƒscCs |jjjS(N(RR<t f_contiguous(R((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pytis_f_contiguous†scCs t|jƒS(N(tstrR(R((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyt__str__‰scCs t|jƒS(N(treprR(R((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyt__repr__ŒscCs t|jƒS(N(tlenR(R((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyt__len__scCs>gtj|jt|t|ƒ|ƒƒD]}t|ƒ^q(S(N(R(tsplitRtrangeROR(RtsectionRta((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyRQ’sN(RRRR=RRtpropertyRRR#R$R&R.R%RR,R2R@RBRCRDRRHRJRLRNRPRQ(((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyR&s.               cCsZ|js |Sg}x6|jD]+}|j|dkr;dn tdƒƒqW|t|ƒS(sG Extract the repeated core of a broadcast array. Broadcast arrays are by definition non-contiguous due to repeated dimensions, i.e., dimensions with stride 0. In order to ascertain memory contiguity and copy the underlying data from such arrays, we must create a view without the repeated dimensions. iN(tstridestappendtsliceR.ttuple(Rt core_indextstride((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyR:™s )cCs;t|ƒ}|jd r7|jd r7ttƒ‚ndS(NR4t F_CONTIGUOUS(R:R<t ValueErrorterrmsg_contiguous_buffer(Rtcore((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyR9«s cCs°|jƒ|jƒ}}|j|jkrJtd|j|jfƒ‚n|j|jkr{td|j|jfƒ‚n|j|jkr¬td|j|jfƒ‚ndS(Nsincompatible dtype: %s vs. %ssincompatible shape: %s vs. %ssincompatible strides: %s vs. %s(tsqueezetdtypet TypeErrorRCR]RV(tary1tary2tary1sqtary2sq((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyR0±sic Csšt|ƒ|dkrƒ|jjdkr1tjn|j}ttjdtjt |ƒƒj |ƒd|jd|j d|j ƒƒS|j |d|ƒdS(NtMmtbufferRaRCRVR(R9R.RatcharR(tint64RR)R8R:tviewRCRVR@(RRR8ttot buffer_dtype((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyt to_device¾s  $    ccs dVdS(N((targ((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pytpinnedÎsR5c Cs"tjd|d|d|d|ƒS(NRCRVRaR3(R(R)(RCRaRVR3((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyt pinned_arrayÓscOs=d|kr|jdƒnd}ttj||Žd|ƒS(NRi(tpopRR(R)(RERFR((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyt device_array×s!cCsttj|ƒƒS(N(RR(R/(RR((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pytdevice_array_likeÜscCs_t|tƒr|tfSt|tjƒsItj|dtdtƒ}nt|||ƒtfS(NR8R7(R RRR(tvoidR;R=Rn(RRR8((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyt auto_deviceàs  cCst|dtƒS(s$Check if an object is a CUDA ndarrayR(RR(R((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyR ìscsWtˆƒ‡fd†}|dtƒ|dtƒ|dtjƒ|dtjƒdS(s,Verify the CUDA ndarray interface for an objcsStˆ|ƒst|ƒ‚nttˆ|ƒ|ƒsOtd||fƒ‚ndS(Ns%s must be of type %s(thasattrRR R(R!ttyp(R(sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyt requires_attrõsRCRVRatsizeN(trequire_cuda_ndarrayRYR(RaRR (RRy((RsGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pytverify_cuda_ndarray_interfaceñs    cCst|ƒstdƒ‚ndS(s9Raises ValueError is is_cuda_ndarray(obj) evaluates Falsesrequire an cuda ndarray objectN(R R](R((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyR{s (!Rt contextlibRtwarningsRtnumpyR(tnumbaRRRR.t DeviceRecordtfrom_record_likeR R^RYR tobjectRR:R9R0R=RnRptfloatRqRsRtRvR|R{(((sGlib/python2.7/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyts,   s