xlnstorch.register_xlnsconf_implementation

xlnstorch.register_xlnsconf_implementation#

xlnstorch.register_xlnsconf_implementation(xlns_function, impl_key)#

Registers an implementation of the sbdb function using a ufunc from the xlnsconf packed. This allows for experimentation with implementations from xlnsconf that haven’t been ported to xlnstorch yet.

Parameters:

  • xlns_function (Callable) – The xlnsconf ufunc that implements the sbdb function.

  • impl_key (str) – The key to register the sbdb function under. This should be unique across all sbdb implementations.

Raises:

ValueError – If an sbdb function with the given key is already registered.

Return type:

None

Contents