pystencils.backend.ast.axes.PsGpuBlockXThreadAxis

pystencils.backend.ast.axes.PsGpuBlockXThreadAxis#

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

Gpu block x thread axis.

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

Parameters: