pystencils.codegen.config.GpuOptions

Contents

pystencils.codegen.config.GpuOptions#

class pystencils.codegen.config.GpuOptions(indexing_scheme=None, manual_launch_grid=None, warp_size=None, assume_warp_aligned_block_size=None)#

Configuration options specific to GPU targets.

Methods

__init__([indexing_scheme, ...])

copy()

Perform a semi-deep copy of this configuration object.

default_warp_size(target)

get_option(name)

Get the value set for the specified option, or the option's default value if none has been set.

is_option_set(name)

override(other)

Attributes

assume_warp_aligned_block_size

Specifies whether block sizes are divisible by the hardware's warp size.

indexing_scheme

Thread indexing scheme for dense GPU kernels.

manual_launch_grid

Always require a manually specified launch grid when running this kernel.

warp_size

Specifies the size of a warp (CUDA) or wavefront (HIP).

Parameters: