sh54SrSSKJr SSKJr SSKrSrSrSr "SS\ 5r "SS \ 5r "S S \ 5rS rS rSrSrSSjr\S5rSr\R,SS4Sjr\R,SS4SjrSrSrSrS SjrSrS!SjrSrSr Sr!g)"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,^\rSrSrSrU4SjrSrU=r$) 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) cv>[U[5(aUS:a [S5e[[U]U5$)Nrztuple index out of range) isinstanceint IndexErrorsuperr __getitem__)selfk __class__s |/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/cuda/simulator/cudadrv/devicearray.pyr FakeShape.__getitem__s4 a  !a%78 8Y1!44)__name__ __module__ __qualname____firstlineno____doc__r __static_attributes__ __classcell__)rs@rrrs 55rrcH\rSrSrSrSrSrSrSrSr Sr S r S r S r g ) FakeWithinKernelCUDAArray"a/ 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. cN[U[5(deXRS'g)N_item)r FakeCUDAArray__dict__ritems r__init__"FakeWithinKernelCUDAArray.__init__+s!$ ....!% grcF[U[5(a [U5$U$N)r r!rr#s r__wrap_if_fake(FakeWithinKernelCUDAArray.__wrap_if_fake/s dM * *,T2 2KrclU[URR5;a/UR[ URRU55$URURR U55$![ a+n[U[5(d [U5UeSnAgSnAff=fr() dirr _ary(_FakeWithinKernelCUDAArray__wrap_if_fakegetattrr Exceptionr AttributeError)rattrnamees r __getattr__%FakeWithinKernelCUDAArray.__getattr__5s 63tzz//**74::??H+MNN**4::+A+A(+KLL 6a00$X.A51 6sAA>)A>> B3!B..B3c:URRX5 gr(r __setitem__)rnmvals r __setattr__%FakeWithinKernelCUDAArray.__setattr__?s r'rcVURURRU55$r()r.r r )ridxs rr %FakeWithinKernelCUDAArray.__getitem__Bs"""4::#9#9##>??rc:URRX5 gr(r7rr>r:s rr8%FakeWithinKernelCUDAArray.__setitem__Es s(rc,[UR5$r()lenr rs r__len__!FakeWithinKernelCUDAArray.__len__Hs4::rc^[X5nSmURS5nU(a[U4SjU55US'[U4SjU55nU"U0UD6$)Nc\[U[5(aURRnU$r()r rr r-objs r convert_fakes@FakeWithinKernelCUDAArray.__array_ufunc__..convert_fakesUs"#899iinnJroutc34># UH nT"U5v M g7fr(r).0orLs r .]s!@Cq-"2"2Cc34># UH nT"U5v M g7fr(r)rParLs rrRrS^s4t!]1%%trT)r/gettuple)rufuncmethodargskwargscallrNrLs @r__array_ufunc__)FakeWithinKernelCUDAArray.__array_ufunc__Ks[u%  jj !!@C!@@F5M4t44T$V$$rrN)rrrrrr%r.r4r;r r8rFr^rrrrrr"s0& 6(@)%rrc(\rSrSrSrSrS(Sjr\S5r\S5r Sr S(Sjr \S 5r S)S jr S rS rS*SjrS(Sjr\S5rSrSrSrSrSrSrSrSrSrSrSrSrSrSr Sr!S r"S!r#S"r$S#r%S$r&S%r'S(S&jr(S'r)g )+r!bz^ Implements the interface of a DeviceArray/DeviceRecord, but mostly just wraps a NumPy array. TcXlX lgr(r-streamraryrds rr%FakeCUDAArray.__init__js   rc.URR$r(r-nbytesrEs r alloc_sizeFakeCUDAArray.alloc_sizensyyrc.URR$r(rirEs rrjFakeCUDAArray.nbytesrsyyrcz[URU5nU$![anSU-n[U5UeSnAff=f)Nz#Wrapped array has no attribute '%s')r/r-r1)rr2attrr3msgs rr4FakeCUDAArray.__getattr__wsC -499h/DK -7(BC %1 , -s :5:c.[URU5$r(r!r-)rrds rbindFakeCUDAArray.bindsTYY//rc"UR5$r() transposerEs rTFakeCUDAArray.Ts~~rNcR[[R"URUS95$)N)axes)r!nprxr-)rr|s rrxFakeCUDAArray.transposesR\\$))$?@@rcURRU5n[U5[R[R 4;aU$[ X RS9$)Nrd)r-r typer}ndarrayvoidr!rd)rr>rets rr FakeCUDAArray.__getitem__sCii##C( 9RZZ1 1J [[9 9rc8URRX5$r()r-r8rAs rr8FakeCUDAArray.__setitem__syy$$S..rcUc![R"UR5nO [X5 [R"XR5 U$r()r} empty_liker-check_array_compatibilitycopytores r copy_to_hostFakeCUDAArray.copy_to_hosts6 ;-- *C %d 0 #yy! rcf[U5 [U5[U5pC[U[5(a[U5 [ X45 OD[ R "UURS(aSOSS[S:aSOSS9n[ X45 [ R"URU5 g) z 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_corer r!rr}arrayflagsrrr-)rrfrd self_coreary_cores rcopy_to_deviceFakeCUDAArray.copy_to_devices $(. 38 c= ) ) c " %i :xx&__^>rcL[URR"U0UD65$r()r!r-reshapers rrFakeCUDAArray.reshapes!TYY..??@@rcL[URR"U0UD65$r()r!r-viewrs rrFakeCUDAArray.viewsTYY^^T,, 1101010001001 rr!cUR(aUR(dU$/nURH%nURUS:XaSO [S55 M' U[ U5$)a/ 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)stridessizeappendslicerX)rf core_indexstrides rrrsS ;;chh J++v{!d < uZ !!rcURRn[[UR5[UR 55H up#US:dM US:wdMX:wa gX-nM" g)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)rfrrrs r is_contiguousrsZ 99  DXcii0(3;;2GH  191~ MD I rc[U5nURS(d$URS(d[[5egg)Nr F_CONTIGUOUS)rr ValueErrorerrmsg_contiguous_buffer)rfcores rrrs7 c?D ::n %djj.H122/I %rcUR5UR5p2URUR:wa'[SUR<SUR<35eURUR:wa'[ SUR<SUR<35eUR UR :wa'[ SUR <SUR <35eg)Nzincompatible dtype: z vs. zincompatible shape: zincompatible strides: )squeezer TypeErrorrrr)ary1ary2ary1sqary2sqs rrr%s\\^T\\^F zzTZZTZZ12 2 ||v||#**djj23 3 ~~',, 67 7(rc [R"U[S:aSOSSS9n[U5 UcURR S;a[R O URn[[R"[R"[U55RU5URURURS9R[U5S95$URXS9 g) NrFTrrMm)bufferrrr)rr)r}rrrrcharint64r!rrrrrrrr)rfrdrto buffer_dtypes r to_devicer2s ((3!.!7T Cc z#&99>>T#9rxxsyy  JJwwz#/44\Biiii   dS d"    #-rc## Sv g7fr(r)args rpinnedrEs scZSHnX!;dM URU5 M [U0UD6$)N)portablewc)pop device_array)r[r\ unused_args r mapped_arrayrJs0(   JJz ")  ( ((rrc,[R"XXS9$N)rrrrr}rrrrrs r pinned_arrayr$Q ::E% MMrc,[R"XXS9$r!r"r#s r managed_arrayr'Ur%rcpSU;aURS5OSn[[R"U0UD6US9$)Nrdrr)rr!r}r)r[r\rds rrrYs4%-%7VZZ !QF T4V4V DDrcURS(d$URS(dURS::ag[UR5Vs/sHoPM nnUR SS9 S/[ UR5-nUR RnUHupVXCU'X@RU-nM [U5$s snf)zO Given an array, compute strides for a new contiguous array of the same shape. rrrNc US$)Nrr)xs r0_contiguous_strides_like_array..ps!A$r)keyr) rndim enumeratersortrDrrrrX)rfr+ stridepermrri_perm_s r_contiguous_strides_like_arrayr5^s yy CIIn$=Q( 46414J6OOO'cC $$G YY  F  ))F##  >7s CcVURS(aURS(dgg)Nrrrr)r)rfs r_order_like_arrayr7{s yy >)Brcn[U5n[U5n[URURUUS9$Nr#)r5r7rrr)rfrdrrs rdevice_array_liker:4,S1G c "E ciisyy'# %%rcn[U5n[U5n[URURUUS9$r9)r5r7r$rr)rfrrs rpinned_array_liker=r;rc[U[5(aUS4$[U[R5(d"[R"U[ S:aSOSSS9n[ XU5S4$)NFrTr)r r!r}rrrr)rfrdrs r auto_devicer?s^#}%%Ez c277 # #hh '&0d S$ ' --rc[USS5$)z$Check if an object is a CUDA ndarrayrF)r/rJs ris_cuda_ndarrayrAs 3*E 22rc^[T5 U4SjnU"S[5 U"S[5 U"S[R5 U"S[5 g)z,Verify the CUDA ndarray interface for an objc>[TU5(d [U5e[[TU5U5(d[U<SU<35eg)Nz must be of type )hasattrr1r r/)rptyprKs r requires_attr4verify_cuda_ndarray_interface..requires_attrsDsD!! & &'#t,c22 D#!FG G3rrrrrN)require_cuda_ndarrayrXr}rr )rKrFs` rverify_cuda_ndarray_interfacerIs?H '5!)U#'288$&#rc:[U5(d [S5eg)z9Raises ValueError is is_cuda_ndarray(obj) evaluates Falsezrequire an cuda ndarray objectN)rArrJs rrHrHs 3  9:: r)rTNr)rT)"r contextlibrnumba.np.numpy_supportrnumpyr} DeviceRecordfrom_record_likerrXrobjectrr!rrrrrrrfloat64r$r'rr5r7r:r=r?rArIrHrrrrRs&0 3 5 5=%=%@X FX v"$  3 7.&  )!jj$cN "zz4sNE :%% .3  ;r