pystencils.codegen.gpu_indexing.GpuLaunchConfiguration

pystencils.codegen.gpu_indexing.GpuLaunchConfiguration#

class pystencils.codegen.gpu_indexing.GpuLaunchConfiguration(hw_props)#

Base class for launch configurations for CUDA and HIP kernels.

Parameters:

  • block_size – A triple of lambdas determining the GPU block size

  • grid_size – A triple of lambdas determining the GPU grid size

  • config_parameters – Set containing all parameters to the given lambdas that are not also parameters to the associated kernel

  • hw_props (HardwareProperties | None)

abstract property parameters: frozenset[Parameter]#

Parameters of this launch configuration

abstract evaluate(**kwargs)#

Compute block and grid size for a kernel launch.

Parameters:

kwargs – Valuation providing a value for each parameter listed in parameters

Return type:

tuple[tuple[int, int, int], tuple[int, int, int]]

abstract jit_cache_key()#

Return a hashable object that represents any user-configurable options of this launch configuration, such that when the configuration changes, the JIT parameter cache is invalidated.

Return type:

Any

Contents