Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
85 lines (66 sloc) 3.75 KB
.. currentmodule:: uarray

Documentation for backend providers

Backend providers can provide a back-end for a defined API within the :obj:`uarray` ecosystem. To find out how to define your own API with :obj:`uarray`, see :ref:`mmauthordocs`. To find out how your backend will be provided, use :ref:`userdocs`.

Backend providers need to be aware of three protocols: __ua_domain__, __ua_function__ and __ua_convert__. The first two are mandatory and the last is optional.

__ua_domain__

__ua_domain__ is a string containing the domain of the backend. This is, by convention, the name of the module (or one of its dependencies or parents) that contain the multimethods. For example, scipy and numpy.fft could both use the numpy domain.

__ua_function__

This is the most important protocol, one that defines the implementation of a multimethod. It has the signature (method, args, kwargs). Note that it is called in this form, so if your backend is an object instead of a module, you should add self. method is the multimethod being called, and it is guaranteed that it is in the same domain as the backend. args and kwargs are the arguments to the function, possibly after conversion (explained below)

Returning :obj:`NotImplemented` signals that the backend does not support this operation.

__ua_convert__

All dispatchable arguments are passed through __ua_convert__ before being passed into __ua_function__. This protocol has the signature (dispatchables, coerce), where dispatchables is an iterable of :obj:`Dispatchable` and coerce is whether or not to coerce forcefully. dispatch_type is the mark of the object to be converted, and coerce specifies whether or not to "force" the conversion. By convention, operations larger than O(log n) (where n is the size of the object in memory) should only be done if coerce is True. In addition, there are arguments wrapped as non-coercible via the coercible attribute, if these must be coerced, then one should return NotImplemented.

A convenience wrapper for converting a single object, :obj:`wrap_single_convertor` is provided.

Returning :obj:`NotImplemented` signals that the backend does not support conversion of the given object.

:obj:`skip_backend`

If a backend consumes multimethods from a domain, and provides multimethods for that same domain, it may wish to have the ability to use multimethods while excluding itself from the list of tried backends in order to avoid infinite recursion. This allows the backend to implement its functions in terms of functions provided by other backends. This is the purpose of the :obj:`skip_backend` decorator.

The process that takes place when the backend is tried

First of all, the backend's __ua_convert__ method is tried. If this returns :obj:`NotImplemented`, then the backend is skipped, otherwise its __ua_function__ protocol is tried. If a value other than :obj:`NotImplemented` is returned, it is assumed to be the final return value. Any exceptions raised are propagated up the call stack, except a :obj:`BackendNotImplementedError`, which signals a skip of the backend. If all backends are exhausted, or if a backend with its only flag set to True is encountered, a :obj:`BackendNotImplementedError` is raised.

Examples

Examples for library authors can be found in the source of unumpy.numpy_backend and other *_backend.py files in this directory.

You can’t perform that action at this time.