J/Ph[t d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
Z
d dl m Z m
Z
mZmZ d dlmZmZ d dlmZ d dlmZ d dlmZ d dlmZ d d l m Z d d
l m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z% d d l&m'Z' d dl(Z( e(j) ej * d d
Z+ e(j) ej, dvd Z- G d de. Z/ G d de(j Z0 G d de(j Z1 G d de(j Z2 G d de Z3 G d de(j Z4e$ G d d e e Z5 G d d e Z6e7d k r e(j8 dS dS )! 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 e Zd ZdZd Zd ZdS )Dummyr c @ t | xj dz
c_ d S N typealiveselfs T/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numba/tests/test_nrt.py__init__zDummy.__init__( " T
A c @ t | xj dz c_ d S r r! r$ s r& __del__z
Dummy.__del__+ r( r) N)__name__
__module__ __qualname__r# r' r+ r) r& r r % s7
E r) r c e Zd ZdZdZd ZdS ) TestNrtMemInfoNotInitializedzn
Unit test for checking the use of the NRT fails if the
initialization sequence has not been run.
Fc dddd}| D ]\ }} | t 5 }dt _ t t | } || d d d n # 1 swxY w Y d}| |t |j dt _ # dt _ w xY wd S )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_failz+TestNrtMemInfoNotInitialized.test_init_fail6 s
"*$)
"--//
#
#JD$
#&&|44 "'EK --BBII
@
c3v'7#8#8999 " d """"
#
#s. B0'A." B0.A2 2B05A2 6-B00B>N)r, r- r. __doc___numba_parallel_test_rD r/ r) r& r1 r1 / s5 "# # # # #r) r1 c \ e Zd ZdZ fdZd Zd Zd Zd Zd Z e
d Z d Z xZ
S )
TestNrtMemInfoz2
Unit test for core MemInfo functionality
c dt _ t j t j t
t | d S )Nr ) r r# r
initializer target_contextsuperrH setUpr% __class__s r& rM zTestNrtMemInfo.setUpN s@
2333
nd##))+++++r) c F t }| t j d d}t j || }| |j d ~| t j d | | |j d | t j d | | |j d ~| t j d d S )Nr J[= r )r assertEqualr# r r4 refcountacquirerelease)r% daddrmis r& test_meminfo_refct_1z#TestNrtMemInfo.test_meminfo_refct_1U s GG a(((
tQ
'
' a(((
a(((
a((( a(((
a((( a(((((r) c t }| t j d d}t j || }| |j d ~| t j d t
d D ]}| | |j d | t j d t
d D ]}| | |j d ~| t j d d S )Nr rQ d e r ) r rS r# r r4 rT rangerU rV )r% rW rX rY ct_s r& test_meminfo_refct_2z#TestNrtMemInfo.test_meminfo_refct_2f s GG a(((
tQ
'
' a(((
a(((** B JJLLLL g... a(((s A JJLLLL a((( a(((((r) c t }| t j d d}t j || }| |j d t
| }| |j d | ||j | |j |
||j | |j
| |j d | |j d ~~| t j d ~| t j d d S )Nr rQ r )r rS r# r r4 rT
memoryviewdata assertFalsereadonlyassertIsobj
assertTrue c_contiguousitemsizendim)r% rW rX rY mviews r& test_fake_memoryviewz#TestNrtMemInfo.test_fake_memoryviewy s= GG a(((
tQ
'
' a(((2 a((( rw''' (((
b%)$$$ *+++ +++ Q'''
a((( a(((((r) c n ddl m}m}m}m} t j t
j }|j dz }t j
|d }|j } | ||j ||dz } t d D ]#}
|
| j |
d $t d D ]}
|
dz | j |
< t | } |
| j | | | j | || j | | j |
| j d |
| j d ~t j || j |j z | } ~ t
j t j | j dz | | dz
} t d D ]&}
|
| j |
|
d z '|
| j j | ~ d S )
Nr c_uint32c_void_pPOINTERcast
Tsafe K r dtypeshapebufferrR ) ctypesrq rr rs rt nprz uint32rk r r5 rd r^ rS contentsrc nbytesre rf rg rh ri rj rl ndarraytesting assert_equalarangesize)
r% rq rr rs rt rz bytesizerY rX c_arrirm arrs
r& test_memoryviewz TestNrtMemInfo.test_memoryview s7 <<<<<<<<<<<<##>B&
5
5
5wXXbg&&2
(>(>??r < B&
5
5
5 a(((wXXd^^WWX]%;%;<<r < .alloc_nrt_memory
s
8Au%% %r) c S Nr/ r s r& keep_memoryz2TestTracemalloc.test_snapshot..keep_memory s ##%% %r) c d S r r/ r s r& release_memoryz5TestTracemalloc.test_snapshot..release_memory s r) r r g=
ףp=?zpUnexpected allocation overhead encountered. May be due to difference in CPython builds or running under coverage)rC z test_nrt.pyg{Gz?)r~ int8r __code__co_firstlinenor assertGreaterEqualr
assertLess tracebackrS ospathbasenamefilenamer )
r% r r alloc_linenodiffstatframer r rz s
@@@r&
test_snapshotz TestTracemalloc.test_snapshot s~
& & & & &
& & & & & & #+:Q> '' 44Aw 1--- 1u9 @ B B B q! ))%.99=III |444 ''77Aw 1t8,,,,,r) N)r, r- r. rE r r r/ r) r& r r s< (- (- (- (- (-r) r c F e Zd Zd Zd Zd Zd Zej d Z dS ) TestNRTIssuec < t d t fd t fd }d}t j |df } ||| }| || }t j || dS )zO
GitHub Issue #1244 https://github.com/numba/numba/issues/1244
c F | \ }}t j |dz |dz z S )NrR )mathsqrt)vectorxys r& calculate_2D_vector_magzNTestNRTIssue.test_issue_with_refct_op_pruning..calculate_2D_vector_mag6 s' DAq9Q!Va1f_-- -r) c t j dt j } | }| \ }}||z |d< ||z |d< |S NrR rz r r )r~ r float64)r normalized_vectormagr r r s r& normalize_2D_vectorzJTestNRTIssue.test_issue_with_refct_op_pruning..normalize_2D_vector< sW ""* = = = ))&11CDAq#$s7 a #$s7 a $ $r) c t j | dft j }t | D ]/}|| } | }|d ||df< |d ||df< 0|S r )r~ r r r^ ) num_vectorsvectorsnormalized_vectorsr r r r s r& normalize_vectorszHTestNRTIssue.test_issue_with_refct_op_pruning..normalize_vectorsH s !#;*:"*!M!M!M ;''
@
@ $7$7$?$?!+.f` s "(3333 3r) r Cy r ) r r Tuple
complex128Arrayr assertPreciseEqualr~ r )r% r cfunczr s r& test_incref_after_castz#TestNRTIssue.test_incref_after_cast] s 4 4 4
K Gek5#3#(;u{As#C#C#F G G J J K KKLN N3 2&&& RXarx%@%@%@AAAAAr) c t d } | \ }}t j || t j |t j dt j d S )Nc t j dt j } t j dt j }| |d d fS )Nru r )r~ onesr )abs r& r z5TestNRTIssue.test_refct_pruning_issue_1511..fl s? "*---A"*---Aad7Nr) ru r )r r~ r r r r )r% r r r s r& test_refct_pruning_issue_1511z*TestNRTIssue.test_refct_pruning_issue_1511k ss
qss1
1%%%
272RZ#@#@#@AAAAAr) c t d }t j g dg }| |dd } ||dd }| || d S )Nc t j | dk }t |d dk rd}d}n |d d }|d d }|dz
|z }|dz
|z }||fS )Nr r )r~ wherelen)imager r next_locy_offsetx_offset
next_loc_x
next_loc_ys r& udtz7TestNRTIssue.test_refct_pruning_issue_1526..udtw sy x
++H8A; 1$$ #A;q>#A;q>a%8+Ja%8+J z) )r) )
r r r r r r r r r r r )r r~ arrayr rS )r% r r expectr s r& test_refct_pruning_issue_1526z*TestNRTIssue.test_refct_pruning_issue_1526v sx
*
*
*
H4445 6 6Q1%%c!Qll %%%%%r) c B ddl m} | t j |d }| t j | | | | t j d S )Nr r c dS )N{ r/ r/ r) r& fooz8TestNRTIssue.test_no_nrt_on_njit_decoration..foo s 3r) )numbar re r r9 rS r ri )r% r r s r& test_no_nrt_on_njit_decorationz+TestNRTIssue.test_no_nrt_on_njit_decoration s
%%%
%%%
...
$$$$$r) N)
r, r- r. r r r r r run_test_in_subprocessr
r/ r) r& r r 1 sv )6 )6 )6V B B B B B B& & &. $% % %$% % %r) r c p e Zd ZdZd Z ej d d Zee d Z
dS )TestRefCtPruninga
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
}
c X | j }t | }d |D }d |D }t j | }t | }d |D }d |D }| || | || t
| t
| z
} t
| t
| z
}
| |
z } | | | |
z d | } dD ]-}
d |
}|
|| .| t t | t | d S )Nc g | ]}d |v | S
NRT_increfr/ .0lns r&
zDTestRefCtPruning.test_refct_pruning_op_recognize.. " IIIlb6H6H"6H6H6Hr) c g | ]}d |v | S
NRT_decrefr/ r s r& r zDTestRefCtPruning.test_refct_pruning_op_recognize.. r r) c g | ]}d |v | S r r/ r s r& r zDTestRefCtPruning.test_refct_pruning_op_recognize.. " IIIlb6H6H6H6H6Hr) c g | ]}d |v | S r r/ r s r& r zDTestRefCtPruning.test_refct_pruning_op_recognize.. r r)
)r rR z ; GONE {}) sample_llvm_irlist
splitlinesr _remove_redundant_nrt_refctassertNotEqualsetrS joinformatr; r )r% input_ir input_linesbefore_increfsbefore_decrefs output_ir output_lines
after_increfs
after_decrefspruned_increfspruned_decrefscombined pruned_linesr gones r& test_refct_pruning_op_recognizez0TestRefCtPruning.test_refct_pruning_op_recognize s &8..0011 II{IIIII{III 6x@@ I 0 0 2 233 IIlIII
IIlIII
NM::: NM::: ^,,s=/A/AA ^,,s=/A/AA "N2 >N#BCCCyy** ! . .A %%a((D MM$ - - - - T,"9"9";"; < <==s8}}MMMMMr) z3Pass removed as it was buggy. Re-enable when fixed.c t d t d t fd t d fd }| d t | |j d }t t
j d | }| t | d d
S ) ztestcase from #2350c V t j | s||| fxx dz
cc< d S d S r r~ isnanr r aggfields r& _append_non_nazITestRefCtPruning.test_refct_pruning_with_branches.._append_non_na s9 8E??
AqD Q
r) c t j | s9t j ||| f r |||| f< d S ||| fxx |z
cc< d S d S r r9 r; s r& _appendzBTestRefCtPruning.test_refct_pruning_with_branches.._append se 8E??
'8C1I&& ' %C1III1III&IIIII
'
'r) c @ | ||| | ||| d S r r/ )r r r< r= r@ r> s r& appendzATestRefCtPruning.test_refct_pruning_with_branches..append s5 N1ae , , , GAq#u % % % % %r) T)no_cpython_wrapperc t | j d D ]-}t | j d D ]} ||| | .d S )Nr r )r^ r{ )r r= r jrB s r& extendzATestRefCtPruning.test_refct_pruning_with_branches..extend& se 39Q<((
-
- sy|,, - -AF1ae,,,,-
-
-r) z(f4[:,::1], f4)r z!(NRT_incref|NRT_decref)\([^\)]+\)N)
r compiler< inspect_llvm
signaturesr" refinditerrS r )r% rF llvmirrefopsr@ r> rB s @@@r& test_refct_pruning_with_branchesz1TestRefCtPruning.test_refct_pruning_with_branches s
' '
'
& & & & &
&
& & & - - - -
' & - ())) V(():1)=>>??bk"FOOPP Va(((((r) c 4 t j t j t j t j t
d t
dg fd }| |d d dS )ztThe InlineAsm class from llvmlite.ir has no 'name' attr the refcount
pruning pass should be tolerant to thisc j d }t t j t j t j |fS )Nc |\ }}t j t j d t j d t j d g }| |dd||fdd }t | ||j | S )N zmov $2, $0; imul $1, $0z=&r,r,rasm_mulF)name side_effect)r FunctionTypeIntTypeasmr
return_type)cgctxbuildersigr@ arg_0arg_1ftymuls r& codegenz>TestRefCtPruning.test_inline_asm..bar..codegen@ s !%objnnrz"~~79z"~~7G H H kk#'@)#(%.y.3 " 5 5 *%#/3OOOr) )r r r )tyctxr r ra s r& barz-TestRefCtPruning.test_inline_asm..bar> s3
P
P
P U[%+u{CCWL Lr) z int32(int32)c * | dz
} | d }|S )Nr rR r/ )r r rc s r& r z-TestRefCtPruning.test_inline_asm..fooJ s
FAAq AHr) ru N)llvmrJ initialize_native_target initialize_native_asmprinterinitialize_native_asmparserr
r rS )r% r rc s @r& test_inline_asmz TestRefCtPruning.test_inline_asm4 s
%''' )+++ (*** M M
M
~
R"%%%%%r) N) r, r- r. r! r6 unittestskiprN
linux_onlyx86_onlyrj r/ r) r& r
r
s FNP N N N> X]HII") ") JI")H
& & X Z& & &r) r
c @ e Zd ZdZ fdZd Zd Zd Zd Zd Z xZ
S )TestNrtExternalCFFIz?Testing the use of externally compiled C code that use NRT
c n t j t t | d S r )r rK rL rp rM rN s r& rM zTestNrtExternalCFFI.setUpW s/ !!
!4((..00000r) c ddl m} | }| ||t g | | t d | }| | t j
| t | }t j | n$# t j | w xY w||fS )Nr FFI) include_dirsz cffi_test_{})tmpdir)
cffirt
set_sourcer cdefr r( rG sysr rB r remove)r% rT sourcery rt ffirv mods r& compile_cffi_modulez'TestNrtExternalCFFI.compile_cffi_module\ s cee tV<>>2BCCC 5 5d ; ;<< 6 """ $ &&C HOOF # # # #CHOOF # # # #Cxs #C !C3c ddl m} | }| dt j d } | }|S )Nr rs z void* (*)()get_api)rw rt rt r c_helpers)r% rt r} nrt_get_apitables r& get_nrt_api_tablez%TestNrtExternalCFFI.get_nrt_api_tablel sJ ceehh}k.CI.NOO
r) c B d | j j }d}d}| ||| \ }}| |j j d | }|j | }| |j j d t |
d| }t j | } | | j
d ~ | |j j d d S ) 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;
r i
r r i )r( rO r, r rS libstatusr test_nrt_apiintrt r MemInforT )
r% rT r| ry r} r~ r outmi_addrrY s
r& test_manage_memoryz&TestNrtExternalCFFI.test_manage_memoryt s &--dn.EFF (
++D&$??S +++&&((g""5)) 111chhx--..
[
!
! a((( 00000r) c d | j j }d}d}| ||| \ }}| }dt j t
j j z }|j
|| }t | d| } t j | }
| |
j d | | d | |
j } t j dt
j | } t
j | g d
d S ) 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);r r r z char [{}])r )r{ rz r| )i
i
i
)r( rO r, r r r~ rz intprk r r r rt r r rS rT r| rd r r r )
r% rT r| ry r} r~ r numbytesr r rY r| r s
r&
test_allocatez!TestNrtExternalCFFI.test_allocate s2 !(( )@AA
:++D&$??S&&((rx((11g""5(33chhx--..
[
!
! a(((CHH[%7%7%A%A27 K KLLjt276BBB
%:%:%:;;;;;r) c ddl m} t d } | }t | d| } | }| || d S )Nr rs c t S r r r/ r) r& r z6TestNrtExternalCFFI.test_get_api..test_nrt_api s == r) r )rw rt r r rt r rS )r% rt r r} r r s r& test_get_apiz TestNrtExternalCFFI.test_get_api s
! !
! ceeSXXh(>(>(@(@AABBlnn %%%%%r) ) r, r- r. rE rM r r r r r r r s @r& rp rp S s 1 1 1 1 1
'1 '1 '1R < < <@
&
&
&
&
&
&
&r) rp c D e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
S ) TestNrtStatisticsc 6 t j | _ d S r )r memsys_stats_enabled _TestNrtStatistics__stats_stater$ s r& rM zTestNrtStatistics.setUp s ( = ? ?r) c d | j rt j d S t j d S r )r r memsys_enable_statsmemsys_disable_statsr$ s r& tearDownzTestNrtStatistics.tearDown s7 / + - - - - - , . . . . .r) c r d}t j }d|d< t || d S )Na if 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)r environcopyr )r% srcr s r& test_stats_env_var_explicit_onz0TestNrtStatistics.test_stats_env_var_explicit_on s@ 8 joo !$ #3''''''r) c , d}t || d S )Na if 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% r r s r& check_env_var_offz#TestNrtStatistics.check_env_var_off s$ #3''''''r) c v t j }d|d< | | d S )N0r )r r r r r% r s r& test_stats_env_var_explicit_offz1TestNrtStatistics.test_stats_env_var_explicit_off s8 joo !$ s#####r) c t j }| dd | | d S )Nr )r r r popr r s r& test_stats_env_var_default_offz0TestNrtStatistics.test_stats_env_var_default_off sD joo !4((( s#####r) c t d }t j | t j t d D ]}t
j }t j | t j | t j | t j t
j } | t
j }|
|| | || d S )Nc d t j d } t j d| d z S )Nr r r )r~ r r )tmps r& r z7TestNrtStatistics.test_stats_status_toggle..foo s' '!**C9QQZ(( (r) rR ) r r r ri r r^ r get_allocation_statsr re rS r )r% r r stats_1stats_2stats_3s r& test_stats_status_togglez*TestNrtStatistics.test_stats_status_toggle s)
) )
)
'))) 8::;;;q . .A022G , . . . [ = ? ? @ @ @ CEEE + - - - OOK < > > ? ? ?022G CEEE022G Wg . . . OOGW - - - -- . .r) c R t j | t j | t
5 }t
j d d d n # 1 swxY w Y | dt |j
d S )NNRT stats are disabled.) r r re r r7 r8 r r r; r< r= )r% rA s r& 5test_rtsys_stats_query_raises_exception_when_disabledzGTestNrtStatistics.test_rtsys_stats_query_raises_exception_when_disabled= s (*** 9;;<<<
|
,
, ) & ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) )
/V5E1F1FGGGGGs A44A8;A8c d}|D ]}t t d| }| | 5 t j | t j | t 5 } | d d d n # 1 swxY w Y | dt |j
d d d n # 1 swxY w Y d S )N)allocfreemi_allocmi_freememsys_get_stats_)
stats_funcr ) r: r subTestr re r r7 r8 r; r< r= )r% method_variationsr? r rA s r& r s
9 8 8 8 8 8 8 8 ' ' ' ' ' ' 2 2 2 2 2 2 - - - - - -F F F F F F F F F F F F F F F F F F + * * * * * X_ !8!8!A!A A .0 0
8?+8+--5GG* , , F # # # # #8#4 # # #4e6 e6 e6 e6 e6X& e6 e6 e6P=- =- =- =- =-h' =- =- =-@r% r% r% r% r%8 r% r% r%jj& j& j& j& j&x( j& j& j&Z s& s& s& s& s&-x s& s& s&lKP KP KP KP KP KP KP KP\ zHMOOOOO r)