pystencils.backend.ast.axes.PsParallelLoopAxis

class pystencils.backend.ast.axes.PsParallelLoopAxis(range, body, *, num_threads=None, schedule=None, collapse=None)

Parallel loop axis.

Will be lowered to a loop parallelized using OpenMP directives.

Parameters:

• range (PsAxisRange)

• body (PsBlock)

• num_threads (int | None)

• schedule (str | None)

• collapse (int | None)

can_hoist: bool = True

Whether or not this axis type is subject to iteration-invariant code motion.

If set to False, the HoistIterationInvariantDeclarations pass will ignore this axis type.

_clone_structural()

Implementation of structural node cloning.

Return type:

PsStructuralNode