1@j70dZddlmZddlmZddlZdZdZdZ Gdde Z Gdd e Z Gd d e Zd Zd ZdZdZd dZedZdZejddfdZejddfdZdZdZdZd!dZdZd"dZdZdZ dZ!dS)#zz The Device Array API is not implemented in the simulator. This module provides stubs to allow tests to import correctly. )contextmanager) numpy_versionNzArray contains non-contiguous buffer and cannot be transferred as a single memory region. Please ensure contiguous buffer with numpy .ascontiguousarray()c"eZdZdZfdZxZS) FakeShapez 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) ct|tr|dkrtdtt||S)Nrztuple index out of range) isinstanceint IndexErrorsuperr __getitem__)selfk __class__s YC:\Projects\Historical\venv\Lib\site-packages\numba/cuda/simulator/cudadrv/devicearray.pyr zFakeShape.__getitem__sJ a   9!a%%788 8Y%%11!444)__name__ __module__ __qualname____doc__r __classcell__)rs@rrrsB 555555555rrcBeZdZdZdZdZdZdZdZdZ dZ d Z d S) FakeWithinKernelCUDAArrayaG Created to emulate the behavior of arrays within kernels, where either array.item or array['item'] is valid (that is, give all structured arrays `numpy.recarray`-like semantics). This behaviour does not follow the semantics of Python and NumPy with non-jitted code, and will be deprecated and removed. cHt|tsJ||jd<dS)N_item)r FakeCUDAArray__dict__r items r__init__z"FakeWithinKernelCUDAArray.__init__+s*$ .....!% grcNt|trt|S|SN)rrrrs r__wrap_if_fakez(FakeWithinKernelCUDAArray.__wrap_if_fake/s' dM * * ,T22 2Krch |t|jjvr-|t |jj|S||j|S#t $r0}t|tst||Yd}~dSd}~wwxYwr!) dirr_ary(_FakeWithinKernelCUDAArray__wrap_if_fakegetattrr ExceptionrAttributeError)r attrnamees r __getattr__z%FakeWithinKernelCUDAArray.__getattr__5s 63tz////**74:?H+M+MNNN**4:+A+A(+K+KLLL 6 6 6a00 6$X..A5 6 6 6 6 6 6 6sAA7 ,A77 B1%B,,B1c<|j||dSr!r __setitem__)r nmvals r __setattr__z%FakeWithinKernelCUDAArray.__setattr__?s  r3'''''rc\||j|Sr!)r&rr )r idxs rr z%FakeWithinKernelCUDAArray.__getitem__Bs&""4:#9#9##>#>???rc<|j||dSr!r.r r4r1s rr/z%FakeWithinKernelCUDAArray.__setitem__Es  sC(((((rc*t|jSr!)lenrr s r__len__z!FakeWithinKernelCUDAArray.__len__Hs4:rct||}d|d}|rtfd|D|d<tfd|D}||i|S)NcHt|tr |jj}|Sr!)rrrr%objs r convert_fakesz@FakeWithinKernelCUDAArray.__array_ufunc__..convert_fakesUs##899 %inJroutc3.K|]}|VdSr!).0or?s r z.]s-!@!@q--"2"2!@!@!@!@!@!@rc3.K|]}|VdSr!rB)rCar?s rrEz.^s-44!]]1%%444444r)r'gettuple)r ufuncmethodargskwargscallr@r?s @r__array_ufunc__z)FakeWithinKernelCUDAArray.__array_ufunc__Ksuf%%    jj  A!!@!@!@!@C!@!@!@@@F5M4444t44444tT$V$$$rN) rrrrrr&r,r2r r/r:rOrBrrrr"s&&& 666(((@@@)))%%%%%rrc.eZdZdZdZd'dZedZedZdZ d'dZ ed Z d(d Z d Z d Zd)dZd'dZedZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ d Z!d!Z"d"Z#d#Z$d$Z%d%Z&d'd&Z'd S)*rzj Implements the interface of a DeviceArray/DeviceRecord, but mostly just wraps a NumPy array. Trc"||_||_dSr!)r%streamr aryrRs rrzFakeCUDAArray.__init__js  rc|jjSr!r%nbytesr9s r alloc_sizezFakeCUDAArray.alloc_sizens yrc|jjSr!rVr9s rrWzFakeCUDAArray.nbytesrsyrc t|j|}|S#t$r}d|z}t||d}~wwxYw)Nz#Wrapped array has no attribute '%s')r'r%r))r r*attrr+msgs rr,zFakeCUDAArray.__getattr__wsV -49h//DK - - -7(BC %%1 , -s =8=c,t|j|Sr!rr%)r rRs rbindzFakeCUDAArray.bindsTY///rc*|Sr!) transposer9s rTzFakeCUDAArray.Ts~~rNcRttj|j|S)N)axes)rnprar%)r rds rrazFakeCUDAArray.transposes!R\$)$???@@@rc|j|}t|tjtjfvr|St ||jS)NrR)r%r typerendarrayvoidrrR)r r4rets rr zFakeCUDAArray.__getitem__sKi##C(( 99RZ1 1 1J T[999 9rc8|j||Sr!)r%r/r6s rr/zFakeCUDAArray.__setitem__sy$$S#...rc|tj|j}nt||tj||j|Sr!)re empty_liker%check_array_compatibilitycopytorSs r copy_to_hostzFakeCUDAArray.copy_to_hostsC ;- **CC %dC 0 0 0 #ty!!! rct|t|t|}}t|tr t|t ||nDt j||jdrdnddtdkrdnd}t ||t j |j |dS) a 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. C_CONTIGUOUSCFTrFN)ordersubokcopy) sentry_contiguous array_corerrrorearrayflagsrrpr%)r rTrR self_coreary_cores rcopy_to_devicezFakeCUDAArray.copy_to_devices $(.. 38 c= ) ) ; c " " " %i : : : :x&_^<Ecc#+f44UU$ @@@H &i : : : ).(+++++rc4t|jjSr!)rr%shaper9s rrzFakeCUDAArray.shapes)))rc@t|jj|i|Sr!)rr%ravelr rLrMs rrzFakeCUDAArray.ravels#_TY_d=f==>>>rc@t|jj|i|Sr!)rr%reshapers rrzFakeCUDAArray.reshapes%.TY.???@@@rc@t|jj|i|Sr!)rr%viewrs rrzFakeCUDAArray.views#^TY^Tz'FakeCUDAArray.split..s0    !     r)resplitr%ranger8)r sectionrRs rrzFakeCUDAArray.splitsC  XdiwD 7)K)KLL    rrr!)Nr)(rrrr__cuda_ndarray__rpropertyrXrWr,r_rbrar r/rqrrrrrrrrrr:rrrrrrrrrrrrrrrBrrrrbsy   X   X ---0000  X AAAA:::///,,,,,**X*???AAA>>>,,,,,, 111111000111000111000000000111000000111      rrc|jr|js|Sg}|jD],}||dkrdntd-|t |S)aG 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. rN)stridessizeappendslicerI)rT core_indexstrides rr|r|sn ;ch J+==v{{!!d <<<< uZ  !!rc|jj}tt|jt|jD]\}}|dkr|dkr||krdS||z} dS)z Returns True iff `ary` is C-style contiguous while ignoring broadcasted and 1-sized dimensions. As opposed to array_core(), it does not call require_context(), which can be quite expensive. rFT)dtypeitemsizezipreversedrr)rTrrrs r is_contiguousrso 9 DXci00(3;2G2GHH v 1991v~~uu EMD 4rct|}|jds!|jdsttdSdS)Nrs F_CONTIGUOUS)r|r~ ValueErrorerrmsg_contiguous_buffer)rTcores rr{r{sM c??D :n %3dj.H312223333rcp||}}|j|jkrtd|jd|j|j|jkrt d|jd|j|j|jkrt d|jd|jdS)Nzincompatible dtype: z vs. zincompatible shape: zincompatible strides: )squeezer TypeErrorrrr)ary1ary2ary1sqary2sqs rroro%s\\^^T\\^^FF zTZiTZZ122 2 |v|##j***djj233 3 ~''j,,, 677 7('rTc tj|tdkrdndd}t|||jjdvr tjn|j}ttjtj t| ||j|j |j  t|S|||dS) NrvFTrzryMm)bufferrrr)rhrg)rer}rr{rcharint64rrirzr|rrrrhr)rTrRrzto buffer_dtypes r to_devicer2s (3!.!7!7T   Cc z#&9>T#9#9rxxsy  Jwz#//44\BBii     dS d""     #f-----rc#KdVdSr!rB)args rpinnedrEs EEEEErcVdD]}||vr||t|i|S)N)portablewc)pop device_array)rLrM unused_args r mapped_arrayrJsB(##    JJz " " "  ( ( ((rrtc2tj||||SN)rrrrxrerirrrrxs r pinned_arrayrQ :E7%u M M MMrc2tj||||Srrrs r managed_arrayrUrrcxd|vr|dnd}ttj|i||S)NrRrrg)rrreri)rLrMrRs rrrYsD%-%7%7VZZ ! ! !QF T4V44V D D DDrcZ|jds|jds |jdkrdSdt|jD}|ddgt |jz}|jj}|D]\}}|||<||j|z}t|S) z[ Given an array, compute strides for a new contiguous array of the same shape. rsrrNcg|]}|SrBrB)rCxs rrz2_contiguous_strides_like_array..os6661666rc|dS)NrrB)rs rz0_contiguous_strides_like_array..ps !A$r)keyr) r~ndim enumeratersortr8rrrrI)rT stridepermrri_perm_s r_contiguous_strides_like_arrayr^s y CIn$=Qt76i 44666JOOO'''cC $$$G Y F$$  #)F## >>rc>|jdr|jdsdSdS)Nrrsrurt)r~)rTs r_order_like_arrayr{s) y >)Bssrcxt|}t|}t|j|j||SNr)rrrrr)rTrRrrxs rdevice_array_liker @,S11G c " "E cisy'# % % %%rcxt|}t|}t|j|j||Sr )rrrrr)rTrrxs rpinned_array_liker r rct|tr|dfSt|tjs$tj|t dkrdndd}t |||dfS)NFrvTr)rrrerjr}rr)rTrRrzs r auto_devicersx#}%%Ez c27 # #h '&00d S&$ ' ' --rc$t|ddS)z$Check if an object is a CUDA ndarrayrF)r'r=s ris_cuda_ndarrayrs 3*E 2 22rctfd}|dt|dt|dtj|dtdS)z,Verify the CUDA ndarray interface for an objct|st|tt||st|d|dS)Nz must be of type )hasattrr)rr')r[typr>s r requires_attrz4verify_cuda_ndarray_interface..requires_attrsesD!! ' && &'#t,,c22 H DDD##!FGG G H HrrrrrN)require_cuda_ndarrayrIrerr )r>rs` rverify_cuda_ndarray_interfacersHHHHH M'5!!!M)U###M'28$$$M&#rcBt|stddS)z9Raises ValueError is is_cuda_ndarray(obj) evaluates Falsezrequire an cuda ndarray objectN)rrr=s rrrs+ 3  ;9:::;;r)rTNr)rT)"r contextlibrnumba.np.numpy_supportrnumpyre DeviceRecordfrom_record_likerrIrobjectrrr|rr{rorrrfloat64rrrrrr r rrrrrBrrr!s<&%%%%%000000 3 5 5 5 5 5 5 5 5=%=%=%=%=%=%=%=%@X X X X X FX X X v"""$    333 7 7 7....&   )))!j$cNNNN "z4sNNNNEEE :%%%%%%% . . . .333    ;;;;;r