tadashi package

Submodules

tadashi.apps module

tadashi.node_type module

class tadashi.node_type.NodeType(*values)[source]

Bases: Enum

Type of the schedule tree node.

Details: ISL online user manual (Schedule Trees).

BAND = 0

CONTEXT = 1

DOMAIN = 2

EXPANSION = 3

EXTENSION = 4

FILTER = 5

LEAF = 6

GUARD = 7

MARK = 8

SEQUENCE = 9

SET = 10

tadashi.passesparser module

class tadashi.passesparser.PassParser[source]

Bases: object

pass_tree()[source]

Return type:: list

parse(begin, end)[source]

Return type:: list

static reassemble(passes)[source]

find(prefix)[source]

split(locs)[source]

tadashi.passesparser.main()[source]

tadashi.scop module

tadashi.translators module

Module contents

class tadashi.TrEnum(*values)[source]

Bases: StrEnum

Enums of implemented transformations.

One of these enums needs to be passed to Node.transform() (with args) to perform the transformation.

TILE_1D = 'tile_1d'

TILE_2D = 'tile_2d'

TILE_3D = 'tile_3d'

INTERCHANGE = 'interchange'

FULL_FUSE = 'full_fuse'

FUSE = 'fuse'

FULL_SPLIT = 'full_split'

SPLIT = 'split'

SCALE = 'scale'

FULL_SHIFT_VAL = 'full_shift_val'

PARTIAL_SHIFT_VAL = 'partial_shift_val'

FULL_SHIFT_VAR = 'full_shift_var'

PARTIAL_SHIFT_VAR = 'partial_shift_var'

FULL_SHIFT_PARAM = 'full_shift_param'

PARTIAL_SHIFT_PARAM = 'partial_shift_param'

SET_PARALLEL = 'set_parallel'

SET_LOOP_OPT = 'set_loop_opt'