pystencils.backend.ast.axes.PsGpuThreadAxis

class pystencils.backend.ast.axes.PsGpuThreadAxis(gpu_dimension, xrange, body)

Gpu thread axis.

Set the axis counter as an affine expression of the block-local GPU thread index in the given dimension; i.e. ctr = start + step * threadIdx.{c}. Only execute the body if ctr < stop.

Parameters:

• gpu_dimension (GpuGridDimension)

• xrange (PsAxisRange)

• body (PsBlock)