a d@sbddlZddlZddlmZmZddlmZmZm Z ddl Z Gddde j Z e dkr^e dS)N)captured_stdoutoverride_config)njit vectorize guvectorizec@s<eZdZddZddZddZddZd d Zd d Zd S) TestFastMathcCsdd}tdd|}t|}||d|d||jd}||jd}|d||d||d||d|dS) NcSs|t|SNmathsinxrb/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/numba/tests/test_fastmath.pyfoo sz"TestFastMath.test_jit..fooTZfastmath?r fadd fast call fast)r assertEqual inspect_llvm signaturesassertIn assertNotInselfrfastfooslowfoofastllvmslowllvmrrrtest_jit s   zTestFastMath.test_jitcCsZdd}tddhd|}tdhd|}||dtjd|t|dtjdS)NcSs |||Srr)r yrrrrsz3TestFastMath.test_jit_subset_behaviour..fooreassocnszrr)rrnpinf assertTrueisnan)rrrrrrrtest_jit_subset_behaviours z&TestFastMath.test_jit_subset_behaviourcCsdd}tddhd|}t|}||d|d||jd}||jd}|d||d ||d ||d ||d ||d ||d |vpd |v||d |vpd |v|dS)NcSs|t|Srr r rrrr!sz.TestFastMath.test_jit_subset_code..foor"r#rrrrrzfadd reassoc nszzcall reassoc nszzfadd nsz reassoczcall nsz reassoc)rrrrrr&rrrrtest_jit_subset_code s,       z!TestFastMath.test_jit_subset_codecCs|t(}tdhddddWdn1s80Y|dt|j|t*}tddiddddWdn1s0Y|dt|j|t&}td dd ddWdn1s0Y|d t|jdS) NZspqrrcSs|dSNrr rrr<z5TestFastMath.test_jit_subset_errors..r+zUnrecognized fastmath flags:FcSs|dSr*rr rrrr,Cr-i9cSs|dSr*rr rrrr,Jr-z!Expected fastmath option(s) to be) assertRaises ValueErrorrrstr exception)rZraisesrrrtest_jit_subset_errors:s$ 6 8 4z#TestFastMath.test_jit_subset_errorsc Cs dd}tdd|}t|}tjdtj}tdd|t}||}Wdn1sb0Y|}t}||} Wdn1s0Y|} Wdn1s0Ytj || | d| | d| | d|| d|dS) NcSs|t|Srr r rrrrQsz(TestFastMath.test_vectorize..fooTrDUMP_OPTIMIZEDrr) rr$randomastypefloat32rrgetvaluetestingassert_almost_equalrr) rrrrr slow_capexpectrfast_capgotrrrrtest_vectorizePs  &&&   zTestFastMath.test_vectorizec Csdd}tjdtj}tdddg}d}t"}t||dd|}Wdn1s`0Y|}t}t|||} Wdn1s0Y|} Wdn1s0Y| |} ||} tj | | | d || d || d | | d | dS) NcSs|t||d<dS)Nrr )r outrrrresz*TestFastMath.test_guvectorize..foor3r4Tz(float32, float32[:])z()->()rrr) r$r5r6r7rrrr8r9r:rr) rrr typessigr=rrr;rrr<r>rrrtest_guvectorizeds$ 0,&   zTestFastMath.test_guvectorizeN) __name__ __module__ __qualname__r r(r)r2r?rCrrrrr s r__main__)r numpyr$Znumba.tests.supportrrZnumbarrrZunittestZTestCaserrDmainrrrrsq