pystencils.jit.cpu.cpujit.ExtensionModuleBuilderBase

pystencils.jit.cpu.cpujit.ExtensionModuleBuilderBase#

class pystencils.jit.cpu.cpujit.ExtensionModuleBuilderBase#

Base class for CPU extension module builders.

abstractmethod static include_dirs()#

List of directories that must be on the include path when compiling generated extension modules.

The Python runtime include directory and the pystencils include directory need not be listed here.

Return type:

list[str]

abstractmethod render_module(kernel, module_name)#

Produce the extension module code for the given kernel.

Return type:

str

Parameters:
abstractmethod get_wrapper(kernel, extension_module)#

Produce the invocation wrapper for the given kernel and its compiled extension module.

Return type:

KernelWrapper

Parameters:
Contents