pystencils.backend.ast.axes.PsGpuBlockAxis

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

Gpu block axis.

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

Parameters:

• gpu_dimension (GpuGridDimension)

• xrange (PsAxisRange)

• body (PsBlock)