<>iZ"lSSKrSSKrSSKr/SQr\R"S5r \R"S5r SSK J r SrSS/rSrSr\R R%S\"\"\55\S 9S 5rSS jr\R R%S\"\"\55\S 9\R R%S \\S 9\R R%S /SQ5S555r\R R%S\"\"\55\S 9\R R%S \\S 9S55r\R R%S\"\"\55\S 9\R R%S \\S 9S55r\R R%S\"\"\55\S 9\R R%S \\S 9S55r\R R%S\"\"\55\S 9\R R%S \\S 9S55rg!\a \R RrGN$f=f)N)uint8int16float32z pyarrow.cudaz numba.cuda) DeviceNDArrayc[RRS5 [R 5nUR 5n[ R5n[RRU5nX4XC4/Ul g)Ni) nprandomseedcudaContextto_numbanb_cudacurrent_context from_numbacontext_choices)modulectx1nb_ctx1nb_ctx2ctx2s d/var/www/html/land-tabula/venv/lib/python3.13/site-packages/pyarrow/tests/test_cuda_numba_interop.py setup_moduler$s[IINN4 <<>DmmoG%%'G << " "7 +D#o?FcU?gN)r)rs rteardown_moduler-src)idsc[UupURURR:XdeURUR5RR:Xde[R R U5nURUR:XdeSnURU5nURURR:Xdeg)N ) rhandlevaluer r r r new_buffercontext)rctxnb_ctxrsizebufs r test_contextr)1s"!$KC ::,, ,, , ::..44 44 4 << " "6 *D :: $$ $ D .. C ::++ ++ +rc[R"U5nUS:XarUS:de[R"XR-5n[R "XBS9n[R RSSU[RS9USS&XT4$US:XaH[USUS9upTURXR-5nURUSURS 9 XV4$[S 5e) z5Return a host or device buffer with random data. hostrdtype)lowhighr'r-Ndevice)targetr-)positionnbyteszinvalid target value)rr-paallocate_bufferitemsize frombufferr randintrmake_random_bufferr#copy_from_hostr' ValueError)r'r2r-r%r(arrdbufs rr:r:>s HHUOE qyy  nn!45mmC-""qs)+#3Ax 8 %d6G~~d^^34 C!CHH=y + ,,rr-r')ric[Uup4[USXS9upV[RU5nUR U5nUR UR :Xde[ R"UR5US9n [ RRXY5 US:Ga}[US-SS5[US-US-*S54H[n UR Xz5n[ R"UR5US9n [ RRXZU 5 M] [R"[SS9 UR USSS25 SSS5 US-n X,-n X-U:XdeUR URX55nUR UR :Xde[ R"UR5US9n [ RRXY5 [R"[SS9 UR URX5SS2SSS245 SSS5 Sn US-n X,U --nX-U-U:XdeUR URXU55nUR UR :Xde[ R"UR5US9n [ RRXY5 [R"[SS9 UR URXU5SSS25 SSS5 "S S 5nUR U"U55nUR UR :Xde[ R"UR5US9n [ RRXY5 g!,(df  GNO=f!,(df  GN=f!,(df  N=f) Nr1r2r-r%r,r@zarray data is non-contiguous)matchc*\rSrSrSr\S5rSrg)test_from_object..MyObjcXlgrdarr)selfrKs r__init__(test_from_object..MyObj.__init__sIrc.URR$r)rK__cuda_array_interface__)rLs rrP8test_from_object..MyObj.__cuda_array_interface__s9955 5rrJN)__name__ __module__ __qualname____firstlineno__rMpropertyrP__static_attributes__rrMyObjrGs   6  6rrY)rr:r to_devicebuffer_from_objectr'rr8 copy_to_hosttesting assert_equalslicepytestraisesr<reshape)rr-r'r%r&r=cbufrKcbuf2arr2srdarrs1s2s3rYs rtest_from_objectrkQs8 "!$KC"4OIC   S !D  " "4 (E :: "" " ==++-U ;DJJC& qyat,a47T24A**473E==!3!3!5UCD JJ # #CFD 1 4 ]]:"@C  " "4!9 -C 1W Xw$&&t||B';<zzTYY&&&}}U//1? * ]]:"@C  " "4<<#73Q3#? @C  1W rE]w|t###&&t||BB'?@zzTYY&&&}}U//1? * ]]:"@C  " "4<<#;CaC#@ AC 66  " "5; /E :: "" " ==++-U ;DJJC&YCCCCCCs$N! +N31(O! N03 O Och[Uup#[R"U5nSnURXAR-5n[ U4UR4XS9nSUSS&SUSS&[R RUR5SSS5 [R RUR5SSS5 [RRU5n[R"UR5US9n[R RXR55 g)Nr gpu_datacXr,) rrr-memallocr7rr]r^r\r CudaBufferrr8) rr-r%r&r'memrKrcres rtest_numba_memallocrus"!$KC HHUOE D //$/ 0C $5>>"3U IDD!HDHJJD--/3R8JJD--/3R8 ?? % %c *D ==**,E :DJJD"3"3"56rc[Uup#Sn[USXS9upVUR5n[URUR UR US9n[RRUR5U5 g)Nr r1rBrm) rr:r rshapestridesr-rr]r^r\) rr-r%r&r'r=rcrtrKs rtest_pyarrow_memallocrysk"!$KC D"4OIC --/C CKKS IDJJD--/5rc^[Uup#Sn[RS [USXS9upVURR UR R :XdeUR5n[URURURUS9n[RRUR5U5 SUS'URR!5 [R""UR5US9n U SS:XdeSSS5 g!,(df  g=f)Nr rr1rBrmror,)rrgpusr:r$r!r"r rrwrxr-rr]r^r\ synchronizer8) rr-r%r&r'r=rcrtrKres rtest_numba_contextr}s"!$KC D a&tH-2= ||""fmm&9&9999mmoSYY SYYM  1 1 3S9Q   "}}T..0>Aw"}} s C3D D,c[Uup#[RS5nSn[USXS9upgSnURUS- -U-n UR 5n [ URURURU S9n XIU4"U 5 URR5 [R"UR5URS9n [RR!XS-5 g) Ncf[RS5nXR:a X==S- ss'gg)Nr?)rgridr')an_arrayposs rincrement_by_one*test_pyarrow_jit..increment_by_ones*ll1o   MQ M rr r1rB r?rmr,)rrjitr:r'r rrwrxr-r$r|rr8r\r]r^) rr-r%r&rr'r=rcthreadsperblock blockspergridrtrKarr1s rtest_pyarrow_jitrs"!$KC [[ D"4OICOXX1!45/IM --/C CKKS IDO34T:LL ==**,CII >DJJD'*r)r+rN)r`pyarrowr5numpyr ImportErrormark pytestmarkdtypes importorskipr rnumba.cuda.cudadrv.devicearrayrrcontext_choice_idsrr parametrizerangelenr)r:rkruryr}rrXrrrs$# ' >*   l +8$l3@eC(:$;</1,1,-&eC(:$;</1&f51H'261H'VeC(:$;</1&f57617(eC(:$;</1&f56616eC(:$;</1&f5 61  eC(:$;</1&f5+61+K#""J#sHH32H3