brainevent.config.set_backend

Contents

brainevent.config.set_backend#

brainevent.config.set_backend(platform, backend)[source]#

Set the global default backend for a platform across all primitives.

After this call, every primitive that has a kernel registered for backend on platform will use it by default, unless overridden by an explicit backend= keyword argument at call time.

Parameters:

  • platform (str) – The platform name (e.g., 'cpu', 'gpu', 'tpu').

  • backend (str | None) – The backend name (e.g., 'warp', 'pallas', 'numba'). Pass None to clear the global default for this platform, reverting to per-primitive defaults.

Raises:

ValueError – If backend is an empty string.

See also

get_backend

Query the current global backend for a platform.

clear_backends

Clear all global backend defaults.

Examples

>>> import brainevent
>>> brainevent.set_backend('gpu', 'warp')
>>> brainevent.get_backend('gpu')
'warp'
>>> brainevent.set_backend('gpu', None)  # clear
>>> brainevent.get_backend('gpu') is None
True
Contents