pystencils.codegen.gpu_indexing.AutomaticLaunchConfiguration

class pystencils.codegen.gpu_indexing.AutomaticLaunchConfiguration(block_size, grid_size, hw_props, assume_warp_aligned_block_size)

Launch configuration that is dynamically computed from kernel parameters.

This launch configuration permits no further user customization.

Parameters:

• block_size (_Dim3Lambda)

• grid_size (_Dim3Lambda)

• hw_props (HardwareProperties)

• assume_warp_aligned_block_size (bool)

property block_size: tuple[int, int, int] | None

Block size is only available when evaluate is called.

property parameters: frozenset[Parameter]

Parameters of this launch configuration

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]]

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