gXu d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
Z
d dl m Z m
Z
d dlmZmZmZmZ d dlmZmZ d dlmZ d dlmZ d d lmZ d dl m Z 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
Z,d
e,_ e+j- ej . d d Z/ e+j- ej0 dvd Z1 G d de2 Z3 G d de+j# Z4 G d de+j# Z5 G d de+j# Z6 G d de# Z7 G d d e+j# Z8e' G d d e"e# Z9 G d d!e# Z:e;d"k r e+j< dS dS )# Nnjit)types)compile_isolatedFlags)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_targetTlinuxzlinux only test)i386x86_64z
x86 only testc e Zd ZdZd Zd ZdS )Dummyr c @ t | xj dz
c_ d S N typealiveselfs J/granary/edev/quantum/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 )Nr1 )r1 r1 )r1 )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_fail9 s
"*$)
"--//
#
#JD$
#&&|44 "'EK --BBII
@
c3v'7#8#8999 " d """"
#
#s. B0'A." B0.A2 2B05A2 6-B00B>N)r. r/ r0 __doc___numba_parallel_test_rF r1 r+ r( r3 r3 2 s5 "# # # # #r+ r3 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_contextsuperrJ setUpr' __class__s r( rO zTestNrtMemInfo.setUpQ 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 r6 refcountacquirerelease)r' daddrmis r( test_meminfo_refct_1z#TestNrtMemInfo.test_meminfo_refct_1X 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" rS d e r ) r rU r% r r6 rV rangerW rX )r' rY rZ r[ ct_s r( test_meminfo_refct_2z#TestNrtMemInfo.test_meminfo_refct_2i 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" rS r )r rU r% r r6 rV
memoryviewdata assertFalsereadonlyassertIsobj
assertTrue c_contiguousitemsizendim)r' rY rZ r[ mviews r( test_fake_memoryviewz#TestNrtMemInfo.test_fake_memoryview| 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" dtypeshapebufferrT ) ctypesrs rt ru rv npr| uint32rm r r7 rf r` rU contentsre nbytesrg rh ri rj rk rl rn ndarraytesting assert_equalarangesize)
r' rs rt ru rv r| bytesizer[ rZ c_arriro arrs
r( test_memoryviewz TestNrtMemInfo.test_memoryview s7 <<<<<<<<<<<<##>B&
5
5
5wXXbg&&2
(>(>??r < B&
5
5
5 a(((wXXd^^WWX]%;%;<<r < r? )r' r predrC s r( test_allocate_invalid_sizez)TestNrtMemInfo.test_allocate_invalid_size s |"a'1,! 1 1D"";//
56#Dt4444
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5 MMITIII f .//
1
1
1
1 1 1s AA A c d}dD ]x}| t 5 }t j || d d d n # 1 swxY w Y d| d}| |t |j yd S )Nir rx z,Cannot allocate a negative number of bytes: .)r9
ValueErrorr r7 r= r> r? )r' r r rC rE s r( test_allocate_negative_sizez*TestNrtMemInfo.test_allocate_negative_size s ! 6 6D"":..
5&#Dt4444
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5HHHHC MM#s6#3 4 4 5 5 5 5 6 6s AA A )r. r/ r0 rG rO r\ rc rp r r r r r
__classcell__rQ s @r( rJ rJ L s , , , , ,) ) )") ) )&) ) ),$ $ $R H 1 1 ]16 6 6 6 6 6 6r+ rJ c e Zd ZdZd Zd ZdS )TestTracemalloczB
Test NRT-allocated memory can be tracked by tracemalloc.
c n dd l }n%# t $ r | d Y nw xY w| | } | }| }~| |d | S # | w xY w)Nr ztracemalloc not availablelineno) tracemalloc ImportErrorskipTeststart
take_snapshot
compare_tostop)r' funcr beforeresafters r( measure_memory_diffz#TestTracemalloc.measure_memory_diff s 7 7 7 7 MM5 6 6 6 6 6 7 ..00F$&&C --//E##FH55 K s ))AB B4c z dt j t fd fd}fd}|j j dz } | | }|d }| |j | |j dz d |j d }|
t j
|j d
|
|j | | | }|d }| |j d z d S ) Ni@B c . t j S )z<
Allocate and return a large array.
)r empty)Nr| s r( alloc_nrt_memoryz7TestTracemalloc.test_snapshot..alloc_nrt_memory
s
8Au%% %r+ c S Nr1 r s r( keep_memoryz2TestTracemalloc.test_snapshot..keep_memory s ##%% %r+ c d S r r1 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)rE z test_nrt.pyg{Gz?)r int8r __code__co_firstlinenor assertGreaterEqualr
assertLess tracebackrU ospathbasenamefilenamer )
r' r r alloc_linenodiffstatframer r r| 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/ r0 rG r r r1 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 )NrT )mathsqrt)vectorxys r( calculate_2D_vector_magzNTestNRTIssue.test_issue_with_refct_op_pruning..calculate_2D_vector_mag9 s' DAq9Q!Va1f_-- -r+ c t j dt j } | }| \ }}||z |d< ||z |d< |S NrT r| 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_vectorsK s !#;*:"*!M!M!M ;''
@
@ $7$7$?$?!+.fc s "(3333 3r+ r1 r" Cy r )
r r Tuple
complex128Arrayr entry_pointassertPreciseEqualr r )r' r creszr s r( test_incref_after_castz#TestNRTIssue.test_incref_after_cast` s 4 4 4
2 % U-=-2[a-M-M-/ !0 !0" "
!!##3 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 )Nrw r )r onesr )abs r( r z5TestNRTIssue.test_refct_pruning_issue_1511..fr s? "*---A"*---Aad7Nr+ rw r )r r r r r r )r' r r r s r( test_refct_pruning_issue_1511z*TestNRTIssue.test_refct_pruning_issue_1511q 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..udt} 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 rU )r' r r expectr s r( test_refct_pruning_issue_1526z*TestNRTIssue.test_refct_pruning_issue_1526| 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{ r1 r1 r+ r( fooz8TestNRTIssue.test_no_nrt_on_njit_decoration..foo s 3r+ )numbar rg r r; rU r rk )r' r r s r( test_no_nrt_on_njit_decorationz+TestNRTIssue.test_no_nrt_on_njit_decoration s
%%%
%%%
...
$$$$$r+ N)
r. r/ r0 r r r r r run_test_in_subprocessr
r1 r+ r( r r 4 sv )6 )6 )6VB 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_increfr1 .0lns r(
zDTestRefCtPruning.test_refct_pruning_op_recognize.. " IIIlb6H6H"6H6H6Hr+ c g | ]}d |v | S
NRT_decrefr1 r s r( r zDTestRefCtPruning.test_refct_pruning_op_recognize.. r r+ c g | ]}d |v | S r r1 r s r( r zDTestRefCtPruning.test_refct_pruning_op_recognize.. " IIIlb6H6H6H6H6Hr+ c g | ]}d |v | S r r1 r s r( r zDTestRefCtPruning.test_refct_pruning_op_recognize.. r r+
)r" rT z ; GONE {}) sample_llvm_irlist
splitlinesr _remove_redundant_nrt_refctassertNotEqualsetrU 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 r< r> s r( _appendzBTestRefCtPruning.test_refct_pruning_with_branches.._append se 8E??
'8C1I&& ' %C1III1III&IIIII
'
'r+ c @ | ||| | ||| d S r r1 )r r r? r@ rC rA 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 jrE 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% refinditerrU r )r' rI llvmirrefopsrC rA rE 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)cgctxbuildersigrB arg_0arg_1ftymuls r( codegenz>TestRefCtPruning.test_inline_asm..bar..codegenF s !%objnnrz"~~79z"~~7G H H kk#'@)#(%.y.3 " 5 5 *%#/3OOOr+ )r r r )tyctxr r rd s r( barz-TestRefCtPruning.test_inline_asm..barD s3
P
P
P U[%+u{CCWL Lr+ z int32(int32)c * | dz
} | d }|S )Nr" rT r1 )r r rf s r( r z-TestRefCtPruning.test_inline_asm..fooP s
FAAq AHr+ rw N)llvmrL initialize_native_target initialize_native_asmprinterinitialize_native_asmparserr r rU )r' r rf s @r( test_inline_asmz TestRefCtPruning.test_inline_asm: s
%''' )+++ (*** M M
M
~
R"%%%%%r+ N) r. r/ r0 r$ r9 unittestskiprQ
linux_onlyx86_onlyrm r1 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 rM rN rs rO rP s r( rO zTestNrtExternalCFFI.setUp] 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)
cffirw
set_sourcer
cdefr r+ rJ sysr rE r remove)r' rW sourcer| rw ffiry mods r( compile_cffi_modulez'TestNrtExternalCFFI.compile_cffi_moduleb s cee tV<>>2BCCC 5 5d ; ;<< 6 """ $ &&C HOOF # # # #CHOOF # # # #Cxs #C !C3c ddl m} | }| dt j d } | }|S )Nr rv z void* (*)()get_api)rz rw rv r
c_helpers)r' rw r nrt_get_apitables r( get_nrt_api_tablez%TestNrtExternalCFFI.get_nrt_api_tabler 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+ rQ r. r rU libstatusr test_nrt_apiintrv r MemInforV )
r' rW r r| r r r outmi_addrr[ s
r( test_manage_memoryz&TestNrtExternalCFFI.test_manage_memoryz 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} r| r~ )i
i
i
)r+ rQ r. r r r r| intprm r r r rv r r rU rV r~ rf r r r )
r' rW r r| r r r numbytesr r r[ 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 rv c t S r r r1 r+ r( r z6TestNrtExternalCFFI.test_get_api..test_nrt_api s == r+ r )rz rw r r rv r rU )r' rw r r r r s r( test_get_apiz TestNrtExternalCFFI.test_get_api s
! !
! ceeSXXh(>(>(@(@AABBlnn %%%%%r+ ) r. r/ r0 rG rO r r r r r r r s @r( rs rs Y s 1 1 1 1 1
'1 '1 '1R < < <@
&
&
&
&
&
&
&r+ rs 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( rO 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+ rT ) r r
r rk r r` r get_allocation_statsr rg rU 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 rg r r9 r: r r r= r> r? )r' rC s r( 5test_rtsys_stats_query_raises_exception_when_disabledzGTestNrtStatistics.test_rtsys_stats_query_raises_exception_when_disabledC 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 rg r r9 r: r= r> r? )r' method_variationsrA r rC s r( r s
7 7 7 7 7 7 7 7 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 + * * * * * 577
X_ !8!8!A!A A .0 0
8?+8+--5GG* , , F # # # # #8#4 # # #4e6 e6 e6 e6 e6X& e6 e6 e6P=- =- =- =- =-h' =- =- =-@u% u% u% u% u%8 u% u% u%pj& 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+