:iAtSSKrSSKrSSKrSSKrSSKrSSKrSSKJr SSK J r SSK J r J r JrJr SSKJrJr SSKJr SSKJr SSKJr SSKJr SS KJr SS KJrJ r J!r!J"r"J#r#J$r$J%r% SS K&J'r' SSK(r(\(RR"\RRUS 5(+S 5r+\(RR"\RX"5S;S5r-"SS\.5r/"SS\(R@5r0"SS\(R@5r1"SS\(R@5r2"SS\ 5r3"SS\(R@5r4\$"SS\\ 55r5"SS\ 5r6\7S :Xa\(Rp"5 gg)!Nnjit)types)rtsysnrtopt _nrt_pythonnrt) intrinsic include_path) signature)impl_ret_untracked)ir NRT_get_api)EnableNRTStatsMixinTestCasetemp_directoryimport_dynamic skip_if_32bitskip_unless_cffirun_in_subprocess) cpu_targetlinuxzlinux only test)i386x86_64z x86 only testc$\rSrSrSrSrSrSrg)Dummy%rc@[U5=RS- slgNtypealiveselfs W/srv/projetos/modelo_ml_acdoc/venv/lib/python3.13/site-packages/numba/tests/test_nrt.py__init__Dummy.__init__( T Ac@[U5=RS-slgr r"r%s r'__del__ Dummy.__del__+r*r+N)__name__ __module__ __qualname____firstlineno__r$r(r-__static_attributes__r/r+r'rr%s Er+rc"\rSrSrSrSrSrSrg)TestNrtMemInfoNotInitialized/zb Unit test for checking the use of the NRT fails if the initialization sequence has not been run. FcfSSSS.nUR5Hvup#UR[5nS[l[ [U5nU"U6 SSS5 SnUR U[WR55 S[lMx g!,(df  NE=f!S[lf=f)Nr/)r/r/)r/)library meminfo_new meminfo_allocFz'Runtime must be initialized before use.T) items assertRaises RuntimeErrorr_initgetattrassertInstr exception)r&methodsmethargsraisesfnmsgs r'test_init_fail+TestNrtMemInfoNotInitialized.test_init_fail6s "*$) "--/JD #&&|4"'EK -BI5 @ c3v'7'7#89" *44# s"B#!B/B# B B## B0r/N)r0r1r2r3__doc___numba_parallel_test_rJr4r/r+r'r6r6/s"#r+r6c`^\rSrSrSrU4SjrSrSrSrSr Sr \ S 5r S r S rU=r$) TestNrtMemInfoIz* Unit test for core MemInfo functionality c>S[l[R"[R 5 [ [U]#5 g)Nr) rr$r initializertarget_contextsuperrOsetUpr& __class__s r'rUTestNrtMemInfo.setUpNs+  223 nd)+r+c4[5nUR[RS5 Sn[R"X!5nURUR S5 AUR[RS5 UR 5 URUR S5 UR[RS5 UR5 URUR S5 AUR[RS5 g)Nr!J[=r)r assertEqualr$rr:refcountacquirerelease)r&daddrmis r'test_meminfo_refct_1#TestNrtMemInfo.test_meminfo_refct_1Us G a(   t ' a(  a(  a( a(  a(  a(r+c|[5nUR[RS5 Sn[R"X!5nURUR S5 AUR[RS5 [ S5HnUR5 M URUR S5 UR[RS5 [ S5HnUR5 M URUR S5 AUR[RS5 g)Nr!rZder) rr\r$rr:r]ranger^r_)r&r`rarbct_s r'test_meminfo_refct_2#TestNrtMemInfo.test_meminfo_refct_2fs G a(   t ' a(  a(*B JJL g. a(sA JJL a(  a(r+c[5nUR[RS5 Sn[R"X!5nURUR S5 [ U5nURUR S5 URX#R5 URUR5 URX4R5 URUR5 URURS5 URURS5 AAUR[RS5 AUR[RS5 g)Nr!rZr)rr\r$rr:r] memoryviewdata assertFalsereadonlyassertIsobj assertTrue c_contiguousitemsizendim)r&r`rarbmviews r'test_fake_memoryview#TestNrtMemInfo.test_fake_memoryviewys G a(   t ' a(2 a( ww' ( b))$ **+ + Q'  a(  a(r+cSSKJnJnJnJn [ R "[ R5nURS-n[R"USS9nURnU"U"UR5U"US-55n [S5H"n URU RU S5 M$ [S5Hn U S-U RU 'M [U5n URU R U5 UR#U R$5 UR'X{R(5 UR+U R,5 URU RS5 URU R.S5 A[ R0"X[R UR-U S9n A [ R2R5[ R6"U R85S-U 5 U S- n [S5H%n URU RU U S -5 M' URU RRU5 A g) Nrc_uint32c_void_pPOINTERcast TsafeKr!dtypeshapebufferr[)ctypesr}r~rrnpruint32rvrr;rorhr\contentsrnnbytesrprqrrrsrtrurwndarraytesting assert_equalarangesize) r&r}r~rrrbytesizerbrac_arrirxarrs r'test_memoryviewTestNrtMemInfo.test_memoryviews<<#>>B&   5wwXbgg&2 (>?rA   U^^A. ;rA !AENN1 2 x0 ( b))$ **+ + Q' jjuLLENN,J %'   #(( 3a 7= qrA   U^^A.A 6 $/ r+cSSKJnJnJnJn [ R "[ R5nURS-n[R"USS9nURURS5 URnU"U"U5U"US-55n [S5H"n URU RU S5 M$ [S5Hn U S-U RU 'M [ R "XVUR-US9n URURS5 A[ R"R%[ R&"U R(5S-U 5 U S- n [S5H%n URU RU U S -5 M' URU RRU5 A g) Nrr|rTrr!rrr[)rr}r~rrrrrrvrr;r\r]rorhrrrrrr) r&r}r~rrrrrbrarrrs r' test_bufferTestNrtMemInfo.test_buffersg<<#>>B&   5 a(wwXd^WX]%;<rA   U^^A. ;rA !AENN1 jju,F "$ a(   #(( 3a 7= qrA   U^^A.A 6 $/ r+c.[RRS-S-nSH^nUR[5n[ R "XS9 SSS5 URSUS3[WR55 M` g!,(df  N<=f)Nr[TFrzRequested allocation of z bytes failed.) rsize_tmaxvalr= MemoryErrorrr;rArBrC)r&rpredrGs r'test_allocate_invalid_size)TestNrtMemInfo.test_allocate_invalid_sizesw||""a'1,!D"";/6##D40 MM4TF.If../ 1"//s B B cSnSH`nUR[5n[R"XS9 SSS5 SUS3nUR U[ WR 55 Mb g!,(df  N>=f)Nirrz,Cannot allocate a negative number of bytes: .)r= ValueErrorrr;rArBrC)r&rrrGrIs r'test_allocate_negative_size*TestNrtMemInfo.test_allocate_negative_sizesd!D"":.&##D4/@aHC MM#s6#3#34 5 "..s A** A8 r/)r0r1r2r3rLrUrcrkryrrrrrr4 __classcell__rWs@r'rOrOIsF,)")&),$RH1166r+rOc$\rSrSrSrSrSrSrg)TestTracemallocz: Test NRT-allocated memory can be tracked by tracemalloc. c.SSKnWR5 UR 5nU"5nUR 5nAUR US5UR 5 $![a URS5 Nxf=f!UR 5 f=f)Nrztracemalloc not availablelineno) tracemalloc ImportErrorskipTeststart take_snapshot compare_tostop)r&funcrbeforeresafters r'measure_memory_diff#TestTracemalloc.measure_memory_diffs 7    ..0F&C--/E##FH5     7 MM5 6 7    sA!9B!A?>A?Bcx^^^ Sm[Rm [UU 4Sj5mU4SjnU4SjnURRS-nT"5 UR U5nUSnUR URT5 URURTS-SS 9 URSnUR[RRUR5S 5 URURU5 UR U5nUSnURURTS -5 g) Ni@Bc2>[R"TT5$)z$ Allocate and return a large array. )rempty)Nrsr'alloc_nrt_memory7TestTracemalloc.test_snapshot..alloc_nrt_memory s 88Au% %r+c>T"5$Nr/rsr' keep_memory2TestTracemalloc.test_snapshot..keep_memorys #% %r+c>T"5 grr/rsr'release_memory5TestTracemalloc.test_snapshot..release_memorys  r+r!rg= ףp=?zpUnexpected allocation overhead encountered. May be due to difference in CPython builds or running under coverage)rIz test_nrt.pyg{Gz?)rint8r__code__co_firstlinenorassertGreaterEqualr assertLess tracebackr\ospathbasenamefilenamer) r&rr alloc_linenodiffstatframerrrs @@@r' test_snapshotTestTracemalloc.test_snapshots   &  &  & #++::Q>  '' 4Aw  1-  1u9@  Bq! ))%..9=I |4''7Aw  1t8,r+r/N)r0r1r2r3rLrrr4r/r+r'rrs (-r+rcP\rSrSrSrSrSrSr\RS5r Sr g) TestNRTIssuei1c^^[S5m[U4Sj5m[U4Sj5nSn[RRUS45nU"X#5nURX#5n[RR XT5 g)z? GitHub Issue #1244 https://github.com/numba/numba/issues/1244 cHUup[R"US-US--5$)Nr[)mathsqrt)vectorxys r'calculate_2D_vector_magNTestNRTIssue.test_issue_with_refct_op_pruning..calculate_2D_vector_mag6s$DA99Q!Va1f_- -r+c>[R"S[RS9nT"U5nUup4X2- US'XB- US'U$Nr[rrr!)rrfloat64)rnormalized_vectormagrrrs r'normalize_2D_vectorJTestNRTIssue.test_issue_with_refct_op_pruning..normalize_2D_vector<sI ""** = )&1CDA#$7 a #$7 a $ $r+c>[R"US4[RS9n[U5H!nXnT"U5nUSX#S4'USX#S4'M# U$r)rrrrh) num_vectorsvectorsnormalized_vectorsrrrrs r'normalize_vectorsHTestNRTIssue.test_issue_with_refct_op_pruning..normalize_vectorsHsj!#;*:"**!M ;' $7$?!+.f`s"((33 3r+r!Cyr) rrTuple complex128ArrayrassertPreciseEqualrr)r&rcfunczrs r'test_incref_after_cast#TestNRTIssue.test_incref_after_cast]s 4 ekk5#3#3#(;;u{{As#C#FGJKKLN 2& RXXarxx%@Ar+c[S5nU"5up#[RRX#5 [RRU[R"S[R S95 g)Nc[R"S[RS9n[R"S[RS9nXSS4$)Nrr)ronesr)abs r'r5TestNRTIssue.test_refct_pruning_issue_1511..fls5"**-A"**-Ad7Nr+rr)rrrrrr)r&rrrs r'test_refct_pruning_issue_1511*TestNRTIssue.test_refct_pruning_issue_1511ksT     s % 2772RZZ#@Ar+c[S5n[R"/SQ/5nURUSS5nU"USS5nUR X45 g)Nc[R"US:H5n[US5S:XaSnSnOUSSnUSSnUS- U-nUS- U-nXg4$)Nr!r)rwherelen)imagerrnext_locy_offsetx_offset next_loc_x next_loc_ys r'udt7TestNRTIssue.test_refct_pruning_issue_1526..udtwsnxx +H8A;1$#A;q>#A;q>a%8+Ja%8+J) )r+) r!rr!rr!rrr!rrr!)rrarrayrr\)r&rrexpectrs r'test_refct_pruning_issue_1526*TestNRTIssue.test_refct_pruning_issue_1526vsU  *  * HH45 6Q1%!Ql %r+c$SSKJn UR[R5 US5nUR[R5 UR U"5UR 55 UR[R5 g)Nrrcg)N{r/r/r+r'foo8TestNRTIssue.test_no_nrt_on_njit_decoration..foosr+)numbarrprr?r\rrt)r&rr(s r'test_no_nrt_on_njit_decoration+TestNRTIssue.test_no_nrt_on_njit_decorationso  %     %  .  $r+r/N) r0r1r2r3rr rr#rrun_test_in_subprocessr+r4r/r+r'rr1s3)6V B B&.$$%%%r+rch\rSrSrSrSr\R"S5S5r\ \ S55r Sr g) TestRefCtPruningia define i32 @"MyFunction"(i8** noalias nocapture %retptr, { i8*, i32 }** noalias nocapture %excinfo, i8* noalias nocapture readnone %env, double %arg.vt.0, double %arg.vt.1, double %arg.vt.2, double %arg.vt.3, double %arg.bounds.0, double %arg.bounds.1, double %arg.bounds.2, double %arg.bounds.3, i8* %arg.xs.0, i8* nocapture readnone %arg.xs.1, i64 %arg.xs.2, i64 %arg.xs.3, double* nocapture readonly %arg.xs.4, i64 %arg.xs.5.0, i64 %arg.xs.6.0, i8* %arg.ys.0, i8* nocapture readnone %arg.ys.1, i64 %arg.ys.2, i64 %arg.ys.3, double* nocapture readonly %arg.ys.4, i64 %arg.ys.5.0, i64 %arg.ys.6.0, i8* %arg.aggs_and_cols.0.0, i8* nocapture readnone %arg.aggs_and_cols.0.1, i64 %arg.aggs_and_cols.0.2, i64 %arg.aggs_and_cols.0.3, i32* nocapture %arg.aggs_and_cols.0.4, i64 %arg.aggs_and_cols.0.5.0, i64 %arg.aggs_and_cols.0.5.1, i64 %arg.aggs_and_cols.0.6.0, i64 %arg.aggs_and_cols.0.6.1) local_unnamed_addr { entry: tail call void @NRT_incref(i8* %arg.xs.0) tail call void @NRT_incref(i8* %arg.ys.0) tail call void @NRT_incref(i8* %arg.aggs_and_cols.0.0) %.251 = icmp sgt i64 %arg.xs.5.0, 0 br i1 %.251, label %B42.preheader, label %B160 B42.preheader: ; preds = %entry %0 = add i64 %arg.xs.5.0, 1 br label %B42 B42: ; preds = %B40.backedge, %B42.preheader %lsr.iv3 = phi i64 [ %lsr.iv.next, %B40.backedge ], [ %0, %B42.preheader ] %lsr.iv1 = phi double* [ %scevgep2, %B40.backedge ], [ %arg.xs.4, %B42.preheader ] %lsr.iv = phi double* [ %scevgep, %B40.backedge ], [ %arg.ys.4, %B42.preheader ] %.381 = load double, double* %lsr.iv1, align 8 %.420 = load double, double* %lsr.iv, align 8 %.458 = fcmp ole double %.381, %arg.bounds.1 %not..432 = fcmp oge double %.381, %arg.bounds.0 %"$phi82.1.1" = and i1 %.458, %not..432 br i1 %"$phi82.1.1", label %B84, label %B40.backedge B84: ; preds = %B42 %.513 = fcmp ole double %.420, %arg.bounds.3 %not..487 = fcmp oge double %.420, %arg.bounds.2 %"$phi106.1.1" = and i1 %.513, %not..487 br i1 %"$phi106.1.1", label %B108.endif.endif.endif, label %B40.backedge B160: ; preds = %B40.backedge, %entry tail call void @NRT_decref(i8* %arg.ys.0) tail call void @NRT_decref(i8* %arg.xs.0) tail call void @NRT_decref(i8* %arg.aggs_and_cols.0.0) store i8* null, i8** %retptr, align 8 ret i32 0 B108.endif.endif.endif: ; preds = %B84 %.575 = fmul double %.381, %arg.vt.0 %.583 = fadd double %.575, %arg.vt.1 %.590 = fptosi double %.583 to i64 %.630 = fmul double %.420, %arg.vt.2 %.638 = fadd double %.630, %arg.vt.3 %.645 = fptosi double %.638 to i64 tail call void @NRT_incref(i8* %arg.aggs_and_cols.0.0) ; GONE 1 tail call void @NRT_decref(i8* null) ; GONE 2 tail call void @NRT_incref(i8* %arg.aggs_and_cols.0.0), !noalias !0 ; GONE 3 %.62.i.i = icmp slt i64 %.645, 0 %.63.i.i = select i1 %.62.i.i, i64 %arg.aggs_and_cols.0.5.0, i64 0 %.64.i.i = add i64 %.63.i.i, %.645 %.65.i.i = icmp slt i64 %.590, 0 %.66.i.i = select i1 %.65.i.i, i64 %arg.aggs_and_cols.0.5.1, i64 0 %.67.i.i = add i64 %.66.i.i, %.590 %.84.i.i = mul i64 %.64.i.i, %arg.aggs_and_cols.0.5.1 %.87.i.i = add i64 %.67.i.i, %.84.i.i %.88.i.i = getelementptr i32, i32* %arg.aggs_and_cols.0.4, i64 %.87.i.i %.89.i.i = load i32, i32* %.88.i.i, align 4, !noalias !3 %.99.i.i = add i32 %.89.i.i, 1 store i32 %.99.i.i, i32* %.88.i.i, align 4, !noalias !3 tail call void @NRT_decref(i8* %arg.aggs_and_cols.0.0), !noalias !0 ; GONE 4 tail call void @NRT_decref(i8* %arg.aggs_and_cols.0.0) ; GONE 5 br label %B40.backedge B40.backedge: ; preds = %B108.endif.endif.endif, %B84, %B42 %scevgep = getelementptr double, double* %lsr.iv, i64 1 %scevgep2 = getelementptr double, double* %lsr.iv1, i64 1 %lsr.iv.next = add i64 %lsr.iv3, -1 %.294 = icmp sgt i64 %lsr.iv.next, 1 br i1 %.294, label %B42, label %B160 } cbURn[UR55nUVs/sH nSU;dM UPM nnUVs/sH nSU;dM UPM nn[R"U5n[UR55nUVs/sH nSU;dM UPM nnUVs/sH nSU;dM UPM n nUR XH5 UR XY5 [ U5[ U5- n [ U5[ U 5- n X-n URXU - 5 SRU 5n SH%nSRU5nURX5 M' UR[[U R555[U 55 gs snfs snfs snfs snf)N NRT_incref NRT_decref )r!r[z ; GONE {}) sample_llvm_irlist splitlinesr_remove_redundant_nrt_refctassertNotEqualsetr\joinformatrAr)r&input_ir input_lineslnbefore_increfsbefore_decrefs output_ir output_lines after_increfs after_decrefspruned_increfspruned_decrefscombined pruned_linesrgones r'test_refct_pruning_op_recognize0TestRefCtPruning.test_refct_pruning_op_recognizes&&8..01 '2I{lb6H"{I'2I{lb6H"{I66x@ I0023 &2Illb6Hl I&2Illb6Hl I N: N:^,s=/AA^,s=/AA"2 N#BCyy* !A%%a(D MM$ -! T,"9"9";<=s8}M5JI JIs. FF F"F" F'F'' F,5F,z3Pass removed as it was buggy. Re-enable when fixed.cd^^^[S5m[S5m[UU4Sj5m[SS9U4Sj5nURS5 [URURS55n[ [ R"S U55nUR[U5S5 g ) ztestcase from #2350cX[R"U5(dX!U4==S- ss'ggr risnanrraggfields r'_append_non_naITestRefCtPruning.test_refct_pruning_with_branches.._append_non_nas#88E??qD Q #r+c[R"U5(d4[R"X!U45(aX2X4'gX!U4==U- ss'ggrrQrSs r'_appendBTestRefCtPruning.test_refct_pruning_with_branches.._appendsB88E??88C1I&& %I1I&I #r+c*>T"XX#5 T"XX#5 grr/)rrrTrUrYrVs r'appendATestRefCtPruning.test_refct_pruning_with_branches..appends 1 , A# %r+T)no_cpython_wrapperc>[URS5H+n[URS5H nT"X2X5 M M- g)Nrr!)rhr)rrUrjr\s r'extendATestRefCtPruning.test_refct_pruning_with_branches..extend&s;399Q<(syy|,A1,-)r+z(f4[:,::1], f4)rz!(NRT_incref|NRT_decref)\([^\)]+\)N) rcompilerB inspect_llvm signaturesr8refinditerr\r)r&rallvmirrefopsrYrVr\s @@@r' test_refct_pruning_with_branches1TestRefCtPruning.test_refct_pruning_with_branchess      '  '  &  &  & - ' - ()V(():):1)=>?bkk"FOP Va(r+c^[R"5 [R"5 [R"5 [S5m[ S/5U4Sj5nUR U"S5S5 g)zlThe InlineAsm class from llvmlite.ir has no 'name' attr the refcount pruning pass should be tolerant to thisczSn[[R[R[R5U4$)Nc  UupE[R"[R"S5[R"S5[R"S5/5nURUSSXE4SSS9n[ XUR U5$)N zmov $2, $0; imul $1, $0z=&r,r,rasm_mulF)name side_effect)r FunctionTypeIntTypeasmr return_type)cgctxbuildersigrFarg_0arg_1ftymuls r'codegen>TestRefCtPruning.test_inline_asm..bar..codegen?sv!%oobjjnrzz"~79zz"~7GHkk#'@)#(.y.3"5*%#//3OOr+)r rr)tyctxrrr~s r'bar-TestRefCtPruning.test_inline_asm..bar=s* PU[[%++u{{CWL Lr+z int32(int32)c$>US- nT"US5nU$)Nr!r[r/)rr rs r'r(-TestRefCtPruning.test_inline_asm..fooIs FAAq AHr+rN)llvminitialize_native_targetinitialize_native_asmprinterinitialize_native_asmparserr rr\)r&r(rs @r'test_inline_asm TestRefCtPruning.test_inline_asm4so %%' ))+ ((*  M  M ~     R"%r+r/N) r0r1r2r3r7rMunittestskiprj linux_onlyx86_onlyrr4r/r+r'r/r/sMFNPN>]]HI")J")H &&r+r/cJ^\rSrSrSrU4SjrSrSrSrSr Sr S r U=r $) TestNrtExternalCFFIiRz?Testing the use of externally compiled C code that use NRT cJ>[R [[U]5 gr)rrSrTrrUrVs r'rUTestNrtExternalCFFI.setUpVs!! !4.0r+cSSKJn U"5nURX[5/S9 UR U5 [ SR U55nURUS9 [RRU5 [U5n[RRU5 XW4$![RRU5 f=f)NrFFI) include_dirsz cffi_test_{})tmpdir) cffir set_sourcer cdefrr>rcsysrr\rremove)r&rqsourcerrffirmods r'compile_cffi_module'TestNrtExternalCFFI.compile_cffi_module[se t<>2BC  5 5d ;< 6 "  $ &C HHOOF #x HHOOF #s  B..!CctSSKJn U"5nURS[RS5nU"5nU$)Nrrz void* (*)()get_api)rrrr c_helpers)r&rr nrt_get_apitables r'get_nrt_api_table%TestNrtExternalCFFI.get_nrt_api_tableks3ehh}k.C.CI.NO   r+cTSRURR5nSnSnURXU5upEUR UR R S5 UR5nUR RU5nUR UR R S5 [URSU55n[R"U5n UR U RS5 A UR UR R S5 g) Nz{}_test_manage_memoryaz #include #include "numba/core/runtime/nrt_external.h" int status = 0; void my_dtor(void *ptr) { free(ptr); status = 0xdead; } NRT_MemInfo* test_nrt_api(NRT_api_functions *nrt) { void * data = malloc(10); NRT_MemInfo *mi = nrt->manage_memory(data, my_dtor); nrt->acquire(mi); nrt->release(mi); status = 0xa110c; return mi; } z; void* test_nrt_api(void *nrt); extern int status; ri  rr!i)r>rWr0rr\libstatusr test_nrt_apiintrr MemInfor]) r&rqrrrrroutmi_addrrbs r'test_manage_memory&TestNrtExternalCFFI.test_manage_memoryss&--dnn.E.EF (  ++D$? +&&(gg""5) 1chhx-. [[ ! a(  0r+cSRURR5nSnSnURXU5upEUR 5nS[ R "[ R5R-nURRXg5n[URSU55n [R"U 5n URU R S5 UR#URSRU5U R$55n [ R&"S[ RU S 9n [ R(R+U /S Q5 g) Nz{}_test_allocatea7 #include #include "numba/core/runtime/nrt_external.h" NRT_MemInfo* test_nrt_api(NRT_api_functions *nrt, size_t n) { size_t *data = NULL; NRT_MemInfo *mi = nrt->allocate(n); data = nrt->get_data(mi); data[0] = 0xded; data[1] = 0xabc; data[2] = 0xdef; return mi; } z(void* test_nrt_api(void *nrt, size_t n);r4rr!z char [{}])r4)rrr)i i i )r>rWr0rrrrintprvrrrrr rr\r]rrorrr) r&rqrrrrrnumbytesrrrbrrs r' test_allocate!TestNrtExternalCFFI.test_allocates!(()@)@A  :++D$?&&(rxx(111gg""53chhx-. [[ ! a(CHH[%7%7%A277KLjjt2776B %:;r+cSSKJn [S5nU"5n[UR SUR 555nU"5nUR XE5 g)Nrrc[5$rrr/r+r'r6TestNrtExternalCFFI.test_get_api..test_nrt_apis = r+r)rrrrrrr\)r&rrrr"rs r' test_get_api TestNrtExternalCFFI.test_get_apisS  !  !eSXXh(>(>(@ABn %r+r/) r0r1r2r3rLrUrrrrrr4rrs@r'rrRs-1  '1R<@ & &r+rcJ\rSrSrSrSrSrSrSrSr Sr S r S r S r g ) TestNrtStatisticsic8[R"5Ulgr)rmemsys_stats_enabled_TestNrtStatistics__stats_stater%s r'rUTestNrtStatistics.setUps(==?r+c|UR(a[R"5 g[R"5 gr)rrmemsys_enable_statsmemsys_disable_statsr%s r'tearDownTestNrtStatistics.tearDowns#     + + -  , , .r+c`Sn[RR5nSUS'[XS9 g)Naif 1: from numba import njit import numpy as np from numba.core.runtime import rtsys, _nrt_python from numba.core.registry import cpu_target @njit def foo(): return np.arange(10)[0] # initialize the NRT before use rtsys.initialize(cpu_target.target_context) assert _nrt_python.memsys_stats_enabled() orig_stats = rtsys.get_allocation_stats() foo() new_stats = rtsys.get_allocation_stats() total_alloc = new_stats.alloc - orig_stats.alloc total_free = new_stats.free - orig_stats.free total_mi_alloc = new_stats.mi_alloc - orig_stats.mi_alloc total_mi_free = new_stats.mi_free - orig_stats.mi_free expected = 1 assert total_alloc == expected assert total_free == expected assert total_mi_alloc == expected assert total_mi_free == expected 1NUMBA_NRT_STATSenv)renvironcopyr)r&srcrs r'test_stats_env_var_explicit_on0TestNrtStatistics.test_stats_env_var_explicit_ons. 8jjoo!$ #'r+cSn[X!S9 g)Naif 1: from numba import njit import numpy as np from numba.core.runtime import rtsys, _nrt_python @njit def foo(): return np.arange(10)[0] assert _nrt_python.memsys_stats_enabled() == False try: rtsys.get_allocation_stats() except RuntimeError as e: assert "NRT stats are disabled." in str(e) r)r)r&rrs r'check_env_var_off#TestNrtStatistics.check_env_var_offs  #'r+cl[RR5nSUS'URU5 g)N0r)rrrrr&rs r'test_stats_env_var_explicit_off1TestNrtStatistics.test_stats_env_var_explicit_off s+jjoo!$  s#r+c[RR5nURSS5 UR U5 g)Nr)rrrpoprrs r'test_stats_env_var_default_off0TestNrtStatistics.test_stats_env_var_default_offs1jjoo !4( s#r+ct[S5n[R"5 UR[R"55 [ S5Hn[ R"5n[R"5 UR[R"55 U"5 [R"5 UR[R"55 [ R"5nU"5 [ R"5nURX45 URXE5 M g)Ncf[R"S5n[R"SUS-5$)Nr4r6r)rrr)tmps r'r(7TestNrtStatistics.test_stats_status_toggle..foos&''!*C99QQZ( (r+r[) rrrrtrrhrget_allocation_statsrrpr\r)r&r(rstats_1stats_2stats_3s r'test_stats_status_toggle*TestNrtStatistics.test_stats_status_toggles  )  ) '')  88:;qA002G  , , .   [==? @ E  + + - OOK<<> ?002G E002G   W . OOG --r+cH[R"5 UR[R"55 UR [ 5n[ R"5 SSS5 URS[WR55 g!,(df  N4=f)NNRT stats are disabled.) rrrprr=r>rrrArBrC)r&rGs r'5test_rtsys_stats_query_raises_exception_when_disabledGTestNrtStatistics.test_rtsys_stats_query_raises_exception_when_disabled<sm ((* 99;<   | ,  & & (- /V5E5E1FG- ,s B B!cSnUHn[[SU35nURUS9 [R"5 UR [R "55 UR [5nU"5 SSS5 URS[WR55 SSS5 M g!,(df  N@=f!,(df  M=f)N)allocfreemi_allocmi_freememsys_get_stats_) stats_funcr) r@rsubTestrrprr=r>rArBrC)r&method_variationsrErrGs r'r sS   9'2-FFF+__!8!8!AA.0 ??8++-5GG* ,F#8#4#4#4e6X&&e6P=-h''=-@r%8r%ji&x((i&Xs&-xs&s&lKPKP\ z MMOr+