xKg1/ddlZddlZddlZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZddlmZmZddlmZmZmZddlmZddlmZddlmZddlZdZd ZGd d eZd ej@fgZ!d Z"dZ#dZ$dZ%GddeZ&Gdde&Z'Gdde'Z(Gdde&Z)Gdde&Z*Gdde'Z+GddeZ,e-dk(rej\yy) N)TestCaseoverride_configoverride_env_configcaptured_stdoutforbid_codegenskip_parfors_unsupported needs_blas)jitnjit)typescompilerutils)NumbaPerformanceWarning)prange)jitclassc|dz}|S)N)someargretvals Z/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/tests/test_debug.pysimple_nopythonrs q[F Mc#K||ywNr)xys r simple_genrs G Gs ceZdZdZy) SimpleClasscd|_y)Nh)selfs r__init__zSimpleClass.__init__s rN)__name__ __module__ __qualname__r&rrrr r srr r$c|jSrr#)objs rsimple_class_userr,#s 55Lrc.tj||Sr)npdot)abs runsupported_parforr2&s 66!Q<rctj|}t|D] }||tj|z||<"|Srr.onesrsinnr0is rsupported_parforr:)s<  A AYtbffQi! Hrctj|}t|D]*}||tj|z||<|dzdkr*J|S)N ir4r7s runsupported_pranger=/sN  A AYtbffQi!2v Hrc\eZdZegdZdZdZdZdZdZ dZ dZ d Z d Z d Zy ) DebugTestBase)bytecodecfgir typeinferllvm func_opt_llvmoptimized_llvmassemblyc8|jtg|i|yr) assertRaisesAssertionError)r%argskwargss r assert_failszDebugTestBase.assert_fails<s.:4:6:rctj|jd}|D] }||vsJd||<t|j D]1\}}t |d|z}|r || |j ||3y)NFTz_check_dump_%s)dictfromkeys all_dumpssorteditemsgetattrrM)r%out dump_names enabled_dumpsnameenabled check_meths rcheck_debug_outputz DebugTestBase.check_debug_output?s dnne< D=( (("&M$ $M$7$7$9:MD' '7$'>?J3!!*c2 ;rctjdvr|jd|ytjdvr|jd|yttj)N)) )r] BINARY_OP))r] )r] BINARY_ADD)r PYVERSIONassertInNotImplementedErrorr%rUs r_check_dump_bytecodez"DebugTestBase._check_dump_bytecodeKsG ??0 0 MM+s + __ 1 1 MM, ,%eoo6 6rc(|jd|y)NzCFG dominatorsrergs r_check_dump_cfgzDebugTestBase._check_dump_cfgSs &,rcB|jd|jz|y)Nz--IR DUMP: %s--re func_namergs r_check_dump_irzDebugTestBase._check_dump_irVs '$..8#>rc(|jd|y)Nz --propagate--rjrgs r_check_dump_typeinferz#DebugTestBase._check_dump_typeinferYs os+rc|jd|tjjdjj dk(r|j |d|yy)Nz --LLVM DUMP auto_parallelFz#store i64 %\"\.\d", i64\* %"retptr")rer FlagsoptionsdefaultrY assertRegexrgs r_check_dump_llvmzDebugTestBase._check_dump_llvm\sL mS) >> ! !/ 2 : : B Be K   S"H# N Lrcf|jd|jz||jd|y)Nz--FUNCTION OPTIMIZED DUMP %sadd nsw i64 %arg.somearg, 1rmrgs r_check_dump_func_opt_llvmz'DebugTestBase._check_dump_func_opt_llvmas) 4t~~EsK 3S9rcf|jd|jz||jd|y)Nz--OPTIMIZED DUMP %srzrmrgs r_check_dump_optimized_llvmz(DebugTestBase._check_dump_optimized_llvmfs) +dnn||~ ||~s AA;;Bctdd5|j}ddd|jdg|jd|jz|t j d}|j||y#1swYcxYw)NrTrBz--GENERATOR INFO: %sz generator state variables: ['x', 'y'] yield point #1: live variables = ['y'], weak live variables = ['x'] yield point #2: live variables = [], weak live variables = ['y'] )rrr[rerntextwrapdedent)r%rUexpected_gen_infos rtest_dump_ir_generatorz/TestGeneratorDebugOutput.test_dump_ir_generatorsw Y -))+C. dV, ,t~~=sC$OO- '-. -s BB N)r'r(r)rnrrrrrrrsI .rrceZdZdZdZdZy)TestDisableJITz: Test the NUMBA_DISABLE_JIT environment variable. ctdd5t5tdt}|j |ddddddddy#1swYxYw#1swYyxYw)N DISABLE_JITTnopythonrr])rrr rr)r%rs rtest_jitzTestDisableJIT.test_jitsT ]D 1!*T*?;''a!4"2 1!!2 1s" A$/AA$A! A$$A-cXtdd5t5ttt}|}|j |j dtdt}|j ||dddddddy#1swYxYw#1swYyxYw)NrTr"r) rrrsimple_class_specr rr$r r,)r%SimpleJITClassr+rs r test_jitclasszTestDisableJIT.test_jitclasss ]D 1!!<*;!<[!I$&''q1*T*+<=''c A6"2 1!!2 1s# B A+BB B B  B)N)r'r(r)__doc__rrrrrrrs5 7rrceZdZdZdZdZy)TestEnvironmentOverridezN Test that environment variables are reloaded by Numba when modified. Fc|j}|j|tdd5|j}|j|gdddd|j}|j|y#1swY+xYw)N NUMBA_DEBUG1)rBrCrDrErFrG)r assertFalserr[rgs r test_debugz"TestEnvironmentOverride.test_debugs{**,   4..0C  # #C*H I 5**, 5 4s %A==BN)r'r(r)r_numba_parallel_test_rrrrrrs " rrcdeZdZdZdZdZdZeedZ eedZ edZ y) TestParforsDebugz5 Tests debug options associated with parfors Fctd}d}|D]}|t|jvsd}n|j|dy)NzX'parallel=True' was specified but no transformation for parallel execution was possible.FT Warning message should be found.strmessage assertTruer% warn_listmsg warning_foundws rcheck_parfors_warningz&TestParforsDebug.check_parfors_warningsA2 Ac!))n$ $   'IJrctd}d}|D]}|t|jvsd}n|j|dy)Nzprange or pndindex loop will not be executed in parallel due to there being more than one entry to or exit from the loop (e.g., an assertion).FTrrrs r(check_parfors_unsupported_prange_warningz9TestParforsDebug.check_parfors_unsupported_prange_warningsC, Ac!))n$ $   'IJrc,tjtjdd}tjd5}tj dt t||fdtddd|jy#1swYxYw)zr Test that using parallel=True on a function that does not have parallel semantics warns. rCTrecordalwaysparallelN) r Arrayfloat64warningscatch_warnings simplefilterrr r2r)r%arr_tyrs r test_warnszTestParforsDebug.test_warnsspU]]As3  $ $D 1Q  ! !(,C D 1D&&!D 12D E2 ""1%2 1s 4B  Bctjd5}tjdtt t j fdtddd|jy#1swYxYw)zG Test that prange with multiple exits issues a warning TrrrN) rrrrr r rr=r)r%rs rtest_unsupported_prange_warnsz.TestParforsDebug.test_unsupported_prange_warnss[  $ $D 1Q  ! !(,C D /D%++$ /0B C2 55a82 1s AA22A;cttjfdtt dd5t 5}ttjfdtdddj jd}|Dcgc] }d|vs| c}}|Dcgc] }d|vs| c}}|Dcgc] }d |vs| c}}ttjd j|d jd }t||t|z}d } d} t||| D]*\} } } | j!| | }|j#|| ,|d z|d zf} d} |D]#} | j | }|j#|| %tj$d t'|gf} d} |D]#} | j | }|j#|| % dddy#1swYxYwcc}wcc}wcc}w#1swYyxYw)zM Test that NUMBA_DEBUG_ARRAY_OPT_STATS produces valid output TrNUMBA_DEBUG_ARRAY_OPT_STATSrN zis produced from patternz is fused intozAfter fusion, functionz #([0-9]+)rr)z"('ones function', 'NumPy mapping'))ruserz6Parallel for-loop #{} is produced from pattern '{}' atz1Parallel for-loop #{} is fused into for-loop #{}.z:After fusion, function {} has {} parallel for-loop(s) #{}.)r r rr:rrrsplitintrecompilesearchgrouprangelenzipformatrer'r)r%rUoutputrparallel_loop_output fuse_outputafter_fusion_output parfor_stateboundspatternfmtr9trialslpatternto_matchs rtest_array_debug_opt_statsz+TestParforsDebug.test_array_debug_opt_statss  ,ekk^d+,<= !> D "c3ekk^d34DE#\\^))$/F"FFq&@A&EFF !#;Fqo&:F; #DFq&>!&CFD rzz,7>>$Q' )).q3L<'#.B*CCEF /GLC'*63G+2(4#68::a2 h/(4 $a')9:GEC%%3::w/ h/& (00!S,5HIGNC-%3::w/ h/.QE D"" G;DE DsY H'G%%+H G2G2H% G7/G73H: G<G<DH%G/ *HH N) r'r(r)rrrrr rrrrrrrrrsg "K K & &994040rr__main__)/osrrrrnumpyr.numba.tests.supportrrrrrrr numbar r numba.corer r rnumba.core.errorsrrnumba.experimentalrunittestrrobjectr int32rr,r2r:r=r?rrrrrrr'mainrrrrs  """--5' &5;;'(  6'H6'r M #33#3L.}.07]7.3*j0xj0Z zHMMOr