i:dZddlZddlZddlmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlZddlZddlZddlZddlmZddlZddlZddlZddlmZddlZddl m!Z!mZddl"m#Z#m$Z$m%Z%m&Z&m'Z'ddl(m)Z)dd l*m+Z+m,Z,m-Z-m.Z.m/Z/dd l0m1Z1dd l2m3Z3ddl4Z4dd l5m6Z6dd l7m8Z8ddl5m9Z:ddl;mZ>m?Z?m@Z@mAZAddlBmCZC ddlDZD ddlFZFeFje,ZHdeH_Ie,ZJdeJ_Ke,ZLe,ZMdeM_Ne!jgdZPePdZQe jdkZSeSZTe4jeTdZVe4je%jdk\dZYe4je%jdk(dZZe4je%jdk(dZ[e4je%jdk\xre jjddZ]e4je%jdk(dZ^e4je jjd xrejd!k(d"Z`e4jeSd#Zae8jd$k\Zce4jecd%Zdeeefe jjd&dZi e4jeid'Zjeke d(d) Zle4jeld*Zmd+Znd,Zod-Zpd.Zqd/Zrd0Zse4jeDduesZte4je)jd1Zvd2Zwe4je jjd  ewZxd3Zye4je jjd eyZzejd4k(Z{e4jdd5Z|e4jejd6k(d7Z}eee jjd8dZ~e4je~d9Ze4je~d:Ze4jejd6k(d;Ze4jejd6k(d<ZejZej d=k(Ze4jed>Zexrejd?k(Z ddlZDdZe4jedAZ ddlZDdZe4jedBZe jjdCddDk(Ze4jedEZ ddlZdZe4jedFZdGZGdHdIe4j,ZGdJdKeZej.dLZej.dMZdNZdOZe j:dPk(r/e j<j?ej@eZnCe j<j?ej@edQe jBZdRZdSZdTZdUZdVZdWZej.dXZdYZdZZej.d[ZGd\d]eZGd^d_eZGd`daeeZej.dbZej.dcZddZdeZedfdfZdsdgZdhZGdidjeZdkZej.dlZdtdmZdudnZdoZGdpdqe.ZdrZy#eE$rdZDY wxYw#eE$rYwxYw#eE$rd@ZY:wxYw#eE$rd@ZY.wxYw#eEef$rd@ZYwxYw)vz& Assorted utilities for use in tests. N) defaultdict)contextmanager)cached_property)testingtypes)errorstypingutilsconfigcpu) cffi_utils) compile_extraFlags DEFAULT_FLAGS CompilerBaseDefaultPassBuilder)IRLegalization) PreserveIR)rtsys) numpy_support) _nrt_python)overload_method typeof_implregister_modelunbox NativeValuemodels) OpaqueModelT) important long_running always_testr!zparfors not supported) zneeds Python 3.10 or laterzneeds Python 3.10r# zneeds Python 3.12r# winz(Not supported on Python 3.13+ on Windowsr#zTest unstable on 3.14linuxaarch64zNot supported on Linux aarch64zNot supported on 32 bitrzNot supported on numpy 2.0+_NUMBA_REDUCED_TESTINGzSkipped for reduced testing_is_gil_enabledcy)NTr3h/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/numba/tests/support.pyr6sdr4zWSkipped [NOT APPLICABLE] if using a free-threading build and free-threading is enabled.cVtjdk(rtj|S|S)N)r# r PYVERSIONunittestexpectedFailurefns r5expected_failure_py311r?$ '!''++ r4cVtjdk(rtj|S|S)Nr%r9r=s r5expected_failure_py312rBr@r4cVtjdk(rtj|S|S)Nr'r9r=s r5expected_failure_py313rDr@r4cVtjdk(rtj|S|S)Nr*r9r=s r5expected_failure_py314rFr@r4cVtjdk(rtj|S|S)Nr.)r numpy_versionr;r<r=s r5expected_failure_np2rIs&""f,''++ r4zSciPy needed for testz requires cffizlinux only testzWindows-only testarmv7lz Test disabledppc64lez?Hits: 'Parameter area must exist to pass an argument in memory'NUMBA_USE_TYPEGUARDzTypeguard is not enabledzBroken if Typeguard is enabledz>Hits: 'mismatch in size of parameter area' in LowerCall_64SVR4zInvalid PPC CTR loopDarwinz-fenv.h-like functionality unreliable on macOSarm64FzLAPACK needs SciPy 1.0+zBLAS needs SciPy 1.0+ SUBPROC_TEST1zneeds subprocess harnesszTest needs setuptoolsctjddtjddtjdy)zUse in testing within a ` warnings.catch_warnings` block to filter out warnings that are unrelated/internally generated by Numba. ignore typeguard)modulez.*TBB_INTERFACE_VERSION.*znumba\.np\.ufunc\.parallel.*)actionmessagecategoryrTN)warningsfilterwarningsr NumbaWarningr3r4r5ignore_internal_warningsr[ s4  H[9 8$@%+%8%8#BDr4ceZdZdZedZdZejdZ ejdZ dZ ejdZ e ejfZeefefej&feej*fgZefefej2gZeefgZeej<fZeej@fZ!dZ"d Z#d Z$d Z% dd Z& ddZ' ddZ(ddZ)dZ*dZ+y )TestCaseTc@tjjdS)N*)nprandom RandomStateselfs r5razTestCase.random syy$$R((r4cnt|trtj|} |`y#t $rYywxYw)z Reset the warnings registry of a module. This can be necessary as the warnings module is buggy in that regard. See http://bugs.python.org/issue4180 N) isinstancestrsysmodules__warningregistry__AttributeError)rdrTs r5reset_module_warningszTestCase.reset_module_warnings$s9 fc "[[(F *   s ( 44c#Ktjtjttf}|j |5}|dddy#1swYyxYww)zr A context manager that asserts the enclosed code block fails compiling in nopython mode. N)r LoweringError TypingError TypeErrorNotImplementedError assertRaises)rd_accepted_errorscms r5assertTypingErrorzTestCase.assertTypingError1sH #00&2D2D%':<   / 0BH1 0 0s >s!B1B'B1B,1B1 $B1c tj|Dcgc]}tj|}}|d}t t |ddD](}||}||k7s|j d|d|d|d*ycc}w)Nrz(Refcount for objects does not match. #0(z) != #(z) does not match.)rwrxrhryrangelenr{)rdr|r}rcrc_0irc_is r5assertRefCountEqualzTestCase.assertRefCountEqualRs *1 2'Qcooa ' 2!us7|$QR(Aa5Dt|  $vVA3av5FHI)3sA?c#xKtj}dtj}|j|jz }|j|jz }|j|jz }|j |j z }|j ||d|j ||dyw)z| A context manager that asserts no NRT leak was created during the execution of the enclosed block. Nz-number of data allocs != number of data freesz3number of meminfo allocs != number of meminfo frees)rget_allocation_statsallocfreemi_allocmi_free assertEqualrdrr total_alloc total_freetotal_mi_alloc total_mi_frees r5assertNoNRTLeakzTestCase.assertNoNRTLeak\s ((* ((*ii#))+ XX(  4 ckk1  jH J N PsB8B:cht|tjryt|tjry|j D]}t||sy|j D]}t||sy|jD]}t||sy|jD]}t||syy)z This function returns a string description of the type family that the object in question belongs to. Possible return values are: "exact", "complex", "approximate", "sequence", and "unknown" ndarrayenumsequenceexactcomplex approximateunknown) rfr`rrEnum_sequence_typesets_exact_typesets_complex_types_approx_typesets)rdnumeric_objecttps r5_detect_familyzTestCase._detect_familyws nbjj 1 ndii 0))B."-!*&&B."-'%%B."- &''B."-$(r4ctjdk(r@tjdkDr-|tjdk(rtjdS|S)z7 Fix the given *dtype* for comparison. win32r"int32int64)rhplatformmaxsizer`dtype)rdrs r5 _fix_dtypezTestCase._fix_dtypes@ LLG # e(; RXXg& &88G$ $Lr4c|jdk(rdg|jzSt|j|jDcgc]\}}|dkDr||j z c}}Scc}}w)z Return the strides of the given array, fixed for comparison. Strides for 0- or 1-sized dimensions are ignored. rr)sizendimrzstridesshapeitemsize)rdarrstriders r5 _fix_strideszTestCase._fix_stridessi 88q=3> !,/s{{CII+F"+FqyS\\)+F" ""sA%c|j|j|jd|j|j|jd|j|j||j|dy)zG Test that two arrays have the same shape and strides. z shapes differzitemsizes differzstrides differN)rrrr)rdfirstseconds r5assertStridesEqualzTestCase.assertStridesEqualsd fllOD :LM **5143D3DV3L) +r4Nc  |j|||||||y#t$r}t|} Yd}~nd}~wwxYw|jd|d|d| y)ag Versatile equality testing function with more built-in checks than standard assertEqual(). For arrays, test that layout, dtype, shape are identical, and recursively call assertPreciseEqual() on the contents. For other sequences, recursively call assertPreciseEqual() on the contents. For scalars, test that two scalars or have similar types and are equal up to a computed precision. If the scalars are instances of exact types or if *prec* is 'exact', they are compared exactly. If the scalars are instances of inexact types (float, complex) and *prec* is not 'exact', then the number of significant bits is computed according to the value of *prec*: 53 bits if *prec* is 'double', 24 bits if *prec* is single. This number of bits can be lowered by raising the *ulps* value. ignore_sign_on_zero can be set to True if zeros are to be considered equal regardless of their sign bit. abs_tol if this is set to a float value its value is used in the following. If, however, this is set to the string "eps" then machine precision of the type(first) is used in the following instead. This kwarg is used to check if the absolute difference in value between first and second is less than the value set, if so the numbers being compared are considered equal. (This is to handle small numbers typically of magnitude less than machine precision). Any value of *prec* other than 'exact', 'single' or 'double' will raise an error. Nzwhen comparing z and : )_assertPreciseEqualAssertionErrorrgr{) rdrrpreculpsmsgignore_sign_on_zeroabs_tolexc failure_msgs r5assertPreciseEqualzTestCase.assertPreciseEqualsZH   $ $UFD$#W .   #c(K # E6;OPs 8 38c  d!fd }j|} j|} d| d| d} r| dz } j| | | | } | dk(rj|j} j| j|jj|j|jd j|j |j d j|j j|j jd jj|j|d |j| k7r|j| }|j| k7r|j| }t|j|jD]\}}j|||||y| d k(rRjt|t|t||D]\}}j|||||y| dk(rd}n| dvrd}n| dk(rRj|j|jj|j |j |||y| dk(r)j|j|jd}nJdt#|dr2t#|dr&j|j|jt%|j&t%|j&k7r8d|jd|jd} r| dz } j)|  t+j,|rt+j,|ry|a|dk(r)t1j2t5|j6}n!t%|t8r|}nt;d|zt=||z |kry|xs|dk(}|sI|dk7rD|dk(rd}n|dk(rd}nt;d|d||z d z z}|t=|t=|zz}nd}t%|j>r=||j@|j@|||jB|jB|yt%|t0jDt0jFfrYt1jH|r9jt1jH|t1jH|y||||y||||y#t.$rYwxYw)"z-Recursive workhorse for assertPreciseEqual().Nc|7||cxk(rdk(s,ntj|stj|rmj||sV tjd|tjd|k7r'j j |d|yyyj|||y#t $rYywxYw)Ngrr != )deltar)mathisinfrcopysignr{_formatMessagerpassertAlmostEqual)rrrrrrds r5_assertNumberEqualz8TestCase._assertPreciseEqual.._assertNumberEquals &!7C!7::e$ 6(:  C 8*==E2dmmAv6NN II $ 3 3C5:F5D!EFO+&&ufEs&K%sAB>> C  C zType Family mismatch. (r)rrrzdifferent number of dimensionszdifferent shapeszdifferent mutabilityzdifferent stridesrrT)rrFrrzunexpected familyrzMismatching return types (z vs. epsz)abs_tol is not "eps" or a float, found %ssingledouble5zunsupported precision r/rN)%rrrrrrflags writeablerastyperzflatrrassertIs __class__valuehasattrrf _bool_typesr{cmathisnanrpr`finfotyperfloat ValueErrorabsrrealimag timedelta64 datetime64isnat)rdrrrrrrrr first_family second_familyassertion_messagecompare_familyrabexact_comparisonrtolbitskrs` `` r5rzTestCase._assertPreciseEquals  L"**51 ++F3 BN   3 0  }:KL& Y &OOEKK0E   UDOOFLL$A B   UZZ= ?   U[[&,,/ 1   U[[22FLL4J4J3 5   T..u5!!&)+> @{{e# U+||u$u-EJJ 41((AtT3)E6*1((AtT3)"T)+>twDGOx! t!EFFdTkAo&AUc&k12EE eT00 1 uzz6;; > uzz6;; >  > ?xx  %"((62BC"5&%8 ufe 4O   s3*T88 UUcx|j}t|j} |||f} djd| D} g} |<|j D])\} }| j | | j |+t jg| dd| }tjj}||d< tjd|d<tj|in|}|j|tj |tj"tj"||d }d |j$d |j&}|j)|j*d |d }||j&vr|j-||S|j/d|j&|S#t$rYwxYw)a Runs named unit test(s) as specified in the arguments as: test_module.test_class.test_name. test_module must always be supplied and if no further refinement is made with test_class and test_name then all tests in the module will be run. The tests will be run in a subprocess with environment variables specified in `envvars`. If given, envvars must be a map of form: environment variable name (str) -> value (str) If given, flags must be a map of form: flag including the `-` (str) -> value (str) It is most convenient to use this method in conjunction with @needs_subprocess as the decorator will cause the decorated test to be skipped unless the `SUBPROC_TEST` environment variable is set to the same value of ``_subproc_test_env`` (this special environment variable is set by this method such that the specified test(s) will not be skipped in the subprocess). Following execution in the subprocess this method will check the test(s) executed without error. The timeout kwarg can be used to allow more time for longer running tests, it defaults to 60 seconds. .c3&K|] }|| ywrr3).0r}s r5 z2TestCase.subprocess_test_runner..s'J5aAM5sz-mznumba.runtestsrOCOVERAGE_RCFILECOVERAGE_PROCESS_STARTT)stdoutstderrtimeoutenvuniversal_newlinesz captured stdout: z captured stderr: rz NO TESTS RANOK) __module__r__name__joinitemsappendrh executableosenvironcopyKeyErrorpytypesMappingProxyTypeupdate subprocessrunPIPErrr returncodeskipTestassertIn)rd test_module test_class test_nameenvvarsrr_subproc_test_envthemodtheclspartsfully_qualified_test flags_argsflagrcmdenv_copystatusstreams no_tests_rans r5subprocess_test_runnerzTestCase.subprocess_test_runnervs2d$$j)4"xx'J5'JJ  ${{} e!!TF,!!UG- -~~% %D%2B%#%::??$#4  13.wrapper..innersi::>>.$74==H"&!8!8J//$(OO#-"&-- ' '*.-- 0Jr4r3)r5r6r rs` r5wrapperz0TestCase.run_test_in_subprocess..wrappers "Lr4)rfr FunctionType) maybefuncrr r7s `` r5run_test_in_subprocesszTestCase.run_test_in_subprocesss* ( i!5!5 69% %Nr4c6|j}tdj|tjfi}|dt |t Gddt}tj|fd}t|d}||fS)zUse to generate a dummy type unique to this test. Returns a python Dummy class and a corresponding Numba type DummyType.zDummyTypeFor{}my_dummyc eZdZy)'TestCase.make_dummy_type..DummyN)r r  __qualname__r3r4r5Dummyr>s r4r@cSrr3)valc dummy_types r5 typeof_dummyz.TestCase.make_dummy_type..typeof_dummys  r4cHt|jjSr)rcontextget_dummy_value)typrrCs r5 unbox_dummyz-TestCase.make_dummy_type..unbox_dummysqyy88:; ;r4) idrformatrOpaquerrobjectrregisterr)rdtest_id DummyTyper@rErJrDs @r5make_dummy_typezTestCase.make_dummy_types ''))009ELL?BO z* !y!+. F    e $  %  y  <  <ir4cBddlm}|s|jdyy)a Call this to ensure the test is skipped if no suitable external compiler is found. This is a method on the TestCase opposed to a stand-alone decorator so as to make it "lazy" via runtime evaluation opposed to running at test-discovery time. r)external_compiler_worksz(No suitable external compiler was found.N)numba.pycc.platformrTr)rdrTs r5skip_if_no_external_compilerz%TestCase.skip_if_no_external_compilers  @&( MMD E)r4)rrNFN)NNN<NrP)NrWN),r r r? longMessagerrarl contextlibrrurrrboolr`bool_rintrgintegerbytesbytes_rrrinexactrtuplelistrfloating _float_typescomplexfloatingrrrrrrrr-r:rRrVr3r4r5r]r]sIK)) --*IPP$"K"SFSFRZZMryy)+O7*rzz; $-2;;'Lr112N<  "+DE9>#'.Q`EF:?$(N5`>BEG=@8t: 0 Fr4r]ceZdZdZdZy) SerialMixinz-Mixin to mark test for serial execution. FN)r r r?__doc___numba_parallel_test_r3r4r5rgrgs !r4rgc#Ktt|}tt|| dtt||y#tt||wxYww)z Return a context manager that temporarily sets Numba config variable *name* to *value*. *name* must be the name of an existing variable in numba.config. N)getattrr setattr)namer old_values r5override_configros@%I FD% ) i(i(s"A;AAAc#Ktjj|}|tj|<tj d|tj|=n|tj|<tjy#|tj|=n|tj|<tjwxYww)zk Return a context manager that temporarily sets an Numba config environment *name* to *value*. N)rrr1r reload_config)rmrrs r5override_env_configrr s **.. CBJJt   ; 4  #BJJt  ; 4  #BJJt sAC  B r)compilerstripeval)rmcodeglobsconss r5compile_functionr{!s2  H 5B BUB d8Or4z numba-testsntriQcV tjty#t$rYywxYwr)rmkdir _trashcan_dirFileExistsErrorr3r4r5_create_trashcan_dirr;s%      s  ((cttjtz }ttjt D]b}tj jt |} tj|}|j|krtj|ddy#t$r }Yd}~td}~wwxYw)NT) ignore_errors) time_trashcan_timeoutsortedrlistdirrpathr statst_mtimeshutilrmtreeOSError)freshness_thresholdr>stes r5_purge_trashcan_dirrAs))+(99RZZ ./ WW\\- , B{{00 b5 0    s&;B$$ B72B7cVttj|dzt}|S)N-)prefixdir)rtempfilemkdtempr)rrs r5_create_trashcan_subdirrNs#   6C<] CD Kr4c,tt|S)a Create a temporary directory with the given *prefix* that will survive at least as long as this process invocation. The temporary directory will be eventually deleted when it becomes stale enough. This is necessary because a DLL file can't be deleted while in use under Windows. An interesting side-effect is to be able to inspect the test files shortly after a test suite run. )rr)rs r5temp_directoryrSs "6 **r4chddl}|jt|tj|S)z Import and return a module of the given name. Care is taken to avoid issues due to Python's internal directory caching. rN) importlibinvalidate_caches __import__rhri)modnamers r5import_dynamicrcs, I!w ;;w r4c#Ktt|}tt|tj tt|tt||y#tt||wxYww)zReturn a context manager used by captured_stdout/stdin/stderr that temporarily replaces the sys stream *stream_name* with a StringIO.N)rkrhrlioStringIO) stream_name orig_stdouts r5captured_outputrpsP#{+K Cbkkm,/c;''[+.[+.s4A1A A1A..A1ctdS)zCapture the output of sys.stdout: with captured_stdout() as stdout: print("hello") self.assertEqual(stdout.getvalue(), "hello ") rrr3r4r5captured_stdoutr{ 8 $$r4ctdS)zCapture the output of sys.stderr: with captured_stderr() as stderr: print("hello", file=sys.stderr) self.assertEqual(stderr.getvalue(), "hello ") rrr3r4r5captured_stderrrrr4c#Kt5}tdd5|ddddddy#1swYxYw#1swYyxYww)N DEBUG_CACHET)rro)outs r5capture_cache_logrs9  c ]D 1I2   1 1  s) A <0< A9 <AAceZdZdZdZdZy)EnableNRTStatsMixinz,Mixin to enable the NRT statistics counters.c,tjyr)_nrtmemsys_enable_statsrcs r5setUpzEnableNRTStatsMixin.setUps   "r4c,tjyr)rmemsys_disable_statsrcs r5tearDownzEnableNRTStatsMixin.tearDowns !!#r4N)r r r?rhrrr3r4r5rrs6#$r4rc(eZdZdZdZdZdZdZy) MemoryLeakTc^tjtj|_yr)rwrxrr_MemoryLeak__init_statsrcs r5memory_leak_setupzMemoryLeak.memory_leak_setups !668r4c>|jr|jyyr)_MemoryLeak__enable_leak_checkassert_no_memory_leakrcs r5memory_leak_teardownzMemoryLeak.memory_leak_teardowns  # #  & & ( $r4cT|j}tj}|j|jz }|j|jz }|j |j z }|j |j z }|j|||j||yr)rrrrrrrrrs r5rz MemoryLeak.assert_no_memory_leaks((*ii#))+ XX(  4 ckk1  j1 7r4cd|_y)NF)rrcs r5disable_leak_checkzMemoryLeak.disable_leak_checks #( r4N)r r r?rrrrrr3r4r5rrs9 )8)r4rc(eZdZfdZfdZxZS)MemoryLeakMixincJtt| |jyr)superrrrrdrs r5rzMemoryLeakMixin.setUps ot*,  r4crtj|jtt|yr)rwrxrrrrrs r5rzMemoryLeakMixin.tearDowns%  !!# ot-/r4)r r r?rr __classcell__)rs@r5rrs!00r4rc#Kddlm}dg}i}d} |D]e}|jd}|}|ddD]}t||}|d}t||}t |s Jd|z||||f<t |||gd|j D]\\}}}t |||y#|j D]\\}}}t |||wxYww) z Forbid LLVM code generation during the execution of the context manager's enclosed block. If code generation is invoked, a RuntimeError is raised. r)codegenz%CPUCodeLibrary._finalize_final_modulectd)Nzcodegen forbidden by test case) RuntimeError)r2r3s r5r{zforbid_codegen..fails;<r rrrs r5rrGs  f  &   "#  HHV__ HHX ##%    sC B6B*BB*'B6B'$B*&B''B**B3 /B66B?c4eZdZejefdZy)CheckWarningsMixinc#\Ktjd5}tjdddddd}D]B}|D];}|t|jvs|j |j ||dz }=D|j |t|y#1swYnxYww)NT)recordalwaysrr)rXcatch_warnings simplefilterrgrVrrWr)rdmessagesrWcatchfoundrms r5check_warningsz!CheckWarningsMixin.check_warningsXs  $ $D 1U  ! !( + 2AAII&$$QZZ:QJE H .2 1s"B,B +B,AB, B)%B,N)r r r?rYrRuntimeWarningr!r3r4r5rrWs0> / /r4rc |syg}|jD]G\}}t|trdj|}|j dj||Idj |S)Nz"{}"z{}={}z, )r rfrgrLrr ) jit_optionsrkeyrs r5_format_jit_optionsr'fsc  C!'') U eS !MM%(E 7>>#u-.* 99S>r4c+LK td}djttjj dd}t jj||dz}td i|}t|d5}|j|}|j|dddtjjd |tj|}|tj j#|dtjj%|t'j(|y#1swYxYw#tj j#dtjj%t'j(|wxYww) aA context manager that creates and imports a temporary module from sources provided in ``source_lines``. Optionally it is possible to provide jit options for ``jit_module`` if it is explicitly used in ``source_lines`` like ``jit_module({jit_options})``. test_temp_moduleztest_temp_module_{}r_z.pyr)r%Nrr3)rrLrguuiduuid4replacerrr r'rwriterhinsertr import_moduleripopremoverr) source_linesr%tempdirtemp_module_nametemp_module_pathflinesrs r5create_temp_moduler9qs. !34077   % %c3 /177<<1AE1IJ)8K8 "C (A ''K'@E GGEN) 7#--.>?  ($/   g) ( ($/   gs8F$BE $D?)AE *AF$?EE AF!!F$cJ|g}tjg|zd|gz}tj|tjtj|}|j |\}}|j dk7r*d}t||j |jfz||fS)zRun a snippet of Python code in a subprocess with flags, if any are given. 'env' is passed to subprocess.Popen(). 'timeout' is passed to popen.communicate(). Returns the stdout and stderr of the subprocess after its termination. -c)rrr)rrz/process failed with code %s: stderr follows %s ) rhrrPopenr communicaterrdecode) rwrrrr(popenrerrrs r5run_in_subprocessrAs } >> e #tTl 2C   S$.OO >E   1HC 1ASE$4$4cjjl#CCDD 8Or4c  tjd5}tj}t j d}| t d|ddt|ddj|d |jg}tj|tj  j}tj| j }|j#d fd } tj$ j& } | j#||j) j+%tj |t,j.n d } | |  j1| |j3} | j|j5 ddd| S# j|j5wxYw#1swY SxYw)a)Runs strace whilst executing the function work() in the current process, captures the listed syscalls (list of strings). Takes an optional timeout in seconds, default is 10, if this is exceeded the process will be sent a SIGKILL. Returns a list of lines that are output by strace. zw+tstraceNz'No valid 'strace' binary could be foundz-qz-pz-e,z-o)rcNj}|dk7rd|d|}t|y)Nrz0strace exited non-zero, process return code was:z. )rr)problemretrr?s r5 check_returnzstrace..check_returns:""CaxIb +"3''r4)rz)If this is SIGKILL, increase the timeout?)r$)rNamedTemporaryFilergetpidrwhichrrgr rmrr<rpid threadingTimerkillrThreadr=flushpollsignalSIGINTwait readlinescancel)worksyscallsrntf parent_pid strace_binaryr( strace_pidthread_timeoutrH thread_commsrF strace_datar?s @r5rCrCs  $ $U +sYY[  X.  FG GS_SXXh'SXX   Z__>YY "%**= (  $%++53D3DEL     F IIKzz|# FMM2DW% JJL N--/K      ! ! #i ,l       ! ! #i ,l s%C G"B-F: G:"GGG)ctjjdsyd}d} t||g}|dj |vS#t$rYywxYw)z)Checks if strace is supported and workingr,FcPtjtjddgy)Nr;zexit())rrrhrr3r4r5 force_clonez%strace_supported..force_clonesh78r4cloner$)rhr startswithrCrr )rcsyscalltraces r5strace_supportedrhsa << " "7 +9G{WJ/ bggen $$ s A AAceZdZdZdZy)IRPreservingTestPipelinez Same as the standard pipeline, but preserves the func_ir into the metadata store after legalisation, useful for testing IR changesctj|jd}d|_|j t t |j|gS)Nir_preserving_custom_pipeF)rdefine_nopython_pipelinestate _finalizedadd_pass_afterrrfinalize)rdpipelines r5define_pipelinesz)IRPreservingTestPipeline.define_pipelinessJ%>> JJ35 $ N;zr4N)r r r?rhrsr3r4r5rjrjs H r4rjc ddl}ddlm}tjj tjj t }tjj|ddd}tjj|s tdttjj |d5}|j}ddd|j|}|d}td }|dd D]5} | d d } | jD]} || d | dxxdz cc<7gd} tjj|g| } tjj| s tdttjj | d5}|j}ddd|j||}|dddd } | jD]} || d | dxxdz cc<tdtdt!|j#D]2\}}|j#D]\}}td|d|dd|4td}t!|j#D]%\}}|j#D] \}}||||<'tdtdd}t!|j#|D]2\}}|j#D]\}}td|dd|d|4y#1swYbxYw#1swY~xYw)a-This is a utility function that prints out the map of NumPy to Python versions and how many of that combination are being tested across all the declared config for azure-pipelines. It is useful to run when updating the azure-pipelines config to be able to quickly see what the coverage is.rN)Loader..zazure-pipelines.ymlz&'azure-pipelines.yml' is not availablertjobsc ttSrrr\r3r4r5r6z$print_azure_matrix..s K$4r4r/ parametersmatrixNUMPYPYTHONr)rvrv buildscriptsazurezazure-windows.ymlz$'azure-windows.yml' is not availablestrategyzNumPy | Python | Countz----------------------- z | z<4z | c ttSrrzr3r4r5r6z$print_azure_matrix..5s +c"2r4z Python | NumPy | CountcBt|djddS)Nrrr)r\r)r}s r5r6z$print_azure_matrix..;ss1Q4::c?1-.r4)r&)yamlrurrdirnameabspathrr isfilerrreadloadrvaluesr rr )rru base_path azure_piper7datapipe_yml templates py2np_maptmpltr|setupwinpath azure_windows windows_ymlnpverpyspyvercountrev_mapsorternpss r5print_azure_matrixrs  9:Iit5JKJ 77>>* %CDD bggooj)4 0Avvx 1yyfy-H I45I2A|$X.]]_E eGn %eHo 6 9 6% IGGGLL5W5M 77>>- (ABB bggoom,d 3qvvx 4))D)0K  #J / 9F%.!%/2A52! "# #$Y__./ sIIKLE5 AeWDr &8 9(0 23GY__./ sIIKLE5$)GEN5 !(0 $% #$ .FW]]_&9 sIIKLE5 AeBZtE7&8 9(:S 1 0" 4 3sL11L>1L;>M)T)NN)r$)rhrrY collectionsrrrwrrrrSrrrhrrMrrrmultiprocessingrrXrrr+rrr functoolsrnumpyr`rrrrr r r r numba.core.typingr numba.core.compilerrrrrrnumba.core.typed_passesrnumba.core.untyped_passesrr;numba.core.runtimernumba.nprrrnumba.core.extendingrrrrrrnumba.core.datamodel.modelsrscipy ImportErrorcoverageprocess_startupenable_pyobj_flagsenable_pyobjectforce_pyobj_flagsforce_pyobjectno_pyobj_flags nrt_flagsnrtmake_tag_decoratortagr!r_32bitis_parfors_unsupportedskipIfskip_parfors_unsupported skipUnlessr:skip_unless_py10_or_laterskip_unless_py10skip_unless_py312reskip_if_py313plus_on_windows skip_if_py314machineskip_if_linux_aarch64 skip_if_32bitrH IS_NUMPY_2skip_if_numpy_2rZr\rr1REDUCED_TESTINGskip_if_reduced_testingrk_free_threadingskip_if_freethreadingr?rBrDrFrI_msgskip_unless_scipy SUPPORTEDskip_unless_cffi _lnx_reason linux_only _win_reason windows_only _is_armv7l disabled_testskip_ppc64le_issue4563 has_typeguardskip_unless_typeguardskip_if_typeguardskip_ppc64le_issue6465skip_ppc64le_invalid_ctr_loopuname_unamesystemIS_MACOSskip_macos_fenv_errorsIS_MACOS_ARM64scipy.linalg.cython_lapack has_lapack needs_lapackscipy.linalg.cython_blashas_blas needs_blas _exec_condneeds_subprocess setuptoolshas_setuptoolsrneeds_setuptoolsr[r]rNrgrorrr{rrmrr  gettempdirgetuidrrrrrrrrrrrrrrrrrr rrrrr'r9rArCrhrjrr3r4r5rs #    % % 99(5530$"24 HW%)"G#'  G  !g  !MN-  *8?? 0H// OOw  '8&& OOw (H'' OOw /x __BCLL$;$;E$B/  x __!8 (LLG$H)9)9)9);y)H$  (AB  ( (F 2 !(//*"?As2::>>*BAFGH *(///*GIDgc#4lCEE(*FH   #HOOETM48&8&&z';';_M X__!8!8!AA; O ! x3<<#:#:5#AA;O X   8 + o6 ))9)9)9);y)H+KM RZZ^^$91=> +++-$HOO3 ))9)9)9);y)H+=?!0H)#!   ==H $(357fnn7%J#x"":#<> #H!X +B C ZZ^^ND 1S 8 &8&&z3MNN'8&&~7NO DVFx  VFr"&"  ) ) , 77d?GGLL!4!4!4!6 FMGGLL!4!4!4!6,99299;!GIM   +   //%%  $&$))8 0): 0 **D ,7?M"$N / / :&>B%&|$6:_& E   XJH WNsZ[[%;[1[?!\ ["!["%[.-[.1[<;[<?\  \  \\