ܙdu 4 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 dl m c m!c m"Z# d d
l$m%Z% d d l&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z, d d l-m.Z. d dl/Z/ e
Z0d
e0_ e/j1 ej 2 d d Z3 e/j1 ej4 dvd Z5 G d de6 Z7 G d de/j( Z8 G d de/j( Z9 G d de/j( Z: G d de( Z; G d d e/j( Z< e/j= e#j> d G d d e'e( Z? G d! d"e( Z@eAd#k r e/jB 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_32bitrun_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 4lib/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 )Nr0 )r0 r0 )r0 )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_fail: 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_rE r0 r* r' r2 r2 3 s5 "# # # # #r* r2 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_contextsuperrI setUpr& __class__s r' rN zTestNrtMemInfo.setUpR 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 r5 refcountacquirerelease)r& daddrmis r' test_meminfo_refct_1z#TestNrtMemInfo.test_meminfo_refct_1Y 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! rR d e r ) r rT r$ r r5 rU rangerV rW )r& rX rY rZ ct_s r' test_meminfo_refct_2z#TestNrtMemInfo.test_meminfo_refct_2j 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! rR r )r rT r$ r r5 rU
memoryviewdata assertFalsereadonlyassertIsobj
assertTrue c_contiguousitemsizendim)r& rX rY rZ 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! dtypeshapebufferrS ) ctypesrr rs rt ru npr{ uint32rl r r6 re r_ rT contentsrd nbytesrf rg rh ri rj rk rm ndarraytesting assert_equalarangesize)
r& rr rs rt ru r{ bytesizerZ rY c_arrirn arrs
r' test_memoryviewz TestNrtMemInfo.test_memoryview s7 <<<<<<<<<<<<##>B&
5
5
5wXXbg&&2
(>(>??r < B&
5
5
5 a(((wXXd^^WWX]%;%;<<r < )r& r predrB 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 rw z,Cannot allocate a negative number of bytes: .)r8
ValueErrorr r6 r< r= r> )r& r r rB rD 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. r/ rF rN r[ rb ro r r r r r
__classcell__rP s @r' rI rI M s , , , , ,) ) )") ) )&) ) ),$ $ $R H 1 1 ]16 6 6 6 6 6 6r* rI 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 Nr0 r s r' keep_memoryz2TestTracemalloc.test_snapshot..keep_memory s ##%% %r* c d S r r0 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)rD z test_nrt.pyg{Gz?)r int8r __code__co_firstlinenor assertGreaterEqualr
assertLess tracebackrT 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. r/ rF r r r0 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 )NrS )mathsqrt)vectorxys r' calculate_2D_vector_magzNTestNRTIssue.test_issue_with_refct_op_pruning..calculate_2D_vector_mag: s' DAq9Q!Va1f_-- -r* c t j dt j } | }| \ }}||z |d< ||z |d< |S NrS 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_vectorsL s !#;*:"*!M!M!M ;''
@
@ $7$7$?$?!+.fd s "(3333 3r* r0 r! Cy r )
r r Tuple
complex128Arrayr entry_pointassertPreciseEqualr r )r& r creszr s r' test_incref_after_castz#TestNRTIssue.test_incref_after_casta 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 )Nrv r )r onesr )abs r' r z5TestNRTIssue.test_refct_pruning_issue_1511..fs s? "*---A"*---Aad7Nr* rv r )r r r r r r )r& r r r s r' test_refct_pruning_issue_1511z*TestNRTIssue.test_refct_pruning_issue_1511r 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 rT )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{ r0 r0 r* r' fooz8TestNRTIssue.test_no_nrt_on_njit_decoration..foo s 3r* )numbar rf r r: rT r rj )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 r0 r* r' r r 5 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_increfr0 .0lns r'
zDTestRefCtPruning.test_refct_pruning_op_recognize.. " IIIlb6H6H"6H6H6Hr* c g | ]}d |v | S
NRT_decrefr0 r s r' r zDTestRefCtPruning.test_refct_pruning_op_recognize.. r r* c g | ]}d |v | S r r0 r s r' r zDTestRefCtPruning.test_refct_pruning_op_recognize.. " IIIlb6H6H6H6H6Hr* c g | ]}d |v | S r r0 r s r' r zDTestRefCtPruning.test_refct_pruning_op_recognize.. r r*
)r! rS z ; GONE {}) sample_llvm_irlist
splitlinesr _remove_redundant_nrt_refctassertNotEqualsetrT 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 r0 )r r r> r? rB 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 jrD 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$ refinditerrT r )r& rH llvmirrefopsrB r@ rD 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)cgctxbuildersigrA arg_0arg_1ftymuls r' codegenz>TestRefCtPruning.test_inline_asm..bar..codegenG s !%objnnrz"~~79z"~~7G H H kk#'@)#(%.y.3 " 5 5 *%#/3OOOr* )r r r )tyctxr r rc s r' barz-TestRefCtPruning.test_inline_asm..barE s3
P
P
P U[%+u{CCWL Lr* z int32(int32)c * | dz
} | d }|S )Nr! rS r0 )r r re s r' r
z-TestRefCtPruning.test_inline_asm..fooQ s
FAAq AHr* rv N)llvmrK initialize_native_target initialize_native_asmprinterinitialize_native_asmparserr r rT )r& r
re s @r' test_inline_asmz TestRefCtPruning.test_inline_asm; s
%''' )+++ (*** M M
M
~
R"%%%%%r* N) r- r. r/ r# r8 unittestskiprP
linux_onlyx86_onlyrl r0 r* r' r r s FNP N N N> X]HII") ") JI")H
& & X Z& & &r* r z
cffi requiredc @ 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 rL rM rr rN rO s r' rN 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)
cffirv
set_sourcer
cdefr r* rI sysr rD r remove)r& rV sourcer{ rv ffirx mods r' compile_cffi_modulez'TestNrtExternalCFFI.compile_cffi_modulec s cee tV<>>2BCCC 5 5d ; ;<< 6 """ $ &&C HOOF # # # #CHOOF # # # #Cxs #C !C3c ddl m} | }| dt j d } | }|S )Nr ru z void* (*)()get_api)ry rv ru r
c_helpers)r& rv r nrt_get_apitables r' get_nrt_api_tablez%TestNrtExternalCFFI.get_nrt_api_tables 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* rP r- r rT libstatusr test_nrt_apiintru r MemInforU )
r& rV r~ r{ r r r outmi_addrrZ s
r' test_manage_memoryz&TestNrtExternalCFFI.test_manage_memory{ 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* rP r- r r r r{ intprl r r r ru r r rT rU r} re r r r )
r& rV r~ r{ r r r numbytesr r rZ 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 ru c t S r r r0 r* r' r z6TestNrtExternalCFFI.test_get_api..test_nrt_api s == r* r )ry rv r r ru r rT )r& rv r r r r s r' test_get_apiz TestNrtExternalCFFI.test_get_api s
! !
! ceeSXXh(>(>(@(@AABBlnn %%%%%r* ) r- r. r/ rF rN r r r r r r r s @r' rr rr Z s 1 1 1 1 1
'1 '1 '1R < < <@
&
&
&
&
&
&
&r* rr 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' rN 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* rS ) r r
r rj r r_ r get_allocation_statsr rf rT 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 rf r r8 r9 r r r< r= r> )r& rB s r' 5test_rtsys_stats_query_raises_exception_when_disabledzGTestNrtStatistics.test_rtsys_stats_query_raises_exception_when_disabledD 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 rf r r8 r9 r< r= r> )r& method_variationsr@ r rB 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 3 3 3 3 3 3 3 3 3 3 3 3 - - - - - -4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 + * * * * * 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*