heterocl.tvm.stmt module

Statement AST Node in TVM.

User do not need to deal with AST node directly. But they can be helpful for developer to do quick proptyping. While not displayed in the document and python file. Each statement node have subfields that can be visited from python side.

x = tvm.var("n")
a = tvm.var("array", tvm.handle)
st = tvm.make.Store(a, x + 1, 1)
assert isinstance(st, tvm.stmt.Store)
assert(st.buffer_var == a)

class Allocate(handle)

Bases: heterocl.tvm.stmt.Stmt

class AssertStmt(handle)

Bases: heterocl.tvm.stmt.Stmt

class AttrStmt(handle)

Bases: heterocl.tvm.stmt.Stmt

class Block(handle)

Bases: heterocl.tvm.stmt.Stmt

class Break(handle)

Bases: heterocl.tvm.stmt.Stmt

class Evaluate(handle)

Bases: heterocl.tvm.stmt.Stmt

class ExternModule(handle)

Bases: heterocl.tvm.stmt.Stmt

class For(handle)

Bases: heterocl.tvm.stmt.Stmt

Parallel = 1

Pipelined = 4

Serial = 0

Unrolled = 3

Vectorized = 2

class Free(handle)

Bases: heterocl.tvm.stmt.Stmt

class IfThenElse(handle)

Bases: heterocl.tvm.stmt.Stmt

class KernelDef(handle)

Bases: heterocl.tvm.stmt.Stmt

class KernelStmt(handle)

Bases: heterocl.tvm.stmt.Stmt

class LetStmt(handle)

Bases: heterocl.tvm.stmt.Stmt

class Partition(handle)

Bases: heterocl.tvm.stmt.Stmt

Block = 1

Complete = 0

Cyclic = 2

class Prefetch(handle)

Bases: heterocl.tvm.stmt.Stmt

class Print(handle)

Bases: heterocl.tvm.stmt.Stmt

class ProducerConsumer(handle)

Bases: heterocl.tvm.stmt.Stmt

class Provide(handle)

Bases: heterocl.tvm.stmt.Stmt

class Realize(handle)

Bases: heterocl.tvm.stmt.Stmt

class Return(handle)

Bases: heterocl.tvm.stmt.Stmt

class Reuse(handle)

Bases: heterocl.tvm.stmt.Stmt

class Stencil(handle)

Bases: heterocl.tvm.stmt.Stmt

class Stmt(handle)

Bases: heterocl.tvm._ffi.node.NodeBase

class Store(handle)

Bases: heterocl.tvm.stmt.Stmt

class StreamStmt(handle)

Bases: heterocl.tvm.stmt.Stmt

class While(handle)

Bases: heterocl.tvm.stmt.Stmt