a a@sjddlmmZddlmZGdddZdeiZddZd d Z d d Z dddZ ddZ e ddS)N) _pocketfftc@s eZdZdZdZeddZdS) _ScipyBackendaDThe default backend for fft calculations Notes ----- We use the domain ``numpy.scipy`` rather than ``scipy`` because in the future, ``uarray`` will treat the domain as a hierarchy. This means the user can install a single backend for ``numpy`` and have it implement ``numpy.scipy.fft`` as well. numpy.scipy.fftcCs(tt|jd}|durtS||i|S)N)getattrr__name__NotImplemented)methodargskwargsfnr b/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/scipy/fft/_backend.py__ua_function__sz_ScipyBackend.__ua_function__N)r __module__ __qualname____doc__ __ua_domain__ staticmethodrr r r rrs rZscipyc Csbt|trLz t|}Wn4tyJ}ztd||WYd}~n d}~00|jdkr^td|S)z8Maps strings to known backends and validates the backendzUnknown backend {}Nrz,Backend does not implement "numpy.scipy.fft") isinstancestr_named_backendsKeyError ValueErrorformatr)backender r r_backend_from_args  & rcCst|}t|dS)aSets the global fft backend The global backend has higher priority than registered backends, but lower priority than context-specific backends set with `set_backend`. Parameters ---------- backend : {object, 'scipy'} The backend to use. Can either be a ``str`` containing the name of a known backend {'scipy'} or an object that implements the uarray protocol. Raises ------ ValueError: If the backend does not implement ``numpy.scipy.fft``. Notes ----- This will overwrite the previously set global backend, which, by default, is the SciPy implementation. Examples -------- We can set the global fft backend: >>> from scipy.fft import fft, set_global_backend >>> set_global_backend("scipy") # Sets global backend. "scipy" is the default backend. >>> fft([1]) # Calls the global backend array([1.+0.j]) N)ruaset_global_backendrr r rr.srcCst|}t|dS)az Register a backend for permanent use. Registered backends have the lowest priority and will be tried after the global backend. Parameters ---------- backend : {object, 'scipy'} The backend to use. Can either be a ``str`` containing the name of a known backend {'scipy'} or an object that implements the uarray protocol. Raises ------ ValueError: If the backend does not implement ``numpy.scipy.fft``. Examples -------- We can register a new fft backend: >>> from scipy.fft import fft, register_backend, set_global_backend >>> class NoopBackend: # Define an invalid Backend ... __ua_domain__ = "numpy.scipy.fft" ... def __ua_function__(self, func, args, kwargs): ... return NotImplemented >>> set_global_backend(NoopBackend()) # Set the invalid backend as global >>> register_backend("scipy") # Register a new backend >>> fft([1]) # The registered backend is called because the global backend returns `NotImplemented` array([1.+0.j]) >>> set_global_backend("scipy") # Restore global backend to default N)rrregister_backendr r r rr!Qs"r!FcCst|}tj|||dS)aOContext manager to set the backend within a fixed scope. Upon entering the ``with`` statement, the given backend will be added to the list of available backends with the highest priority. Upon exit, the backend is reset to the state before entering the scope. Parameters ---------- backend : {object, 'scipy'} The backend to use. Can either be a ``str`` containing the name of a known backend {'scipy'} or an object that implements the uarray protocol. coerce : bool, optional Whether to allow expensive conversions for the ``x`` parameter. e.g., copying a NumPy array to the GPU for a CuPy backend. Implies ``only``. only : bool, optional If only is ``True`` and this backend returns ``NotImplemented``, then a BackendNotImplemented error will be raised immediately. Ignoring any lower priority backends. Examples -------- >>> import scipy.fft as fft >>> with fft.set_backend('scipy', only=True): ... fft.fft([1]) # Always calls the scipy implementation array([1.+0.j]) )coerceonly)rr set_backend)rr"r#r r rr$wsr$cCst|}t|S)aContext manager to skip a backend within a fixed scope. Within the context of a ``with`` statement, the given backend will not be called. This covers backends registered both locally and globally. Upon exit, the backend will again be considered. Parameters ---------- backend : {object, 'scipy'} The backend to skip. Can either be a ``str`` containing the name of a known backend {'scipy'} or an object that implements the uarray protocol. Examples -------- >>> import scipy.fft as fft >>> fft.fft([1]) # Calls default SciPy backend array([1.+0.j]) >>> with fft.skip_backend('scipy'): # We explicitly skip the SciPy backend ... fft.fft([1]) # leaving no implementation available Traceback (most recent call last): ... BackendNotImplementedError: No selected backends had an implementation ... )rr skip_backendr r r rr%sr%)FF) Zscipy._lib.uarrayZ_libZuarrayrrrrrrr!r$r%r r r rs #&