pystencils.codegen.driver.KernelFactory

class pystencils.codegen.driver.KernelFactory(ctx)

Factory for wrapping up backend and IR objects into exportable kernels and function objects.

Parameters:

ctx (KernelCreationContext)

create_lambda(expr)

Create a Lambda from an expression.

Return type:

Lambda

Parameters:

expr (PsExpression)

create_generic_kernel(platform, body, function_name, target_spec, jit)

Create a kernel for a generic target

Return type:

Kernel

Parameters:

• platform (Platform)

• body (PsBlock)

• function_name (str)

• target_spec (Target)

• jit (JitBase)

create_gpu_kernel(platform, body, function_name, target_spec, jit, launch_config_factory)

Create a kernel for a GPU target

Return type:

GpuKernel

Parameters:

• platform (Platform)

• body (PsBlock)

• function_name (str)

• target_spec (Target)

• jit (JitBase)

• launch_config_factory (Callable[[], GpuLaunchConfiguration])