pystencils.codegen.GpuKernel

class pystencils.codegen.GpuKernel(body, target, name, parameters, required_headers, jit, launch_config_factory)

Internal representation of a kernel function targeted at CUDA GPUs.

Parameters:

• body (PsBlock)

• target (Target)

• name (str)

• parameters (Sequence[Parameter])

• required_headers (set[str])

• jit (JitBase)

• launch_config_factory (Callable[[], GpuLaunchConfiguration])

get_launch_configuration()

Object exposing the total size of the launch grid this kernel expects to be executed with.

Return type:

GpuLaunchConfiguration