Skip to content

remove_loops

InlineVertical2DWrite

Bases: ScheduleNodeVisitor

Inline K index value for 2D write vertical while removing for loop.

Transforming: 

for __k = 0; __k < 1; __k = __k + 1:
    map __j, __i:
        field[__i, __j] = tasklet(field_in[__i, __j, __k])

Into 

map __j, __i:
    field[__i, __j] = tasklet(field_in[__i, __j, 0])