2@jdZddlZddlZddlZddlmZmZddlmZddl m Z m Z dZ dZ dZd ZGd d eZed krejdSdS) 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]7}|d|d|d|d|d8|dd|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 MC:\Projects\Historical\venv\Lib\site-packages\numba/tests/test_controlflow.py_generate_large_cfg_sourcers & 'E 5\\;; 6666777 999q999:::: LL   99U  ctj|}t|}t|}||jS)z;Create a CFGraph from a Python function for direct testing.)r from_functionrrrungraph)funcfunc_idbccfas r_create_cfg_from_functionr#sA,T22G '  B b ! !CGGIII 9rc&d}|r|nd}d}|r|nd}|S)z7 Test function for probing stack depth during compile. Tr)xdep_0val_0dep_1val_1s r _complex_funcr$,s- E AAAE E AAAE Hrcdgtt}tjfd}|t_ |t_n#t_wxYwdS)zj Return the Python frame depth at CFGraph._find_topo_order during a trivial njit compilation. rctj}d}||dz }|j}||dkr|d<|SNrrsys _getframef_back)selffd depth_sampleoriginals r_probing_topo_orderz;_measure_trivial_compile_depth.._probing_topo_order>sW MOO m FAAm |A  LOx~~r)rr$r_find_topo_orderprocess)cfgr1r/r0s @@r_measure_trivial_compile_depthr55s 3L #M 2 2C'H 3G# "("""" ?s AA(c.eZdZejdZdS)TestCFGTopoOrderNonRecursivecttdz}d}t|}td}i}t t |dd|t |d}tj}tj ||zd} | n#t$r } | }Yd} ~ nd} ~ wwxYwtj |n#tj |wxYw|dSdd tj|jz} || dS) 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. c3TKtjd}||j}dV| dSdSr'r()r-s rframeszVTestCFGTopoOrderNonRecursive.test_topo_order_no_recursion_in_large_cfg..framesps9 a  A-H-----rdzexec_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,244q899    &&((mm (-- '       (,=(>??.00 ek1222  2  " " " "   JJJJJJ   !. 1 1 1 1C !. 1 1 1 1   F$'')-j.FGGHHI  IIcNNNNNs0+C?C. C C C.CC..DN)__name__ __module__ __qualname__rrun_test_in_subprocessrRrrrr7r7Vs6 $DD%$DDDrr7__main__)__doc__r)rFunittestnumba.core.controlflowrrnumba.tests.supportrnumba.core.bytecoderrrrr$r5r7rSmainrrrr^s   ????????((((((::::::::   BGGGGG8GGGT zHMOOOOOr