sh<SSKrSSKJr SSKJrJrJrJr SSKJ r SSKJ r SSK J r J r SSKrSSKrSSKJrJr SSKJr \4S jr\ "S 5"S S \ 55r\ "S 5"S S\ 55r\S:Xa\R2"5 gg)N) namedtuple)voidint32float32float64) guvectorize)cuda)skip_on_cudasim CUDATestCase)NumbaPerformanceWarning TypingError)override_configc v[[USS2SS24USS2SS24USS2SS245/SSS9S5nU$)Nz(m,n),(n,p)->(m,p)r targetc URup4URupE[U5HFn[U5H4nSX&U4'[U5HnX&U4==XU4XU4-- ss'M M6 MH gNrshaperange) ABCmnpijks w/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/cuda/tests/cudapy/test_gufunc.py matmulcore*_get_matmulcore_gufunc..matmulcorespwwwwqA1XQ$qAdGqAwa400G")rr)dtyper!s r _get_matmulcore_gufuncr%sM$uQT{E!Q$Kq!t=>% 1 1 r#z&ufunc API unsupported in the simulatorc\rSrSrSrSrSrSrSrSr Sr S r S r S r S rS rSrSrSrSrSrSrSrSrSrSrSrSrg)TestCUDAGufuncc[5nSn[R"US-S-[RS9R USS5n[R"US-S-[RS9R USS5nU"X45n[R "X45nUR [R"XV55 gNr$r%nparangerreshapematmul assertTrueallcloseselfgufunc matrix_ctrrrGolds r test_gufunc_small TestCUDAGufunc.test_gufunc_small!s') IIi!ma'rzz : B B9aCD F IIi!ma'rzz : B B9aCD F 1Lyy  A,-r#c[5nSn[R"US-S-[RS9R USS5n[R"US-S-[RS9R USS5n[ R "U5nU"X55R5n[R"X45nUR[R"Xg55 gr*) r%r0r1rr2r to_device copy_to_hostr3r4r5)r7r8r9rrdBrr:s r test_gufunc_auto_transfer(TestCUDAGufunc.test_gufunc_auto_transfer/s') IIi!ma'rzz : B B9aCD F IIi!ma'rzz : B B9aCD F^^A  1M & & (yy  A,-r#c[5nSn[R"US-S-[RS9R USS5n[R"US-S-[RS9R USS5nU"X45n[R "X45nUR [R"XV55 g)Nr+r,r-r.r/r6s r test_gufuncTestCUDAGufunc.test_gufunc?s') IIi!ma'rzz : B B9aCD F IIi!ma'rzz : B B9aCD F 1Lyy  A,-r#c[5nSn[R"US-S-[RS9R SSSS5n[R"US-S-[RS9R SSSS5nU"X45n[R "X45nUR [R"XV55 g)Ndr+r,r-r.r/r6s r test_gufunc_hidim TestCUDAGufunc.test_gufunc_hidimMs') IIi!ma'rzz : B B1b!Q O IIi!ma'rzz : B B1b!Q O 1Lyy  A,-r#c[[S9n[RR SSS5n[RR SS5n[R "X#5nU"X#5n[R RXE5 U"U[R"US55n[R RXF5 g)Nr- )rMrO) r%rr0randomrandnr3testingassert_allclosetile)r7r8XYgoldres1res2s r test_gufunc_new_axis#TestCUDAGufunc.test_gufunc_new_axisYs'g6 IIOOB1 % IIOOAq !yya| ""4.aJ/0 ""4.r#c[5nSn[R"US-S-[RS9R USS5n[R"US-S-[RS9R USS5n[ R "5n[ R"X55n[ R"XE5n[ R"SURUS9nU"XgXS9nURUS 9n UR5 [R"X45n UR[R"X55 g) NrDr+r,r-r.)rDr+r.)rr$stream)outr])r])r%r0r1rr2r r]r> device_arrayr$r? synchronizer3r4r5) r7r8r9rrr]dAr@dCrr:s r test_gufunc_stream!TestCUDAGufunc.test_gufunc_streamhs') IIi!ma'rzz : B B9aCD F IIi!ma'rzz : B B9aCD F ^^A & ^^A &   \ P B 2 OO6O *yy  A,-r#c[[[SS[SS5/SSS9S5n[R"S[RS9S-n[R "U5nU"X#S9 [R RX#5 g) N(x)->(x)r rcH[UR5H nXX'M gNrsizerrrs r copy&TestCUDAGufunc.test_copy..copy166]t#r#rMr-rOr^rrrr0r1 zeros_likerRrSr7rlrrs r test_copyTestCUDAGufunc.test_copysu d71:wqz23" $  $  IIb +a / MM!  Q ""1(r#c[[SS[SS4/SSS9S5n[R"S[RS9S-n[R"U5nU"X#S9 UR [R "X#55 g) Nrfr rcH[UR5H nXX'M grhrirks r rl9TestCUDAGufunc.test_copy_unspecified_return..copyrnr#rMr-rOro)rrr0r1rqr4r5rrs r test_copy_unspecified_return+TestCUDAGufunc.test_copy_unspecified_returnsx wqz71:./" $  $  IIb +a / MM!  Q  A)*r#c&[[[SS[SS5/SSS9S5n[R"S[RS9S-n[R "U5nU"X#S9 UR [R"X#55 g) Nrfr rcH[UR5H nXX'M grhrirks r rl*TestCUDAGufunc.test_copy_odd..copyrnr# r-rOro)rrrr0r1rqr4r5rrs r test_copy_oddTestCUDAGufunc.test_copy_oddsx d71:wqz23" $  $  IIb +a / MM!  Q  A)*r#c ^[[[SS2SS24[SS2SS245/SSS9S5n[R"S[RS9R SS5S -n[R "U5nU"X#S 9 UR[R"X#55 g) Nz(x, y)->(x, y)r rc[URS5H,n[URS5H nXU4XU4'M M. g)NrrOrr)rrxys r copy2d*TestCUDAGufunc.test_copy2d..copy2dsC1771:&qwwqz*A1gAdG+'r#r(r-r.rOro) rrrr0r1r2rqr4r5)r7rrrs r test_copy2dTestCUDAGufunc.test_copy2ds d71a4='!Q$-89%" $ & $ & IIb + 3 3Aq 9A = MM! q  A)*r#c|^[[[SS[SS5/SSS9S5m[RU4Sj5n[ R "S5RS5n[ R"U5nSnUR[U5 US "X#5 SSS5 g!,(df  g=f) Nz(n)->(n)r rcN[URS5H nXX'M grr)rbrs r gufunc_copyDTestCUDAGufunc.test_not_supported_call_from_jit..gufunc_copys$1771:&t'r#c>T"X5$rh)rrrs r cuda_jitATestCUDAGufunc.test_not_supported_call_from_jit..cuda_jitsq$ $r#rz#Untyped global name 'gufunc_copy'.*)rOrO) rrrr jitr0r1astyperqassertRaisesRegexr )r7rrrmsgrs @r test_not_supported_call_from_jit/TestCUDAGufunc.test_not_supported_call_from_jits d58U1X./ 0  0   %  % IIi ' ' 0 MM! 3  # #K 5 TN1 6 5 5s  B-- B;c [S/SSS9S5n[RRS5R S5n[RRS5R S5n[R "UR S5R S5n[S S 5 [R"S S 9nU"X#U5 URUSR[5 URS [USR55 URS[USR55 SSS5 SSS5 g!,(df  N=f!,(df  g=f)N(void(float32[:], float32[:], float32[:]) (n),(n)->(n)r rc\URSn[U5HnXX-X$'M grrardistlenrs r numba_dist_cudaMTestCUDAGufunc.test_inefficient_launch_configuration..numba_dist_cuda.''!*C3Z$+ r#rrrCUDA_LOW_OCCUPANCY_WARNINGSrOTrecordz Grid sizez low occupancy)rr0rPrandrzerosrrwarningscatch_warnings assertEqualcategoryr assertInstrmessage)r7rrrrws r %test_inefficient_launch_configuration4TestCUDAGufunc.test_inefficient_launch_configurations @A#F 4 & 4 & IINN9 % , ,Y 7 IINN9 % , ,Y 7xx #**95 :A >((5d+  10GH k3qt||+<= os1Q4<55? >s%-E!A=E?E! E E!! E/c:[S/SSSS9S5n[RRS5R S5R S 5n[RRS5R S5R S 5n[R "U5n[S S 5 [R"SS 9nU"X#U5 UR[U5S 5 SSS5 SSS5 g!,(df  N=f!,(df  g=f)NrrTr )nopythonrc\URSn[U5HnXX-X$'M grrrs r numba_dist_cuda2LTestCUDAGufunc.test_efficient_launch_configuration..numba_dist_cuda2rr#ir)ir+rrOrr) rr0rPrrr2rqrrrrr)r7rrrrrs r #test_efficient_launch_configuration2TestCUDAGufunc.test_efficient_launch_configurations @A#d6 C & C & IINN: & - -i 8 GK  IINN: & - -i 8 GK  }}Q :A >((5 t,  Q+6? >55? >s$0D %C;*D ; D D  Dc ^Sn[[[SS[SS5/SSSS9"U5 UR[5n[[[SS[SS5/SSSS9"U5 SSS5 UR S[ WR55 g!,(df  N4=f)Ncgrhrrrs r foo.TestCUDAGufunc.test_nopython_flag..foo r#rfr T)rrFznopython flag must be True)rrr assertRaises TypeErrorrr exception)r7rraisess r test_nopython_flag!TestCUDAGufunc.test_nopython_flags  T'!*gaj12Jv! ##& (  y )V gaj'!*56 % 77: <* 5s6;K;K7LM* )s ,B B,c SnUR[5n[[[SS[SS5/SSSSS9"U5 SSS5 Sn[ WR 5nURUS[U5U5 U[U5SR5RS5nUVs/sHofRS 5PM nnUR[S S /5[U55 g!,(df  N=fs snf) Ncgrhrrs r r.TestCUDAGufunc.test_invalid_flags..foorr#rfr TF)rwhat1ever2z/The following target options are not supported:,z'" rr) rrrrrrrrrstripsplitset)r7rrheadritemsrs r test_invalid_flags!TestCUDAGufunc.test_invalid_flagss   y )V gaj'!*56 %T @@C E*A&""# Zc$i$/CIJ%%'--c2*/0%Q%0 gw/0#e*=* )1s-C52D5 DcZ[[[SS[SS5/SSS9S5n[R"S[RS9=p#UR [ 5nU"X#US9 SSS5 SnUR[WR5U5 g!,(df  N6=f) Nrfr rcgrhr)inpr^s r r2TestCUDAGufunc.test_duplicated_output..foos r#rMr-roz()r rc^Sn[[U55HnX0UX-- nM X2S'gr)rr)rrrsrs r gu_reduce1TestCUDAGufunc.check_tuple_arg..gu_reduces4A3q6]qTAD[ #aDr#rO)axis)rrr0sumasarrayrR assert_equal)r7rrrrexpecteds r check_tuple_argTestCUDAGufunc.check_tuple_argsx wqz71:wqz:;]" $  $  aO66"**Q-"**Q-7a@ ,r#c.SnSnURX5 g)N)?@@@@@)?@ @@@@)rr7rrs r test_tuple_of_tuple_arg&TestCUDAGufunc.test_tuple_of_tuple_arg+s   Q"r#c~[SS5nU"SSSS9U"SSSS94nU"S S S S9U"S S SS94nURX#5 g)NPointrrzrrrrrrrrrrrr)rr)r7rrrs r test_tuple_of_namedtuple_arg+TestCUDAGufunc.test_tuple_of_namedtuple_arg2sZ7O4 SC3 ' SC3 ' ) SC3 ' SC3 ' ) Q"r#c[R"S5[R"S54n[R"S5[R"S54nURX5 g)Nrrrr)r0rrrs r test_tuple_of_array_arg&TestCUDAGufunc.test_tuple_of_array_arg:sN ZZ ( ZZ ( * ZZ ( ZZ ( * Q"r#cP[5nURURS5 g)Nr!)r%r__name__)r7r8s r test_gufunc_nameTestCUDAGufunc.test_gufunc_nameAs') ,7r#c (UR[5n[[[SS[SS5/SSS9S5nSSS5 [ WR 5nUR SU5 UR SU5 g!,(df  NH=f)Nz(m)->(m)r rcgrhr)rrs r f.TestCUDAGufunc.test_bad_return_type..fGsr#z+guvectorized functions cannot return valueszspecifies int32 return type)rrrrrrr)r7ter rs r test_bad_return_type#TestCUDAGufunc.test_bad_return_typeEs|   y )R %a%(34j P Q * ",, CSI 3S9* )s +B Bc[[SS[SS[SS4/SSS9S5n[R"S5nUR [ 5nU"U5 SSS5 [ WR5nURSU5 URSU5 URSU5 UR [ 5nU"X"X"5 SSS5 [ UR5nURSU5 URSU5 URS U5 g!,(df  N=f!,(df  Nk=f) Nz (m),(m)->(m)r rcgrhrrs r r ;TestCUDAGufunc.test_incorrect_number_of_pos_args..fP r#r.%gufunc accepts 2 positional argumentszor 3 positional argumentsGot 1 positional argument.zGot 4 positional arguments. rrr0r1rrrrrr7r arrr rs r !test_incorrect_number_of_pos_args0TestCUDAGufunc.test_incorrect_number_of_pos_argsOs uQxq5845#F 4  4 iil  y )R cF*",, =sC 137 2C8  y )R c !*",, =sC 137 3S9* )* )s D% D6% D36 ErN)r __module__ __qualname____firstlineno__r;rArErJrZrcrsrxr~rrrrrrrrrrrrrr__static_attributes__rr#r r'r'sw .. . . /.0 ) + + +!(B&,& N > 5 -###8::r#r'c2\rSrSrSrSrSrSrSrSr g) TestMultipleOutputsijc [[[SS[SS[SS5/SSS9S5n[R"S[RS9S-n[R "U5n[R "U5nU"X#U5 [R RX#5 [R RX$5 g)N (x)->(x),(x)r rcT[UR5HnXX'XX#'M grhrirrrrs r rlKTestMultipleOutputs.test_multiple_outputs_same_type_passed_in..copyms(166]tt#r#rMr-rOrp)r7rlrrrs r )test_multiple_outputs_same_type_passed_in=TestMultipleOutputs.test_multiple_outputs_same_type_passed_inls d71:wqz71:>?#" $  $  IIb +a / MM!  MM!  Q1  ""1( ""1(r#c [[[SS[SS[SS5/SSS9S5n[R"S[RS9S-n[R "U5n[R "U5nU"X#U5 [R RX#5 [R RUS-U5 g) Nr"r rcZ[UR5HnXX'XS-X#'M gNr+rir$s r copy_and_doubleRTestMultipleOutputs.test_multiple_outputs_distinct_values..copy_and_double~,166]ttax#r#rMr-rOr+rpr7r+rrrs r %test_multiple_outputs_distinct_values9TestMultipleOutputs.test_multiple_outputs_distinct_values|s d71:wqz71:>?#" $  $ IIb +a / MM!  MM! a  ""1( ""1q5!,r#c L[[[SS[SS[SS5/SSS9S5n[R"S[RS9S-nU"U5up4[R R X#5 [R R US-U5 g) Nr"r rcZ[UR5HnXX'XS-X#'M gr*rir$s r r+LTestMultipleOutputs.test_multiple_output_allocation..copy_and_doubler-r#rMr-rOr+)rrrr0r1rRrSr.s r test_multiple_output_allocation3TestMultipleOutputs.test_multiple_output_allocations d71:wqz71:>?#" $  $ IIb +a /q! ""1( ""1q5!,r#c [[[SS[SS[SS5/SSS9S5n[R "S[RS9S-n[R "U5n[R "U[RS9nU"X#U5 [RRX#5 [RRU[R"S5-U5 g) Nr"r rcZ[UR5HnXX'XS-X#'M g)Nrrir$s r copy_and_multiplyJTestMultipleOutputs.test_multiple_output_dtypes..copy_and_multiplys,166]ttcz#r#rMr-rOr) rrrrr0r1rqrRrS)r7r8rrrs r test_multiple_output_dtypes/TestMultipleOutputs.test_multiple_output_dtypess d58U1Xwqz:;#" $ " $ " IIb )A - MM!  MM!2:: .!" ""1( ""1rzz##6:r#c[[SS[SS[SS[SS4/SSS9S5n[R"S5nUR [ 5nU"U5 SSS5 [ WR5nURSU5 URSU5 URSU5 UR [ 5nU"X"X"U5 SSS5 [ UR5nURSU5 URSU5 URS U5 g!,(df  N=f!,(df  Nk=f) Nz(m),(m)->(m),(m)r rcgrhr)rrrrs r r @TestMultipleOutputs.test_incorrect_number_of_pos_args..frr#r.rzor 4 positional argumentsrzGot 5 positional arguments.rrs r r5TestMultipleOutputs.test_incorrect_number_of_pos_argss uQxq58U1X>?' 8  8 iil  y )R cF*",, =sC 137 2C8  y )R c# &*",, =sC 137 3S9* )* )s D. D?. D<? E rN) rrrrr&r/r4r:rrrr#r r r js) -" -;":r#r __main__)numpyr0 collectionsrnumbarrrrrr numba.cuda.testingr r unittestrnumba.core.errorsr r numba.tests.supportrr%r'r rmainrr#r rIs"//<B/")  9:H:\H:;H:V 9:Y:,Y:;Y:x z MMOr#