o 0 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)Zsomeargretvalrrb/home/app/PaddleOCR-VL-test/.venv_paddleocr/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__rrrrrrs rrcCs|jSrr)objrrrsimple_class_user#sr$cCs t||Sr)npdot)abrrrunsupported_parfor&s r)cCs2t|}t|D] }||t|||<q |Srr%Zonesrsinnr'irrrsupported_parfor)s  r/cCsBt|}t|D]}||t|||<|ddksJq |S)N ir*r,rrrunsupported_prange/s  r1c@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_dumpssorteditemsgetattrr?)routZ dump_namesZ enabled_dumpsnameenabledZ check_methrrrcheck_debug_output?s   z DebugTestBase.check_debug_outputcCs>tjdvr |d|dStjdvr|d|dSttj)N)) )rJ )rJr0Z BINARY_OP))rJ BINARY_ADD)r Z PYVERSIONassertInNotImplementedErrorrrFrrr_check_dump_bytecodeKs   z"DebugTestBase._check_dump_bytecodecC|d|dS)NzCFG dominatorsrOrQrrr_check_dump_cfgSzDebugTestBase._check_dump_cfgcCs|d|j|dS)Nz--IR DUMP: %s--rO func_namerQrrr_check_dump_irVszDebugTestBase._check_dump_ircCrS)Nz --propagate--rTrQrrr_check_dump_typeinferYrVz#DebugTestBase._check_dump_typeinfercCs6|d|tjjdjjdkr||d|dSdS)Nz --LLVM DUMPZ auto_parallelFz#store i64 %\"\.\d", i64\* %"retptr")rOr FlagsoptionsdefaultrH assertRegexrQrrr_check_dump_llvm\s zDebugTestBase._check_dump_llvmcC"|d|j||d|dS)Nz--FUNCTION OPTIMIZED DUMP %sadd nsw i64 %arg.somearg, 1rWrQrrr_check_dump_func_opt_llvmasz'DebugTestBase._check_dump_func_opt_llvmcCr`)Nz--OPTIMIZED DUMP %srarWrQrrr_check_dump_optimized_llvmfsz(DebugTestBase._check_dump_optimized_llvmcCs2|d|j|tdvr|d|dSdS)Nz --ASSEMBLY %s)x86_64ZAMD64i386i686Zxorl)rOrXplatformmachinerQrrr_check_dump_assemblyjs z"DebugTestBase._check_dump_assemblyN)r r!r"setrBr?rIrRrUrYrZr_rbrcrirrrrr27s   r2c@seZdZdZddZdS)FunctionDebugTestBasercCsTt}ttjft}||ddWd|S1s!wY|S)NrJ)rr r int64rassertPreciseEqualgetvaluerrFcfuncrrrcompile_simple_nopythonts z-FunctionDebugTestBase.compile_simple_nopythonN)r r!r"rXrrrrrrrkps rkc@sDeZdZddZddZddZddZd d Zd d Zd dZ dS)TestFunctionDebugOutputcCDtdd |}Wdn1swY||dgdS)NZ DUMP_BYTECODETr3rrrrIrQrrrtest_dump_bytecode~  z*TestFunctionDebugOutput.test_dump_bytecodecCrt)NDUMP_IRTr5rurQrrr test_dump_irrwz$TestFunctionDebugOutput.test_dump_ircCrt)NZDUMP_CFGTr4rurQrrr test_dump_cfgrwz%TestFunctionDebugOutput.test_dump_cfgcCrt)NZ DUMP_LLVMTr7rurQrrrtest_dump_llvmrwz&TestFunctionDebugOutput.test_dump_llvmcCrt)NZ DUMP_FUNC_OPTTr8rurQrrrtest_dump_func_opt_llvmrwz/TestFunctionDebugOutput.test_dump_func_opt_llvmcCrt)NZDUMP_OPTIMIZEDTr9rurQrrrtest_dump_optimized_llvmrwz0TestFunctionDebugOutput.test_dump_optimized_llvmcCrt)NZ DUMP_ASSEMBLYTr:rurQrrrtest_dump_assemblyrwz*TestFunctionDebugOutput.test_dump_assemblyN) r r!r"rvryrzr{r|r}r~rrrrrs|s rsc@ eZdZdZddZddZdS)TestGeneratorDebugOutputrcCsbt"}ttjtjft}|t|ddddgWd|S1s(wY|S)Nrlr)rr r rmrrnlistrorprrrcompile_simple_gens z+TestGeneratorDebugOutput.compile_simple_gencCsltdd |}Wdn1swY||dg|d|j|td}|||dS)NrxTr5z--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'] )rrrIrOrXtextwrapdedent)rrFZexpected_gen_inforrrtest_dump_ir_generators   z/TestGeneratorDebugOutput.test_dump_ir_generatorN)r r!r"rXrrrrrrrs rc@r)TestDisableJITz: Test the NUMBA_DISABLE_JIT environment variable. c Cstdd3ttddt}||ddWdn1s#wYWddSWddS1s;wYdS)N DISABLE_JITTZnopythonrlrJ)rrr rrn)rrqrrrtest_jits "zTestDisableJIT.test_jitc CstddCt'ttt}|}||jdtddt}|||dWdn1s3wYWddSWddS1sKwYdS)NrTrr) rrrsimple_class_specrrnrr r$)rZSimpleJITClassr#rqrrr 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)NZ NUMBA_DEBUG1)r5r6r7r8r9r:)rrZ assertFalserrIrQrrr 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 assertTruerZ warn_listmsgZ 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. rlCTrecordalwaysZparallelN) r ZArrayZfloat64warningscatch_warnings simplefilterrr r)r)rZarr_tyrrrr test_warnss  zTestParforsDebug.test_warnscCsZtjdd}tdtttjfddtWdn1s!wY||dS)zG Test that prange with multiple exits issues a warning TrrrN) rrrrr r rmr1r)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 TrZNUMBA_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 rmr/rrrosplitintrecompilesearchgrouprangelenzipformatrOr rj)rrFoutputZparallel_loop_outputZ fuse_outputZafter_fusion_outputZ parfor_stateZboundspatternfmtr.ZtrialsZlpatternZto_matchrrrtest_array_debug_opt_statssN         "z+TestParforsDebug.test_array_debug_opt_statsN) r r!r"rrrrrrrrrrrrrrs      r__main__)/osrgrrrnumpyr%Znumba.tests.supportrrrrrrrZnumbar r Z numba.corer r r Znumba.core.errorsrrZnumba.experimentalrZunittestrrobjectrZint32rr$r)r/r1r2rkrsrrrrr mainrrrrs>$    9 &m