o i3/@sxddlZddlZddlZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZddlmZmZddlmZmZmZddlmZddlmZddlmZddlZdd Zd d ZGd d d eZdej fgZ!ddZ"ddZ#ddZ$ddZ%GdddeZ&Gddde&Z'Gddde'Z(Gddde&Z)Gdd d e&Z*Gd!d"d"e'Z+Gd#d$d$eZ,e-d%kre.dSdS)&N)TestCaseoverride_configoverride_env_configcaptured_stdoutforbid_codegenskip_parfors_unsupported needs_blas)jitnjit)typescompilerutils)NumbaPerformanceWarning)prange)jitclasscCs |d}|S)N)someargretvalrrS/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/numba/tests/test_debug.pysimple_nopythonsrccs|V|VdSNr)xyrrr simple_gens rc@seZdZddZdS) SimpleClasscCs d|_dS)Nh)selfrrr__init__s zSimpleClass.__init__N)__name__ __module__ __qualname__r rrrrrs rrcCs|jSrr)objrrrsimple_class_user#sr%cCs t||Sr)npdot)abrrrunsupported_parfor&s r*cCs2t|}t|D] }||t|||<q |Srr&onesrsinnr(irrrsupported_parfor)s  r1cCsBt|}t|D]}||t|||<|ddksJq |S)N ir+r.rrrunsupported_prange/s  r3c@sheZdZegdZddZddZddZdd Zd d Z d d Z ddZ ddZ ddZ ddZdS) DebugTestBase)bytecodecfgir typeinferllvm func_opt_llvmoptimized_llvmassemblycOs|jtg|Ri|dSr) assertRaisesAssertionError)rargskwargsrrr assert_fails<szDebugTestBase.assert_failscCsnt|jd}|D] }||vsJd||<q t|D]\}}t|d|}|r.||q|||qdS)NFTz_check_dump_%s)dictfromkeys all_dumpssorteditemsgetattrrA)rout dump_names enabled_dumpsnameenabled check_methrrrcheck_debug_output?s   z DebugTestBase.check_debug_outputcCs>tjdvr |d|dStjdvr|d|dSttj)N)) )rO )rOr2 BINARY_OP))rO BINARY_ADD)r PYVERSIONassertInNotImplementedErrorrrHrrr_check_dump_bytecodeKs   z"DebugTestBase._check_dump_bytecodecC|d|dS)NzCFG dominatorsrVrXrrr_check_dump_cfgSzDebugTestBase._check_dump_cfgcCs|d|j|dS)Nz--IR DUMP: %s--rV func_namerXrrr_check_dump_irVszDebugTestBase._check_dump_ircCrZ)Nz --propagate--r[rXrrr_check_dump_typeinferYr]z#DebugTestBase._check_dump_typeinfercCs6|d|tjjdjjdkr||d|dSdS)Nz --LLVM DUMP auto_parallelFz#store i64 %\"\.\d", i64\* %"retptr")rVr FlagsoptionsdefaultrL assertRegexrXrrr_check_dump_llvm\s zDebugTestBase._check_dump_llvmcC"|d|j||d|dS)Nz--FUNCTION OPTIMIZED DUMP %sadd nsw i64 %arg.somearg, 1r^rXrrr_check_dump_func_opt_llvmasz'DebugTestBase._check_dump_func_opt_llvmcCrh)Nz--OPTIMIZED DUMP %srir^rXrrr_check_dump_optimized_llvmfsz(DebugTestBase._check_dump_optimized_llvmcCs2|d|j|tdvr|d|dSdS)Nz --ASSEMBLY %s)x86_64AMD64i386i686xorl)rVr_platformmachinerXrrr_check_dump_assemblyjs z"DebugTestBase._check_dump_assemblyN)r!r"r#setrDrArNrYr\r`rargrjrkrsrrrrr47s   r4c@seZdZdZddZdS)FunctionDebugTestBasercCsTt}ttjft}||ddWd|S1s!wY|S)NrO)rr r int64rassertPreciseEqualgetvaluerrHcfuncrrrcompile_simple_nopythonts z-FunctionDebugTestBase.compile_simple_nopythonN)r!r"r#r_r|rrrrrups ruc@sDeZdZddZddZddZddZd d Zd d Zd dZ dS)TestFunctionDebugOutputcCDtdd |}Wdn1swY||dgdS)N DUMP_BYTECODETr5rr|rNrXrrrtest_dump_bytecode~  z*TestFunctionDebugOutput.test_dump_bytecodecCr~)NDUMP_IRTr7rrXrrr test_dump_irrz$TestFunctionDebugOutput.test_dump_ircCr~)NDUMP_CFGTr6rrXrrr test_dump_cfgrz%TestFunctionDebugOutput.test_dump_cfgcCr~)N DUMP_LLVMTr9rrXrrrtest_dump_llvmrz&TestFunctionDebugOutput.test_dump_llvmcCr~)N DUMP_FUNC_OPTTr:rrXrrrtest_dump_func_opt_llvmrz/TestFunctionDebugOutput.test_dump_func_opt_llvmcCr~)NDUMP_OPTIMIZEDTr;rrXrrrtest_dump_optimized_llvmrz0TestFunctionDebugOutput.test_dump_optimized_llvmcCr~)N DUMP_ASSEMBLYTr<rrXrrrtest_dump_assemblyrz*TestFunctionDebugOutput.test_dump_assemblyN) r!r"r#rrrrrrrrrrrr}|s r}c@ eZdZdZddZddZdS)TestGeneratorDebugOutputrcCsbt"}ttjtjft}|t|ddddgWd|S1s(wY|S)Nrvr)rr r rwrrxlistryrzrrrcompile_simple_gens z+TestGeneratorDebugOutput.compile_simple_gencCsltdd |}Wdn1swY||dg|d|j|td}|||dS)NrTr7z--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'] )rrrNrVr_textwrapdedent)rrHexpected_gen_inforrrtest_dump_ir_generators   z/TestGeneratorDebugOutput.test_dump_ir_generatorN)r!r"r#r_rrrrrrrs rc@r)TestDisableJITz: Test the NUMBA_DISABLE_JIT environment variable. c Cstdd3ttddt}||ddWdn1s#wYWddSWddS1s;wYdS)N DISABLE_JITTnopythonrvrO)rrr rrx)rr{rrrtest_jits "zTestDisableJIT.test_jitc CstddCt'ttt}|}||jdtddt}|||dWdn1s3wYWddSWddS1sKwYdS)NrTrr) rrrsimple_class_specrrxrr r%)rSimpleJITClassr$r{rrr test_jitclasss  "zTestDisableJIT.test_jitclassN)r!r"r#__doc__rrrrrrrs rc@seZdZdZdZddZdS)TestEnvironmentOverridezN Test that environment variables are reloaded by Numba when modified. FcCsj|}||tdd|}||gdWdn1s%wY|}||dS)N NUMBA_DEBUG1)r7r8r9r:r;r<)r| assertFalserrNrXrrr test_debugs  z"TestEnvironmentOverride.test_debugN)r!r"r#r_numba_parallel_test_rrrrrrs rc@sPeZdZdZdZddZddZeeddZ eed d Z ed d Z d S)TestParforsDebugz5 Tests debug options associated with parfors FcC8d}d}|D] }|t|jvrd}nq||ddS)NzX'parallel=True' was specified but no transformation for parallel execution was possible.FT Warning message should be found.strmessage assertTruer warn_listmsg warning_foundwrrrcheck_parfors_warningsz&TestParforsDebug.check_parfors_warningcCr)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).FTrrrrrr(check_parfors_unsupported_prange_warningsz9TestParforsDebug.check_parfors_unsupported_prange_warningcCsjttjdd}tjdd}tdtt||fddtWdn1s)wY| |dS)zr Test that using parallel=True on a function that does not have parallel semantics warns. rvCTrecordalwaysparallelN) r Arrayfloat64warningscatch_warnings simplefilterrr r*r)rarr_tyrrrr test_warnss  zTestParforsDebug.test_warnscCsZtjdd}tdtttjfddtWdn1s!wY||dS)zG Test that prange with multiple exits issues a warning TrrrN) rrrrr r rwr3r)rrrrrtest_unsupported_prange_warnss  z.TestParforsDebug.test_unsupported_prange_warnsc Csttjfddttddt}ttjfddtWdn1s(wY|d}dd|D}d d|D}d d|D}tt d  |d  d }t ||t|}d}d} t|||D]\} } } | | | } || | qk|d |d f}d} |D] } | j|} || | qtjd t|gf}d} |D] } | j|} || | qWddS1swYdS)zM Test that NUMBA_DEBUG_ARRAY_OPT_STATS produces valid output TrNUMBA_DEBUG_ARRAY_OPT_STATSrN cSg|]}d|vr|qS)zis produced from patternr.0rrrr -z?TestParforsDebug.test_array_debug_opt_stats..cSr)z is fused intorrrrrr/rcSr)zAfter fusion, functionrrrrrr1rz #([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 rwr1rrrysplitintrecompilesearchgrouprangelenzipformatrVr!rt)rrHoutputparallel_loop_output fuse_outputafter_fusion_output parfor_stateboundspatternfmtr0trialslpatternto_matchrrrtest_array_debug_opt_statssN         "z+TestParforsDebug.test_array_debug_opt_statsN) r!r"r#rrrrrrrrrrrrrrs      r__main__)/osrqrrrnumpyr&numba.tests.supportrrrrrrrnumbar r numba.corer r r numba.core.errorsrrnumba.experimentalrunittestrrobjectrint32rr%r*r1r3r4rur}rrrrr!mainrrrrs>$    9 &m