pystencils.codegen.gpu_indexing.ManualLaunchConfiguration

class pystencils.codegen.gpu_indexing.ManualLaunchConfiguration(hw_props, assume_warp_aligned_block_size=False)

Manual GPU launch configuration.

This launch configuration requires the user to set block and grid size.

Parameters:

• hw_props (HardwareProperties)

• assume_warp_aligned_block_size (bool)

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

Returns desired block size if available.

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