Ë èÿæi%ãóp—ddlZddlmZmZddlmZddlmZmZddl m Z ddlmZm Z mZmZmZmZddlmZmZmZmZddlmZmZmZdd lmZd Zedej<jj?ej@«fg«Gd„d e!««Z"Gd„de«Z#e$dk(rejJ«yy)éN)ÚTestCaseÚunittest)Ú cpu_target)ÚCompilerBaseÚFlags)ÚPassManager)ÚtypesÚirÚbytecodeÚcompilerÚir_utilsÚregistry)ÚExtractByteCodeÚTranslateByteCodeÚ FixupArgsÚIRProcessing)ÚNopythonTypeInferenceÚtype_inference_stageÚDeadCodeElimination)ÚjitclasséÚvalcó—eZdZd„Zy)ÚDummycó—||_y©N)r)Úselfrs ún/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/numba/tests/test_ir_utils.pyÚ__init__zDummy.__init__s €ØˆóN)Ú__name__Ú __module__Ú__qualname__r©r rrrs„ór rcó(—eZdZdZd„Zd„Zd„Zd„Zy)ÚTestIrUtilszA Tests ir handling utility functions like find_callname. cóŽ—d„}tj|«}tj}tj}t|||dd«}t j||jdjdj|j«}|jt|t«xrt|«dk(xr|ddk(«y)zGTest matching of an object method (other than Array see #3449) cóR—tdg«}|jjd«y)Néé)rrÚappend)Úds rÚ test_funcz2TestIrUtils.test_obj_func_match..test_func!s€Üqc“ ˆAØ E‰EL‰L˜Or r$Nrér*r+)rÚrun_frontendrÚtyping_contextÚtarget_contextrr Ú find_callnameÚblocksÚbodyÚvalueÚtypemapÚ assertTrueÚ isinstanceÚtupleÚlen)rr-Útest_irÚ typingctxÚ targetctxÚ typing_resÚmatched_calls rÚtest_obj_func_matchzTestIrUtils.test_obj_func_matchs´€ò ô×'Ñ'¨ Ó2ˆÜ×-Ñ-ˆ Ü×-Ñ-ˆ Ü)Øy '¨2¨tó5ˆ ä×-Ñ-ØW—^‘^ AÑ&×+Ñ+¨AÑ.×4Ñ4°j×6HÑ6HóJˆà‰œ <´Ó7ò4Ü˜LÓ)¨QÑ.ò4à$ Q™¨8Ñ3õ 5r cóâ‡—Gd„dt«}ˆfd„}ˆfd„}d„}|jtjf«}|j |«}||«}|jtjf«}|j |d¬«}||«‰jt |jdj«t |«z t |jdj««y) Ncó2—eZdZe dd„«Zdd„Zy)ú6TestIrUtils.test_dead_code_elimination..TesterNc ó¼—|€i}|s t«}d|_|€tjj}|€tjj }||||||||«S)NT)rÚnrtrrr0r1)ÚclsÚargsÚreturn_typeÚflagsÚlocalsÚlibraryr0r1s rÚmk_pipelinezBTestIrUtils.test_dead_code_elimination..Tester.mk_pipeline4si€ð>ØFÙÜ!›GEØ ” Ø!Ð)Ü%-×%8Ñ%8×%GÑ%GNØ!Ð)Ü%-×%8Ñ%8×%GÑ%GNÙ˜>¨>¸7ÀDØ&¨¨vó7ð7r có—tjj|«}||j_t«j |j«|j}d}t|«}|jtd«|jtd«|jtd«|jtd«|dur|jtd«|j«|j|«|j S)z7 Compile and return IR ÚDCE_testingzanalyzing bytecodezfix up argsz processing IRznopython frontendTzDCE after typing)rÚFunctionIdentityÚ from_functionÚstateÚfunc_idrÚrun_passrÚadd_passrrrrrÚfinalizeÚrunÚfunc_ir)rÚfuncÚDCErRrQÚnameÚpms rÚ compile_to_irzDTestIrUtils.test_dead_code_elimination..Tester.compile_to_irKsÄ€ô#×3Ñ3×AÑAÀ$ÓGØ%,— ‘ Ô"ÜÓ!×*Ñ*¨4¯:©:Ô6ØŸ ™ à$Ü Ó&Ø—‘Ô-Ð/CÔDØ—‘œI }Ô5Ø—‘œL¨/Ô:Ø—‘Ô1Ð3FÔGØ˜$‘;Ø—K‘KÔ 3Ð5GÔHØ—‘” Ø—‘u” Ø—}‘}Ð$r )NNNNNN)F)r!r"r#ÚclassmethodrLr\r$r rÚTesterrC2s-„à ð!ØØØØ#Ø#ò 7óð 7ô, %r r^cóR•—‰jt|j«d«|jd}g}|jtj «D]S}t t|dd«tj«sŒ)dt|jdd«vsŒC|j|«ŒU‰jt|«d«|D]Z}|j|j«}‰jtd|jjzd «|j«Œ\|S) Nr)rÚtargetÚdeadrZÚr*z0x%sé)ÚassertEqualr:r3Ú find_instsr ÚAssignr8ÚgetattrÚVarr`r+Úget_definitionr5r7ÚintrZ)Úthe_irÚblockÚdeadsÚxr,Ú const_valrs €rÚcheck_initial_irz@TestIrUtils.test_dead_code_elimination..check_initial_ir`såø€ð ×ÑœS §¡Ó/°Ô3Ø—M‘M !Ñ$ˆEØˆEØ×%Ñ%¤b§i¡iÖ0Üœg a¨°4Ó8¼"¿&¹&ÕAØ¤¨¯©°6¸2Ó!>Ò>ØŸ™ Qð1ð ×ÑœS ›Z¨Ô+Ûà"×1Ñ1°!·'±'Ó: Ø—‘¤ F¨Q¯X©X¯]©]Ñ$:¸BÓ ?Ø )§¡õ1ððˆLr có\•—‰jt|j«d«|jd}g}g}|jtj «D]ˆ}t t|dd«tj«r*dt|jdd«vr|j|«t t|dd«tj«sŒx|j|«ŒŠ‰jt|«d«|D])}‰j|jjdv«Œ+y) Nr)rr`rarZrbr5)éÞì^[=)rdr:r3rer rfr8rgrhr`r+ÚConstr7r5)rkrlrmÚconstsrnrs €rÚcheck_dce_irz.check_dce_irysãø€Ø×ÑœS §¡Ó/°Ô3Ø—M‘M !Ñ$ˆEØˆEØˆFØ×%Ñ%¤b§i¡iÖ0Üœg a¨°4Ó8¼"¿&¹&ÔAØ¤¨¯©°6¸2Ó!>Ñ>ØŸ™ QœÜœg a¨°$Ó7¼¿¹ÕBØ—M‘M !Õ$ð1ð ×ÑœS ›Z¨Ô+óØ—‘ §¡§ ¡ Ð5IÐ IÕJñr có,—|dz}d}|dz}d}||z}|S)Nr)rrr*rsr$)rnÚyraÚzÚdeaddeadÚrets rÚfooz3TestIrUtils.test_dead_code_elimination..foo‹s,€ØA‘ˆAØˆDØA‘ˆAØ!ˆHØa‘%ˆCØˆJr T)rYr) rrLr Úintpr\rdr:r3r4) rr^rprvr|Ú test_pipelineÚno_dceÚremovedÚw_dces ` rÚtest_dead_code_eliminationz&TestIrUtils.test_dead_code_elimination0sÊø€ô, %”\ô, %ô\ ô2 Kò$ ð×*Ñ*¬E¯J©J¨=Ó9ˆ Ø×,Ñ,¨SÓ1ˆÙ" 6Ó*ˆà×*Ñ*¬E¯J©J¨=Ó9ˆ Ø×+Ñ+¨C°TÐ+Ó:ˆÙUÔð ×Ñœ˜VŸ]™]¨1Ñ-×2Ñ2Ó3´c¸'³lÑBÜ˜UŸ\™\¨!™_×1Ñ1Ó2õ 4r cób‡—dŠˆfd„}tj|«}|jd}d}d}|jD]È}t |t j«rH|jjdk(r/tjtj||j«}t |t j«sŒ€|jjdk(sŒštjtj||j«}ŒÊ|j|t«|j|‰«y)z– Test find_const() for values in globals (ir.Global) and freevars (ir.FreeVar) that are considered constants for compilation. écó$•—t}‰}||z|zSr)ÚGLOBAL_B)ÚaÚbÚcÚ FREEVAR_Cs €rr|z/TestIrUtils.test_find_const_global..foo¦sø€ÜˆAØˆAØq‘5˜1‘9Ðr rNrˆr‰)rr/r3r4r8r rfr`rZr ÚguardÚ find_constrdr†)rr|Úf_irrlÚconst_bÚconst_cÚinstrŠs @rÚtest_find_const_globalz"TestIrUtils.test_find_const_globalŸsäø€ð ˆ ô ô ×$Ñ$ SÓ)ˆØ—‘˜A‘ˆØˆØˆà—J”JˆDÜ˜$¤§ ¡ Ô*¨t¯{©{×/?Ñ/?À3Ò/FÜ"Ÿ.™.Ü×'Ñ'¨¨t¯{©{ó<ä˜$¤§ ¡ Õ*¨t¯{©{×/?Ñ/?À3Ó/FÜ"Ÿ.™.Ü×'Ñ'¨¨t¯{©{ó<‘ðð ×Ñ˜¤(Ô+Ø×Ñ˜ )Õ,r cóš‡—d„}d„}d„}ˆfd„}||«}||«}||«}‰j||«‰j||«y)z tests flatten_labels có²—d}|dkDr|dz }|dkDr=y|dkr,|dk\r|dz }td«D]}|dz}|d ksŒn|d z }ntd«‚d}d }d}d}|S) Nrér)éé5éèr„é r*r.úsome stringé©ÚrangeÚ ValueError©r‡ÚaccrnÚ py310_defeat1Ú py310_defeat2Ú py310_defeat3Ú py310_defeat4s rr|z,TestIrUtils.test_flatten_labels..foo½sˆ€ØˆCØ1ŠuØq‘Ør’6ØØT’Ø˜’7Ø˜1‘HCÜ˜ržAØ˜1‘HCØ˜Q“wÙð#ð ˜1‘H‘Cä Ó/Ð/àˆMØˆMØˆMØˆMØˆJr cóè—d}d}|dkDr|dz }|dz }|dkDrQ|dz }y|dkr;|dk\r |dz }|dz }td«D]}|dz }|dz}|d ksŒn|dz }|d z }ntd«‚d}d }d}d}|S) Nrr„r”r)r•r–r—r˜r*r.r™ršr›)r‡rŸryrnr r¡r¢r£s rÚbarz,TestIrUtils.test_flatten_labels..barÕsº€ØˆCØˆAØ1ŠuØq‘ØR‘Ør’6Ø˜‘GAØØT’Ø˜’7Ø˜‘GAØ˜1‘HCÜ˜ržAØ˜‘GAØ˜1‘HCØ˜Q“wÙð #ð˜‘GAØ˜1‘H‘Cä Ó/Ð/ØˆMØˆMØˆMØˆMØˆJr có´—d}|dkDr|dz }|dkDryy|dkr,|dk\r|dz }td «D]}|dz}|d ksŒn|dz }ntd«‚d}d }d}d }|S)Nrr”r)r•r–é7r—r„r˜r*r.r™ršr›ržs rÚbazz,TestIrUtils.test_flatten_labels..bazòs‹€ØˆCØ1ŠuØq‘Ør’6ØàØT’Ø˜’7Ø˜1‘HCÜ˜ržAØ˜1‘HCØ˜Q“wÙð#ð ˜1‘H‘Cä Ó/Ð/ØˆMØˆMØˆMØˆMØˆJr có&•—tj|t««}tj|j«}‰jt |j««dzt|j««tj|«S)Nr)) r Úcompile_to_numba_irÚdictÚflatten_labelsr3rdÚmaxÚkeysr:Úcompute_cfg_from_blocks)rXrWÚflat_blocksrs €rÚget_flat_cfgz5TestIrUtils.test_flatten_labels..get_flat_cfgshø€Ü×2Ñ2°4¼»Ó@ˆGÜ"×1Ñ1°'·.±.ÓAˆKØ×ÑœS ×!1Ñ!1Ó!3Ó4°qÑ8¼#¸g¿n¹nÓ:MÔNÜ×3Ñ3°KÓ@Ð@r N)rdÚassertNotEqual)rr|r¥r¨r±Úfoo_cfgÚbar_cfgÚbaz_cfgs` rÚtest_flatten_labelszTestIrUtils.test_flatten_labels»sXø€ò ò0 ò: ô2 Añ˜sÓ#ˆÙ˜sÓ#ˆÙ˜sÓ#ˆà×Ñ˜ 'Ô*Ø×Ñ˜G WÕ-r N)r!r"r#Ú__doc__r@r‚r‘r¶r$r rr&r&s„ñò5ò&m4ò^-ó8[.r r&Ú__main__)&ÚnumbaÚnumba.tests.supportrrÚnumba.core.registryrÚnumba.core.compilerrrÚnumba.core.compiler_machineryrÚ numba.corer r rrr rÚnumba.core.untyped_passesrrrrÚnumba.core.typed_passesrrrÚnumba.experimentalrr†ÚcoreÚListr}Úobjectrr&r!Úmainr$r rÚrÆs¦ðÛß2Ý*ß3Ý5ßH×H÷AóA÷PñPå'ð€ñ ˆE5—:‘:×#Ñ#×(Ñ(¨¯©Ó4Ð5Ð 6Ó7ôˆFóó8ðô ~.(ô~.ðBˆzÒØ€H‡MM…Oðr