pystencils.jit.cpu.cpujit.ExtensionModuleBuilderBase

class pystencils.jit.cpu.cpujit.ExtensionModuleBuilderBase

Base class for CPU extension module builders.

abstract static include_dirs()

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

Return type:

list[str]

abstract render_module(kernel, module_name)

Produce the extension module code for the given kernel.

Return type:

str

Parameters:

• kernel (Kernel)

• module_name (str)

abstract get_wrapper(kernel, extension_module)

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

Return type:

KernelWrapper

Parameters:

• kernel (Kernel)

• extension_module (ModuleType)