def add_par_assign_stmt(self, par_assign_stmt: ParAssignStmt) -> "StencilBuilder":
if len(self._vertical_loops) == 0:
self._vertical_loops.append( # TODO builder
VerticalLoop(
interval=Interval(start=AxisBound.start(), end=AxisBound.end()),
loop_order=LoopOrder.FORWARD,
body=[],
temporaries=[],
)
)
self._vertical_loops[-1].body.append(par_assign_stmt)
return self
def __init__(self) -> None:
self._interval = Interval(start=AxisBound.start(), end=AxisBound.end())
self._loop_order = LoopOrder.PARALLEL
self._temporaries = []
self._body = []
def interval(copy_assign):
yield Interval(
loc=SourceLocation(line=2, column=11, source="copy_gtir"),
start=AxisBound(level=LevelMarker.START, offset=0),
end=AxisBound(level=LevelMarker.END, offset=0),
)