iKjBUdZddlZddlZddlmZddlmZmZerddlm Z e e dze e fZeed<dZde fd Zd ed efd edefd Zddd ed efd eddfdZdefdZde defdZdae dzed<e Ze e ed<de fdZedk(rddlZeeyy)aN This is a toolkit for determining which module set the "flush to zero" flag. For details, see the docstring and comments in `identify_ftz_culprit()`. This module is defined outside the main Hypothesis namespace so that we can avoid triggering import of Hypothesis itself from each subprocess which must import the worker function. N)Callable) TYPE_CHECKING TypeAlias)Queue FTZCulprits)1zarchive-pdf-toolsz bgfx-pythonzbicleaner-ai-gloveBTreescadbiom ctranslate2dyNETdyNET38geventzglove-python-binaryhigrahybridqikomiaiohz jij-cimodlavavuz lavavu-osmesa MulticoreTSNEzneural-compressornwhyopenjij openturns perfmetricspHashPyz pyace-litepyapr pycompadrezpycompadre-serialPyKEPpykepz pylimer-toolspyqubopyscfPyTATz python-prtreez qiskit-aerzqiskit-aer-gpu RelStoragezsail-ml segmentationsentesinrsnapmlsuperman symenginez systran-alignz texture-toolztsne-mpxcsfreturnc ddk(S)Ngrr*]/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/_hypothesis_ftz_detector.py flush_to_zeror-.s ?r+fn.argscddl}|jdd|j}|jt||g|}|j |j }|j|S)NrspawnT)force)targetr/)multiprocessingset_start_methodrProcessr3startgetjoin)r.r/mpqpretvals r,run_in_processr>3s_ t,HHJA &2~~ 6AGGI UUWFFFH Mr+r;zQueue[FTZCulprits]c,|j||yN)put)r;r.r/s r,r3r3?sEE"d)r+cHtttjfSr@)r-setsysmodulesr*r+r,always_imported_modulesrFEs ?C , ,,r+modcttj} tj|ttj|z }t |fS#t $rdtfcYSwxYw)z7Return the set of modules imported transitively by mod.N)rCrDrE importlib import_module Exceptionr-)rGbeforeimportss r,modules_imported_byrNIsb  F$#++'G ?G ## SU{sAA0/A0 KNOWN_FTZ CHECKED_CACHEctrtStt\}}|r tdtj |dt dtttt ffd}ttjtz }i}|rdt||}|j|tt|\}}|tz}|r |||<||z}n||z}tj ||rdtd|D}|D chc]} | j!|r| } } t#| D]3} t#| D]#} | || vs | || vs| j| %5t| atScc} w)zAFind the modules in sys.modules which cause "mod" to be imported.z3Python is always in FTZ mode, even without imports!namer(c6|tv|jd|fS)zEPrefer known-FTZ modules, then top-level packages, then alphabetical..)KNOWN_EVER_CULPRITScount)rRs r,keyz"identify_ftz_culprits..keyks//C$GGr+)rWc3&K|] }|dz yw)rTNr*).0ns r, z(identify_ftz_culprits..s9QW9s)rOr>rF RuntimeErrorrPupdatestrtupleboolintrCrDrEmindiscardrN startswithsorted) always_enables_ftzalways_importsrW candidatestriggering_modulesrG enables_ftzrMprefixeskresultabs r,identify_ftz_culpritsrpZs~*88O)P&PQQ( H#H%c3/H S[[!M1J *#&3-.A3G W=  &- s # ' !J ' !J   ) "9&899H+ JA1<<3Ia JF JF^" "A&q))a7I!7L.Lq! ""F I Ks E0E0__main__)__doc__rIrDcollections.abcrtypingrrr4rr_r`rCr^r__annotations__rUr-objectr>r3rFrNrOrPrp__name__ grequestsprintr*r+r,rzs $+%td{CH45 Y5  t x[ 01 & [ !)#{*:!;DJ  --$S$[$ 3:% s3x8s8v z   !"r+