registry module

Full Documentation for hippynn.custom_kernels.registry module. Click here for a summary page.

class MessagePassingKernels(impl_name: str, envsum_impl, sensesum_impl, featsum_impl, wrap=True, compiler=None)

Bases: object

classmethod get_available_implementations(hidden=False)

Return the available implementations of the custom kernels.

Parameters:

hidden – Show all implementations, even those which have no improved performance characteristics.

Returns:

classmethod get_implementation(impl_name)

Parameters:

impl_name

Returns:

:raises CustomKernelError if implementation is not available or known.