pystencils.jit.cpu.default_module_builder.DefaultExtensionModuleBuilder

class pystencils.jit.cpu.default_module_builder.DefaultExtensionModuleBuilder(compiler_info)

Parameters:

compiler_info (CompilerInfo)

class ParamExtraction(argstruct_members=<factory>, kernel_kwarg_refs=<factory>, array_proxy_defs=<factory>, extract_kernel_args=<factory>, precondition_checks=<factory>, kernel_invocation_args=<factory>)

Parameters:

• argstruct_members (list[str])

• kernel_kwarg_refs (list[str])

• array_proxy_defs (list[str])

• extract_kernel_args (list[str])

• precondition_checks (list[str])

• kernel_invocation_args (list[str])

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]

render_module(kernel, module_name)

Produce the extension module code for the given kernel.

Return type:

str

Parameters:

• kernel (Kernel)

• module_name (str)

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)