GJjdZddlZddlZddlZddlmZmZddlmZddl m Z m Z dZ dZ dZd ZGd d eZed k(rej$yy) aB Regression test for issue #5611 / PR #10482. CFGraph._find_topo_order must be stack-based (not recursive) so that functions with large control-flow graphs do not exhaust the Python call stack. The test is run in a subprocess so that sys.setrecursionlimit manipulation is completely isolated from the rest of the suite. N)CFGraphControlFlowAnalysis)TestCase)FunctionIdentityByteCodecdg}t|D]/}|jd|d|jd|d|d1|jddj|S) z/ Source for function with unusually large CFG. zdef _large_cfg_func(x):z dep_z = 1 if True else 0z val_z = x if dep_z else 0z return x )rangeappendjoin)countlinesis f/Users/ahmed/devFolder/claude-voice/.venv/lib/python3.12/site-packages/numba/tests/test_controlflow.py_generate_large_cfg_sourcersf & 'E 5\ xs"567 xs,qc9: LL 99U ctj|}t|}t|}|j |j S)z;Create a CFGraph from a Python function for direct testing.)r from_functionrrrungraph)funcfunc_idbccfas r_create_cfg_from_functionr#s9,,T2G ' B b !CGGI 99rc*d}|r|nd}d}|r|}|Sd}|S)z7 Test function for probing stack depth during compile. r)xdep_0val_0dep_1val_1s r _complex_funcr$,s- EAAE EAE HE Hrcdgtt}tjfd}|t_ |jt_dS#t_wxYw)zj Return the Python frame depth at CFGraph._find_topo_order during a trivial njit compilation. rctj}d}||dz }|j}||dkDr|d<|SNrrsys _getframef_back)selffd depth_sampleoriginals r_probing_topo_orderz;_measure_trivial_compile_depth.._probing_topo_order>sR MMO m FAAm |A LO~r)rr$r_find_topo_orderprocess)cfgr1r/r0s @@r_measure_trivial_compile_depthr55s] 3L #M 2C''H 3G# " ?#s A A$c0eZdZejdZy)TestCFGTopoOrderNonRecursivecPttdz}d}t|}td}i}t t |dd|t |d}tj}tj||zd} |jtj||ydd jtj|jz} |j!| y#t$r } | }Yd} ~ kd} ~ wwxYw#tj|wxYw) a  Regression tests for PR #10482: _find_topo_order must not use recursive Python calls, so that large CFGs cannot exhaust the stack. Steps ----- 1. Measure the frame depth at CFGraph.topo_order() during a trivial njit compilation by patching the method to walk the frame chain once; call this bsize. Add 2 for headroom. 1a. Determine current frame_depth 2. Lower sys.setrecursionlimit to bsize + frame_depth 3. Attempt to compile a function with 100 ternary-expression pairs. 4. Catch any RecursionError. 5. Restore sys.setrecursionlimit unconditionally. 6. Fail the test and print the traceback if a RecursionError was raised. c3bKtjd}||j}d|yywr'r()r-s rframeszVTestCFGTopoOrderNonRecursive.test_topo_order_no_recursion_in_large_cfg..framesps. a A-HH-s*//dzexec_large_cfg_funcNzeUnexpected RecursionError.Potential regression related to finding toplogical order. Full traceback: )intr5sumrr=compilerr)getrecursionlimitsetrecursionlimitr2RecursionErrorr traceback format_tb __traceback__fail) r,bsizer; frame_depthfunenvr4original_limit caught_excexcmsgs r)test_topo_order_no_recursion_in_large_cfgzFTestCFGTopoOrderNonRecursive.test_topo_order_no_recursion_in_large_cfgXs",24q89  &(m (- '    (,=(>?..0 ek12  2  "  ! !. 1   $'')--j.F.FGHI  IIcN# J   ! !. 1s* C66 D ?DDD  DD%N)__name__ __module__ __qualname__rrun_test_in_subprocessrRrrrr7r7Vs $$D%Drr7__main__)__doc__r)rFunittestnumba.core.controlflowrrnumba.tests.supportrnumba.core.bytecoderrrrr$r5r7rSmainrrrr^s\  ?(: BG8GT zHMMOr