a _$a @sDddlmZmZmZmZmZmZmZmZddgZ ddZ ddZ dS))halffloat32asarrayndarray longdoublefloat64 longcomplexcomplex___upcast_float16_array__downcast_float128_arraycCst|dr@|j}|tkr$t|tdS|tkr@|tks@td|t|t st|}|j}|tkrlt|tdS|tkr|tkstd||S|S)zb Used in _scipy_fft to upcast float16 to float32, instead of float64, as mkl_fft would dodtyper ztype %s is not supported) hasattrr rrrrr ValueError isinstancerxZxdtZ__xr3lib/python3.9/site-packages/mkl_fft/_float_utils.pyr s      cCst|drH|j}|tkr,|tks,t|tdS|tkrH|tksHt|tdSt|tst|}|j}|tkr||tks|t|tdS|tkr|tkst|tdS|S|S)zp Used in _numpy_fft to unsafely downcast float128/complex256 to complex128, instead of raising an errorr r ) rr rrrrr rrrrrrr 7s      N) Znumpyrrrrrrrr __all__r r rrrrs(