g<2ddlZddlZddlZddlZddlZddlmZddlZddlm Z m Z ddl m Z m Z ddlmZmZmZmZmZddlmZddlmZddlmZmZdd lmZmZmZeZd e_ eZ!d e!_"eZ#Gd d e$Z%Gd de%Z&Gdde$Z'dZ(dZ)dZ*dZ+dZ,dZ-Gdde.Z/dZ0dZ1dZ2dZ3dZ4dZ5Gdd eZ6Gd!d"eeZ7e8d#krej9dSdS)$N) takewhile)njitjit)CFGraphLoop) compile_extracompile_isolatedFlags CompilerBaseDefaultPassBuilder) PreserveIR)IRLegalization)typesir)TestCaseoverride_config SerialMixinTc eZdZdZdZdZdS)_DummycL||_||_||dSN)recordername _add_dummy)selfrrs S/granary/edev/quantum/lib/python3.11/site-packages/numba/tests/test_obj_lifetime.py__init__z_Dummy.__init__s*   D!!!!!cxt|tsJt|j|jd|jS)Nz + ) isinstancerrr)rothers r__add__z_Dummy.__add__"s9%(((((dm$)))UZZ%HIIIrc<t|jd|jzS)Nziter(%s))_DummyIteratorrrrs r__iter__z_Dummy.__iter__&sdmZ$)-CDDDrN)__name__ __module__ __qualname__rr"r&rrrrsI""" JJJEEEEErrceZdZdZdZeZdS)r$rc|jdkrt|xjdz c_t|j|jd|jS)N#)count StopIterationrrrr%s r__next__z_DummyIterator.__next__.sC :??  a dm 4::%FGGGrN)r'r(r)r0r2nextr*rrr$r$*s, EHHH DDDrr$c`eZdZdZdZdZdZeZdZdZ e dZ e dZ d S) RefRecorderz An object which records events when instances created through it are deleted. Custom events can also be recorded to aid in diagnosis. c^tjt|_g|_i|_dSr) collections defaultdictint_counts_events_wrsr%s rrzRefRecorder.__init__>s&".s33   rc"t||S)zK Make an object whose deletion will be recorded as *name*. )r)rrs r make_dummyzRefRecorder.make_dummyCsdD!!!rcXtj||j}|j|j|<dSr)weakrefref _on_disposalrr<)rdummywrs rrzRefRecorder._add_dummyIs' [ 1 2 2  " rc|j|dz}||j|<|j||dS)zr Manually append *event* to the recorded events. *event* can be formatted using format(). r.)r0N)r:r;appendformat)reventr0s rmarkzRefRecorder.markOsK  U#a'# U ELLuL5566666rcn|j|}|j|dSr)r<popr;rF)rrDrs rrBzRefRecorder._on_disposalXs1y}}R   D!!!!!rc$d|jDS)zC A list of objects which haven't been deleted yet. c"g|] }| Sr*r*).0rDs r z%RefRecorder.alive..as))))))r)r<r%s ralivezRefRecorder.alive\s *)ty))))rc|jS)z, A list of recorded events. )r;r%s rrecordedzRefRecorder.recordedcs |rN) r'r(r)__doc__rr>r__call__rIrBpropertyrPrRr*rrr5r57s  """ ###H777"""**X* Xrr5c|d}|d}|d}||z}|d||z}|d|S)Nabc--1----2--rI)recrWrXrYds rsimple_usecase1r_ksd CA CA CA AAHHW AAHHW Hrch|d}|d}|d|}|}d}|S)NrWrXrZr\)r]rWrXxys rsimple_usecase2rcusB CA CAHHW A A A Hrc|d}|d}|d}|}|D]}||z}|d|d||z}|S)NrWrXrY--loop bottom-- --loop exit--r\)r]rWrXrYrarbs rlooping_usecase1rg~s| CA CA CA A $$ E "####HH_ AA Hrc|d}|d}|d}|D]}|d||z}||z}|d|D]1}|d||z}|d2|d|r||z}nn+|d |d |d |S) NrWrXcum--outer loop top--z--inner loop entry #{count}--z--inner loop top #{count}--z--inner loop bottom #{count}--z--inner loop exit #{count}--z--outer loop bottom #{count}----outer loop else----outer loop exit--r\)r]rWrXrirazrbs rlooping_usecase2rns CA CA #e**C (( %&&&Ag E 0111 7 7A HH2 3 3 3'C HH5 6 6 6 6 /000  a%CC E 12222 &'''HH "### Jrc#FK|d}|d}|V|VdSNrWrXr*r]rWrXs rgenerator_usecase1rrs6 CA CA GGG GGGGGrc#PK|d}|d}|D]}|V|VdSrpr*)r]rWrXras rgenerator_usecase2rtsG CA CA  GGGGGrceZdZdS)MyErrorN)r'r(r)r*rrrvrvsDrrvc t|rrv)ras rdo_raiserys !**rc|d}|d}|d}|rtd|d}||z||zdS)NrWrXr^foorYry)r]rWrXr^rYs rraising_usecase1r}s_ CA CA CA CHH AEEEErct|d}|d}|r|d}t|||zdS)NrWrXrYr|)r]rWrXrYs rraising_usecase2rsI CA CA CHH EEEErcT|d}|d}|rt|dSrprxrqs rraising_usecase3rs: CA CAajjrcd}t|D]}|t|d}t|D]}dS|dkrB|dkr,t|D]}|ccS|d|d|d|ddS)zN This test reveal a bug that there is a del on uninitialized variable rFAILEDOK)rangerIstr)r]nijs rdel_before_definitionrs A 1XX   Q q  A1111uuaqAHHHHH""" HHX    HHTNNN 2rc#K |ddV|d}|r|d =)z^ test to reveal bug of invalid liveness when infinite loop has multiple backedge. TyieldNpbrar\)r]rs rinf_loop_multiple_back_edgersR    CHH   HHUOOO  rcneZdZdZdZddZdZdZdZdZ d Z d Z d Z d Z d ZdZdZdZdZdS)TestObjLifetimezH Test lifetime of Python objects inside jit-compiled functions. cRttjfdd|}|S)NTF)forceobjlooplift)rrpyobject)rpyfunccfuncs rcompilezTestObjLifetime.compiles-FU^%uEEEfMM rNct}||}|9||5||dddn #1swxYwYn |||Sr)r5r assertRaises)rrraisesr]rs rcompile_and_recordz"TestObjLifetime.compile_and_recordsmm V$$  ""6**  c                 E#JJJ s AAAcg}|j}t|}|D]0}||vr*||||1|||d|dS)zk Check that the *expected* markers occur in that order in *rec*'s recorded events. z%the full list of recorded events is: N)rRlistrFremove assertEqual)rr]expectedactualrR remainingr^s rassertRecordOrderz!TestObjLifetime.assertRecordOrder s <NN  $ $AI~~ a     ### FNhP R R R R Rrc|t}||j||gd||gd||gddS)N)rWrXrZ)rWrYrZ)rZzb + cr[)rr_ assertFalserPrrr]s r test_simple1zTestObjLifetime.test_simple1s%%o66 ### s$7$7$7888 s$7$7$7888 s$?$?$?@@@@@rc|t}||j||gddS)N)rXrZrW)rrcrrPrrs r test_simple2zTestObjLifetime.test_simple2"sN%%o66 ### s$7$7$788888rc|t}||j||gd||gddS)N)rWrXrfrY) iter(a)#1re iter(a)#2re iter(a)#3reiter(a)rf)rrgrrPrrs r test_looping1zTestObjLifetime.test_looping1's|%%&677 ### s$D$D$DEEE s%'%'%' ( ( ( ( (rc^|t}||j||ddg||gd||gd||gd||ddgdS)NrWrj)rrkrl)iter(b)z--inner loop exit #1--rz--inner loop exit #2--rz--inner loop exit #3--)rz--inner loop entry #1--rz--inner loop entry #2--rz--inner loop entry #3--ziter(a)#1 + iter(a)#1z--outer loop bottom #1--)rrnrrPrrs r test_looping2zTestObjLifetime.test_looping23s%%&677 ### sS*>$?@@@ s%<%<%< = = = s%'%'%' ( ( ( s%'%'%' ( ( ( s%<%?%' ( ( ( ( (rc||}t}||5||}t|||jt |||jdddn #1swxYwYt}||5||}~tj ||jdddn #1swxYwYt}||5||}t|||j~tj ||jddddS#1swxYwYdSr) rr5assertRefCountr3 assertTruerPrrgccollect)rgenfuncrr]gens rexercise_generatorz"TestObjLifetime.exercise_generatorIsJ W%%mm   % % ( (%**C III OOCI & & & III   SY ' ' '  ( ( ( ( ( ( ( ( ( ( ( ( ( ( (mm   % % ( (%**C JLLL   SY ' ' '  ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( mm   % % ( (%**C III OOCI & & & JLLL   SY ' ' '  ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s7AB##B'*B':DDDA#F55F9<F9c:|tdSr)rrrr%s rtest_generator1zTestObjLifetime.test_generator1d  233333rc:|tdSr)rrtr%s rtest_generator2zTestObjLifetime.test_generator2grrct|t}||jgddS)N)012)rrrrRrs rtest_del_before_definitionz*TestObjLifetime.test_del_before_definitionjs8%%&;<< 77777rc|t5|tt}||jddddS#1swxYwYdSN)r)rryrr}rvrrPrs r test_raising1zTestObjLifetime.test_raising1n   * * ( ())*:7)KKC   SY ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (.get_ir..IRPreservingCompilerctj|j}|tt ||gSrr define_nopython_pipelinestateadd_pass_afterr rfinalizerpms rdefine_pipelineszqTestExtendingVariableLifetimes.test_lifetime_basic..get_ir..IRPreservingCompiler.define_pipelines>+DTZPPB%%j.AAAKKMMM4KrNr'r(r)rr*rrIRPreservingCompilerr#     rr)pipeline_classc"d}d}||z}||z }|SN r*rWrXrYr^s rr{zOTestExtendingVariableLifetimes.test_lifetime_basic..get_ir..foo#EErEXTEND_VARIABLE_LIFETIMESr preserved_ir)r rr overloads signaturesmetadata)extend_lifetimesrr{cresfunc_irs rget_irzBTestExtendingVariableLifetimes.test_lifetime_basic..get_irs     |   !5 6 6 6  7 6 !!<>NOO 8 8}S^A%67-7 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 Ns0A99A=A=c t|jdtt |j}t ||jD]\}}||dS)Nr.) rlenblocksr3rvalueszipbodyassertIsInstance)rexpectblk expect_classgot_stmtrs rcheckzATestExtendingVariableLifetimes.test_lifetime_basic..checks   S00! 4 4 4tGN11334455C+.fch*?*? > >& h%%h ==== > >rFr-T)rAssignDelReturn)rrrdel_after_use_irrdel_at_block_end_irs` rtest_lifetime_basicz2TestExtendingVariableLifetimes.test_lifetime_basics   4 > > > > >"6%==0RYL1$0rv0rv0ry0"&0)0V0%'Y0 '''$fTllORYL1$NryNRVIMNBIN !6*****rcd}|d}|dd}|dd}d}|||||||||dS)NcGddt}tdd|i|d}||j|jd}|jd}|S)NceZdZdZdS)fTestExtendingVariableLifetimes.test_dbg_extend_lifetimes..get_ir..IRPreservingCompilerctj|j}|tt ||gSrrrs rrzwTestExtendingVariableLifetimes.test_dbg_extend_lifetimes..get_ir..IRPreservingCompiler.define_pipelinesrrNrr*rrrrrrrrc"d}d}||z}||z }|Srr*rs rr{zUTestExtendingVariableLifetimes.test_dbg_extend_lifetimes..get_ir..foorrrrr*)r rrrr)optionsrr{rrs rrzHTestExtendingVariableLifetimes.test_dbg_extend_lifetimes..get_irs     |    A A!5 A A A  B A  CEEE=!23DmN3GNrT)debug)r_dbg_extend_lifetimesFct|j\}d|jD}tt t j|}t|}tt t j||d}|t|z}|t|dz kS)NcBg|]}t|tjSr*)r rr)rNstmts rrOzoTestExtendingVariableLifetimes.test_dbg_extend_lifetimes..is_del_grouped_at_the_end..s$III:dBF33IIIrr.) rrrrroperatornot_rtruth)firr inst_is_delnot_delsbeginall_delsends ris_del_grouped_at_the_endz[TestExtendingVariableLifetimes.test_dbg_extend_lifetimes..is_del_grouped_at_the_endsJ%%''ESIIIIIKIhm[AABBHMMEIhnk%&&6IJJKKH#h--'C#k**Q.. .r)rr)rrir_debug ir_debug_extir_debug_no_extr!s rtest_dbg_extend_lifetimesz8TestExtendingVariableLifetimes.test_dbg_extend_lifetimess   26%%%vDEEE  &t5III / / / 11(;;<<< 11,??@@@ 22?CCDDDDDrN)r'r(r)r r%r*rrrrs8 0+0+0+d3E3E3E3E3Err__main__):r7sysr@rr itertoolsrunittestnumbarrnumba.core.controlflowrrnumba.core.compilerrr r r r numba.core.untyped_passesr numba.core.typed_passesr numba.corerrnumba.tests.supportrrrenable_pyobj_flagsenable_pyobjectforceobj_flagsforce_pyobjectno_pyobj_flagsobjectrr$r5r_rcrgrnrrrt RuntimeErrorrvryr}rrrrrrr'mainr*rrr9sd  00000000CCCCCCCCCCCCCC000000222222 FFFFFFFFFFUWW%)" $ E E E E EV E E E     V   11111&111h          2           l         .   SNSNSNSNSNhSNSNSNljEjEjEjEjE[(jEjEjEZ zHMOOOOOr