Skip to content

memlet

MemletCollector

Bases: ScheduleNodeVisitor

Gathers in_memlets and out_memlets of TaskNodes and LibraryCalls.

WriteDependencyCollector

Bases: ScheduleNodeVisitor

Collect write dependency for all data as list of memlet used as inputs in tasklet.

axis_from_backend(backend)

Return axis in order of execution for a given backend

no_data_dependencies_on_cartesian_axis(first, second, axis)

Check for read after write, write after write and write after read, with different offsets.